Format Files

Author: cheran-senthil

codeforces/api.py

@@ -11,16 +11,13 @@
 
 __all__ = ['call']
 
-
 CODEFORCES_API_URL = "https://codeforces.com/api/"
 
 
 def _generate_api_sig(method, args, secret):
     rand = "%06d" % secrets.randbelow(999999)
     url_args = urllib.parse.urlencode(sorted(args.items()))
-    return rand + hashlib.sha512(
-        ("%s/%s?%s#%s" % (rand, method, url_args, secret)).encode('utf-8')
-    ).hexdigest()
+    return rand + hashlib.sha512(("%s/%s?%s#%s" % (rand, method, url_args, secret)).encode('utf-8')).hexdigest()
 
 
 def call(method, key=None, secret=None, **kwargs):
@@ -56,10 +53,7 @@ def call(method, key=None, secret=None, **kwargs):
 
     with requests.get(url, params=params) as res:
         if res.status_code == 404:
-            data = {
-                'status': 'FAILED',
-                'comment': "%s: No such method" % method
-            }
+            data = {'status': 'FAILED', 'comment': "%s: No such method" % method}
         elif res.status_code in (429, 503):
             time.sleep(1)
             return call(method, key, secret, **kwargs)

codeforces/cf_run.py

@@ -17,37 +17,27 @@ def main(argv=None):
     parser = argparse.ArgumentParser()
 
     parser.add_argument(
-        'contestId',
-        type=int,
-        help=
-        "Id of the contest. It is not the round number. It can be seen in contest URL."
-    )
+        'contestId', type=int, help="Id of the contest. It is not the round number. It can be seen in contest URL.")
 
     parser.add_argument(
         'index',
         type=str,
-        help=
-        "A letter or a letter followed by a digit, that represent a problem index in a contest."
-    )
+        help="A letter or a letter followed by a digit, that represent a problem index in a contest.")
 
-    parser.add_argument(
-        'program', type=str, help="Path to executable that needs to be tested")
+    parser.add_argument('program', type=str, help="Path to executable that needs to be tested")
 
     parser.add_argument(
         '-t',
         '--timeout',
         type=int,
         default=10,
-        help=
-        "Timeout for program in seconds, -1 for no time limit (default: 10)")
+        help="Timeout for program in seconds, -1 for no time limit (default: 10)")
 
     parser.add_argument(
         '-g',
         '--gym',
         action='store_true',
-        help=
-        "If true open gym contest instead of regular contest. (default: false)"
-    )
+        help="If true open gym contest instead of regular contest. (default: false)")
 
     if argv:
         args = parser.parse_args(argv)
@@ -56,8 +46,7 @@ def main(argv=None):
 
     args.timeout = None if args.timeout == -1 else args.timeout
 
-    title, time_limit, memory_limit, sample_tests = problem.get_info(
-        args.contestId, args.index, gym=args.gym)
+    title, time_limit, memory_limit, sample_tests = problem.get_info(args.contestId, args.index, gym=args.gym)
 
     print(title)
     print("time limit per test:", time_limit)

codeforces/problem.py

@@ -39,15 +39,9 @@ def get_info(contest_id, index, gym=False, lang='en'):
 
     """
     if gym:
-        problem_url = urllib.parse.urljoin(
-            CODEFORCES_URL,
-            "gym/%d/problem/%s" % (contest_id, index)
-        )
+        problem_url = urllib.parse.urljoin(CODEFORCES_URL, "gym/%d/problem/%s" % (contest_id, index))
     else:
-        problem_url = urllib.parse.urljoin(
-            CODEFORCES_URL,
-            "contest/%d/problem/%s" % (contest_id, index)
-        )
+        problem_url = urllib.parse.urljoin(CODEFORCES_URL, "contest/%d/problem/%s" % (contest_id, index))
 
     with requests.get(problem_url, params={'lang': lang}) as res:
         soup = BeautifulSoup(res.text, 'html.parser')
@@ -57,13 +51,9 @@ def get_info(contest_id, index, gym=False, 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.get_text('\n').lstrip('\n') for i in soup.find_all("div", class_="input")
-    ]
+    inputs = [i.pre.get_text('\n').lstrip('\n') for i in soup.find_all("div", class_="input")]
 
-    outputs = [
-        i.pre.get_text('\n') for i in soup.find_all("div", class_="output")
-    ]
+    outputs = [i.pre.get_text('\n').lstrip('\n') for i in soup.find_all("div", class_="output")]
 
     sample_tests = zip(inputs, outputs)
 

codeforces/submission.py

@@ -26,9 +26,10 @@ def get_submission(submission_id):
         csrf_token = soup.find('meta', attrs={'name': 'X-Csrf-Token'})['content']
 
         res = sess.post(
-            'https://codeforces.com/data/submitSource',
-            {'submissionId': submission_id, 'csrf_token': csrf_token},
-            headers={'X-Csrf-Token': csrf_token}
-        )
+            'https://codeforces.com/data/submitSource', {
+                'submissionId': submission_id,
+                'csrf_token': csrf_token
+            },
+            headers={'X-Csrf-Token': csrf_token})
 
     return json.loads(res.text)['source']

setup.py

@@ -10,12 +10,9 @@
     long_description=long_description,
     long_description_content_type='text/markdown',
     classifiers=[
-        'Development Status :: 5 - Production/Stable',
-        'Intended Audience :: Developers',
-        'License :: OSI Approved :: MIT License',
-        'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.6',
-        'Programming Language :: Python :: 3.7'
+        'Development Status :: 5 - Production/Stable', 'Intended Audience :: Developers',
+        'License :: OSI Approved :: MIT License', 'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.6', 'Programming Language :: Python :: 3.7'
     ],
     keywords='codeforces',
     url='https://github.com/Mukundan314/python-codeforces',
@@ -25,13 +22,10 @@
     packages=find_packages(exclude=['docs']),
     install_requires=['beautifulsoup4', 'colorama', 'requests'],
     extras_requires={'docs': ['sphinx', 'sphinx_rtd_theme']},
-    entry_points={
-        'console_scripts': ['cf-run = codeforces.cf_run:main']
-    },
+    entry_points={'console_scripts': ['cf-run = codeforces.cf_run:main']},
     python_requires='>=3.6,<4',
     project_urls={
         "Bug Tracker": "https://github.com/Mukundan314/python-codeforces/issues/",
         "Documentation": "https://python-codeforces.readthedocs.io/en/stable/",
         "Source": "https://github.com/Mukundan314/python-codeforces"
-    }
-)
+    })

test/test_problem.py

@@ -9,7 +9,7 @@ def test_normal_problem(self):
 
         title, time_limit, memory_limit, sample_tests = codeforces.problem.get_info(contest_id, index)
 
-        assert(title == 'A. Theatre Square')
-        assert(time_limit == '1 second')
-        assert(memory_limit == '256 megabytes')
-        assert(list(sample_tests) == [('6 6 4', '4')])
+        assert (title == 'A. Theatre Square')
+        assert (time_limit == '1 second')
+        assert (memory_limit == '256 megabytes')
+        assert (list(sample_tests) == [('6 6 4', '4')])