Added get_tag method and fixed error in CodeforcesAPI class

Resolving #5

Author: VadVergasov

codeforces_api/api_requests.py

@@ -13,10 +13,11 @@ class CodeforcesApi(CodeforcesApiRequestMaker):
 
     session = None
 
-    def __init__(self):
+    def __init__(self, api_key=None, secret=None, random_number=1000000):
         """
         Initializing class. All we will need is session to optimize performance.
         """
+        super().__init__(api_key, secret, random_number)
         self.session = requests.Session()
 
     def blog_entry_comments(self, blog_entry_id):

codeforces_api/parse_methods.py

@@ -3,11 +3,13 @@
 """
 import requests
 from lxml import html
+from codeforces_api.api_requests import CodeforcesApi
 
 
 class CodeforcesParser:
 
     session = None
+    problem_tags = dict()
 
     def __init__(self):
         """
@@ -36,3 +38,27 @@ def get_solution(self, contest_id, submit_id):
         if len(code) == 0:
             raise ValueError("Incorrect contest_id or submit_id " + str(code))
         return code[0].replace("\r", "")
+
+    def get_tags(self, contest_id, index):
+        """
+        Get tags of given problem.
+
+        contest_id is number of contest.
+
+        index is number of problem, better to be capital letter. Also could be integer or lowercase letter.
+        """
+        if self.problem_tags == dict():
+            cf_api = CodeforcesApi()
+            for problem in cf_api.problemset_problems()["result"]["problems"]:
+                if str(problem["contestId"]) not in self.problem_tags.keys():
+                    self.problem_tags[str(problem["contestId"])] = dict()
+                self.problem_tags[str(problem["contestId"])][
+                    str(problem["index"])
+                ] = problem["tags"]
+        if isinstance(index, int):
+            index = chr(ord("A") + index)
+        elif isinstance(index, str):
+            if index.isnumeric():
+                index = chr(ord("A") + int(index))
+            index = index.capitalize()
+        return self.problem_tags[str(contest_id)][index]

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="CodeforcesApiPy",
-    version="1.4.0",
+    version="1.4.1",
     description="Implementation of codeforces.com API",
     platforms="any",
     url="https://github.com/VadVergasov/CodeforcesApiPy",

tests/main.py

@@ -8,6 +8,8 @@
 
 MAIN = codeforces_api.CodeforcesApi(conf.API_KEY, conf.API_SECRET)
 
+ANONIM_API = codeforces_api.CodeforcesApi()
+
 PARSER = codeforces_api.CodeforcesParser()
 
 COMMENTS = MAIN.blog_entry_comments(74185)
@@ -31,5 +33,6 @@
 USER_STATUS = MAIN.user_status("VadVergasov")
 
 SOLUTION = PARSER.get_solution(1322, 72628149)
+TAGS = PARSER.get_tags(1322, "D")
 
 print("No errors found!")