feat(problem): Add support for gym problems

Author: Mukundan314

codeforces/problem.py

@@ -10,7 +10,7 @@
 CODEFORCES_URL = "https://codeforces.com"
 
 
-def get_info(contest_id, index, lang='en'):
+def get_info(contest_id, index, gym=False, lang='en'):
     """
     Get info for a contest problem.
 
@@ -23,6 +23,9 @@ def get_info(contest_id, index, lang='en'):
         Usually a letter of a letter, followed by a digit, that
         represent a problem index in a contest. It can be seen in
         problem URL. For example: /contest/566/**A**
+    gym: bool
+        If true gym problem is returned otherwise regular problem is
+        returned.
 
     Returns
     -------
@@ -36,10 +39,16 @@ def get_info(contest_id, index, lang='en'):
         Sample tests given for the problem.
 
     """
-    problem_url = os.path.join(
-        CODEFORCES_URL,
-        "contest/%d/problem/%s?lang=%s" % (contest_id, index, lang)
-    )
+    if gym:
+        problem_url = os.path.join(
+            CODEFORCES_URL,
+            "gym/%d/problem/%s?lang=%s" % (contest_id, index, lang)
+        )
+    else:
+        problem_url = os.path.join(
+            CODEFORCES_URL,
+            "contest/%d/problem/%s?lang=%s" % (contest_id, index, lang)
+        )
 
     with urllib.request.urlopen(problem_url) as res:
         soup = BeautifulSoup(res.read(), 'html.parser')
@@ -49,8 +58,13 @@ def get_info(contest_id, index, lang='en'):
     time_limit = soup.find_all("div", class_="time-limit")[0].text[19:]
     memory_limit = soup.find_all("div", class_="memory-limit")[0].text[21:]
 
-    inputs = [i.pre.text.lstrip('\n') for i in soup.find_all("div", class_="input")]
-    outputs = [i.pre.text.lstrip('\n') for i in soup.find_all("div", class_="output")]
+    inputs = [
+        i.pre.text.lstrip('\n') for i in soup.find_all("div", class_="input")
+    ]
+
+    outputs = [
+        i.pre.text.lstrip('\n') for i in soup.find_all("div", class_="output")
+    ]
 
     sample_tests = zip(inputs, outputs)