Improved get_tags method, added comments

VadVergasov · Jul 6, 2024 · 8a52dc2 · 8a52dc2
1 parent 976374f
commit 8a52dc2
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/codeforces_api/parse_methods.py b/codeforces_api/parse_methods.py
@@ -92,6 +92,11 @@ def get_tags(self, contest_id, index, include_rating=False):
                 index = chr(ord("A") + int(index))
             index = index.capitalize()
         try:
-            return sorted(self.problem_tags[str(contest_id)][index])
+            if include_rating:
+                return self.problem_tags[str(contest_id)][index]
+            return self.problem_tags[str(contest_id)][index][1:]
         except KeyError:
-            return sorted(self.problem_tags[str(int(contest_id) - 1)][index])
+            # If problem included in more than one division.
+            if include_rating:
+                return self.problem_tags[str(int(contest_id) - 1)][index]
+            return self.problem_tags[str(int(contest_id) - 1)][index][1:]