Added termforces module

termforces/cmds/__init__.py

@@ -0,0 +1,2 @@
+from termforces.cmds.session_manager import *
+from termforces.cmds.submit_interface import *

termforces/cmds/session_manager.py

@@ -0,0 +1,43 @@
+import click
+import os
+from termforces.termforces_shell import termforces_shell, State
+from termforces.session_manager import load_session, store_session
+from getpass import getpass
+from codeforces_scraper import ScraperError
+
+
+@termforces_shell.command()
+@click.argument('handle')
+@click.option('--password',
+              help="Will be prompted, so do not pass, unless you're sure for safety")
+@click.option('--session-file', help='Tries to store session on successful login')
+@click.option('--no-getpass', is_flag=True,
+              show_default=True, default=False, help='Read password from stdin instead of using getpass')
+@click.pass_obj
+def login(state: State, handle, password, session_file, no_getpass):
+    if password is None:
+        if no_getpass:
+            password = input()
+        else:
+            password = getpass(f'Codeforces password for {handle}: ')
+    try:
+        state.scraper.login(handle, password)
+    except ScraperError:
+        print('Failed to login, check your credentials')
+        return
+    if session_file is not None:
+        store_session(state.scraper, os.path.expanduser(session_file))
+
+
+@termforces_shell.command(name='load-session')
+@click.argument('session_file')
+@click.pass_obj
+def load_session_cmd(state: State, session_file):
+    load_session(state.scraper, os.path.expanduser(session_file))
+
+
+@termforces_shell.command(name='whoami')
+@click.pass_obj
+def whoami(state: State):
+    state.scraper.update_current_user()
+    print(state.scraper.current_user)

termforces/cmds/submit_interface.py

@@ -0,0 +1,64 @@
+import click
+from termforces.termforces_shell import termforces_shell, State
+from codeforces_scraper import Verdict
+from codeforces_scraper.languages import some_compiler_by_ext
+from termforces.utils import tcolors, str_from_timestamp
+
+
+@termforces_shell.command(name='enter-contest')
+@click.argument('contest_id')
+@click.pass_obj
+def enter_contest(state: State, contest_id: int):
+    state.contest_id = contest_id
+
+
+@termforces_shell.command(name='submit')
+@click.argument('problem_index')
+@click.argument('source_file')
+@click.option('--contest-id')
+@click.option('--lang-code')
+@click.pass_obj
+def submit(state: State, problem_index, source_file, contest_id, lang_code):
+    if contest_id is None:
+        if state.contest_id is not None:
+            contest_id = state.contest_id
+        else:
+            print('Specify contest id or enter contest via enter-contest command')
+            return
+    if lang_code is None:
+        ext = '.' + source_file.split('.')[-1]
+        compiler = some_compiler_by_ext(ext)
+        if compiler is None:
+            print('Please specify language code')
+            return
+        lang_code = compiler.id
+    with open(source_file, 'r') as f:
+        source_code = f.read()
+    state.scraper.submit(contest_id, problem_index, source_code, lang_code)
+
+
+@termforces_shell.command(name='results-my')
+@click.option('--contest-id')
+@click.pass_obj
+def results_my(state: State, contest_id):
+    if contest_id is None:
+        if state.contest_id is not None:
+            contest_id = state.contest_id
+        else:
+            print('Specify contest id or enter contest via enter-contest command')
+            return
+    if state.scraper.current_user is None:
+        print('Please login to view your results')
+        return
+    subms = state.scraper.get_submissions(contest_id, state.scraper.current_user)
+    subms.sort(key=lambda subm: (subm.problem.index, subm.creation_time_seconds))
+
+    for subm in subms:
+        if subm.verdict == Verdict.TESTING:
+            color = tcolors.WARNING
+        elif subm.verdict == Verdict.OK:
+            color = tcolors.OKGREEN
+        else:
+            color = tcolors.FAIL
+        verdict_str = f'{color}{subm.verdict.name}{tcolors.ENDC}'
+        print(f'{subm.problem.index} \t {str_from_timestamp(subm.creation_time_seconds)} \t {verdict_str}')