codeforces 0.0.6__tar.gz → 0.0.8__tar.gz

{codeforces-0.0.6 → codeforces-0.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codeforces
-Version: 0.0.6
+Version: 0.0.8
 Summary: A simple command line tool to move your competitive programming workflow to your terminal.
 Home-page: https://github.com/Nirlep5252/codeforces-cli
 Author: Nirlep5252
@@ -47,9 +47,10 @@ $ cf --help
 `cf config` - save your username, password, problems-directory \
 `cf contests` - list all the current or upcoming contests \
 `cf contests {ID}` - view all the problems of an ongoing contest \
-`cf parse {Contest ID} {Problem ID | Optional}` - parse the problem and its test cases \
+`cf parse {Contest ID} {Problem ID | Optional} {--lang | Optional}` - parse the problem and its test cases \
 `cf run {FILE}` - check the test cases for the current problem (works based on current directory) \
-`cf submit {FILE}` - submit the problem (requires config) (works based on current directory)
+`cf submit {FILE}` - submit the problem (requires config) (works based on current directory) \
+`cf unsolved` - return the list of all your unsolved problems
 
 #### TODO commands:
 
@@ -59,7 +60,6 @@ $ cf --help
 #### TODO features:
 
 - [ ] Support all languages in run
-- [ ] Add a templating system
 - [ ] A problem recommendation system, maybe?
 
 

{codeforces-0.0.6 → codeforces-0.0.8}/README.md

@@ -34,9 +34,10 @@ $ cf --help
 `cf config` - save your username, password, problems-directory \
 `cf contests` - list all the current or upcoming contests \
 `cf contests {ID}` - view all the problems of an ongoing contest \
-`cf parse {Contest ID} {Problem ID | Optional}` - parse the problem and its test cases \
+`cf parse {Contest ID} {Problem ID | Optional} {--lang | Optional}` - parse the problem and its test cases \
 `cf run {FILE}` - check the test cases for the current problem (works based on current directory) \
-`cf submit {FILE}` - submit the problem (requires config) (works based on current directory)
+`cf submit {FILE}` - submit the problem (requires config) (works based on current directory) \
+`cf unsolved` - return the list of all your unsolved problems
 
 #### TODO commands:
 
@@ -46,5 +47,4 @@ $ cf --help
 #### TODO features:
 
 - [ ] Support all languages in run
-- [ ] Add a templating system
 - [ ] A problem recommendation system, maybe?

{codeforces-0.0.6 → codeforces-0.0.8}/cf/__init__.py

@@ -4,6 +4,8 @@ from .contests import contests
 from .parse import parse
 from .submit import submit
 from .run import run
+from .unsolved import unsolved
+from .edit import edit_cmd
 from rich.console import Console
 from rich.table import Table
 from typing import Dict
@@ -49,3 +51,5 @@ commands.add_command(contests)
 commands.add_command(parse)
 commands.add_command(run)
 commands.add_command(submit)
+commands.add_command(unsolved)
+commands.add_command(edit_cmd)

codeforces-0.0.8/cf/edit.py

@@ -0,0 +1,39 @@
+import click
+import os
+import subprocess
+from rich.console import Console
+from utils import get_config
+
+console = Console()
+
+editor_cmds = {
+    "vscode": "code {path}",
+    "neovim": "nvim {path}",
+    "vim": "vim {path}",
+}
+
+
+@click.command(name="edit")
+@click.argument("contest_id", required=True)
+@click.option("--editor", type=click.Choice(["vscode", "neovim", "vim"]), prompt="Select an editor: ")
+def edit_cmd(contest_id: str, editor: str):
+    conf = get_config(console)
+    if conf is None:
+        return
+
+    cf_dir = conf.get("dir")
+    if cf_dir is None:
+        console.print("[bold red]ERROR: [/]The default directory is not set.\nPlease run the `cf config` command.")
+        return
+
+    if not os.path.isdir(cf_dir):
+        console.print("[bold red]ERROR: [/]The default directory does not exist.\nPlease run the `cf config` command.")
+        return
+
+    contest_dir = os.path.join(cf_dir, contest_id)
+    if not os.path.isdir(contest_dir):
+        console.print(f"\n[bold red]ERROR: [/]The contest directory `{contest_dir}` does not exist.")
+        console.print(f"[bold green]TIP: [/]Use `cf parse {contest_id}` to create the directory and parse the contest.")
+        return
+
+    subprocess.run(editor_cmds[editor].format(path=os.path.join(cf_dir, contest_id)), shell=True)

{codeforces-0.0.6 → codeforces-0.0.8}/cf/parse.py

@@ -3,12 +3,12 @@ import os
 import requests
 from rich.console import Console
 from bs4 import BeautifulSoup
-from utils import get_config
+from utils import get_config, get_bp
 
 console = Console()
 
 
-def parse_problem(contest_id: int, problem: str, cf_dir: str, print_info: bool = True):
+def parse_problem(contest_id: int, problem: str, cf_dir: str, print_info: bool = True, bp: str = "_"):
     slash = "/" if os.name == "posix" else "\\\\"
     r = requests.get(url=f"https://codeforces.com/contest/{contest_id}/problem/{problem}")
     if len(r.history) > 0:
@@ -50,6 +50,15 @@ def parse_problem(contest_id: int, problem: str, cf_dir: str, print_info: bool =
         with open(f"{contest_dir}{slash}{problem}.{i}.output.test", "w") as f:
             f.write(out)
 
+    if bp != "_":
+        bp_text = get_bp(bp)
+        if bp_text is None:
+            console.print(f"[bold red]ERROR: [/]No boilerplate file found for `{bp}`.")
+        else:
+            with open(f"{contest_dir}{slash}{problem}.{bp}", "w") as f:
+                f.write(bp_text)
+                console.print(f"[bold green]INFO: [/]Created boilerplate `{problem}.{bp}` file.")
+
     console.print(f"[bold green]Problem {contest_id} {problem} parsed successfully.[/]\n")
     if print_info:
         console.print(f"Use `cd {contest_dir}` to move the contest directory.")
@@ -59,7 +68,8 @@ def parse_problem(contest_id: int, problem: str, cf_dir: str, print_info: bool =
 @click.command()
 @click.argument("contest_id", required=True)
 @click.argument("problem", default="_", required=False)
-def parse(contest_id: int, problem: str):
+@click.option("--lang", required=False, default="_")
+def parse(contest_id: int, problem: str, lang: str):
     """
     Parse the sample test cases for a problem OR a contest.
     """
@@ -93,6 +103,6 @@ def parse(contest_id: int, problem: str):
         for i, p in enumerate(problems):
             items = p.find_all('td')
             p_id = items[0].a.string.strip().lower()
-            parse_problem(contest_id, p_id, cf_dir, print_info=(i == len(problems) - 1))
+            parse_problem(contest_id, p_id, cf_dir, print_info=(i == len(problems) - 1), bp=lang)
     else:
-        parse_problem(contest_id, problem, cf_dir)
+        parse_problem(contest_id, problem, cf_dir, bp=lang)

{codeforces-0.0.6 → codeforces-0.0.8}/cf/submit.py

@@ -127,17 +127,15 @@ Memory:     [bold white]{sub_mem.strip()}[/]
         sub_data = json.loads(sub['text'])['d']
         live_sub_id = sub_data[1]
         if live_sub_id == sub_id:
-            status = sub_data[6]
-            test_case = sub_data[8]
+            status = sub_data[6].strip()
+            test_case = sub_data[8].strip()
 
             if status == "OK":
                 status_text = "[bold green]ACCEPTED[/]"
-            elif status in ("WRONG_ANSWER", "COMPILATION_ERROR", "RUNTIME_ERROR", "TIME_LIMIT_EXCEEDED", "MEMORY_LIMIT_EXCEEDED", "IDLENESS_LIMIT_EXCEEDED", "SECURITY_VIOLATED", "CRASHED", "INPUT_PREPARATION_CRASHED", "CHALLENGED", "SKIPPED", "PARTIAL", "REJECTED"):
-                status_text = f"[bold red]{' '.join(status.split('_'))}[/] on test case: {test_case}"
             elif status == "TESTING":
                 status_text = f"[bold]Running on test case: {test_case}[/]"
             else:
-                status_text = status
+                status_text = f"[bold red]{' '.join(status.split('_'))}[/] on test case: {test_case}"
 
             timee = sub_data[9]
             memory = int(sub_data[10]) // 1000
@@ -150,4 +148,5 @@ Memory:     [bold]{memory} KB[/]
             live.refresh()
             if status != "TESTING":
                 live.stop()
+                sub_watcher.close()
                 break

codeforces-0.0.8/cf/unsolved.py

@@ -0,0 +1,60 @@
+import click
+from bs4 import BeautifulSoup
+from rich.console import Console
+from rich.table import Table
+from utils import CFClient, get_config
+
+console = Console()
+
+
+@click.command()
+def unsolved():
+    """Show unsolved problems"""
+    conf = get_config(console)
+    if conf is None:
+        return
+
+    if "username" not in conf and "password" not in conf:
+        console.print("[bold red]ERROR: [/]Username and password not found in config file")
+        return
+
+    client = CFClient(conf["username"], conf["password"])
+    if not client.login():
+        console.print("[bold red]ERROR: [/]Login failed")
+        return
+
+    ps = client.session.get("https://codeforces.com/problemset")
+    if ps.status_code != 200:
+        console.print("[bold red]ERROR: [/]Failed to fetch unsolved problems.")
+        return
+
+    soup = BeautifulSoup(ps.text, "html.parser")
+    unsolved_table = soup.find('table', {'class': 'rtable'})
+    if unsolved_table is None:
+        console.print("[bold red]ERROR: [/]Failed to fetch unsolved problems.")
+        return
+    problems = unsolved_table.find_all('tr')[1:]  # type: ignore
+
+    if len(problems) == 0:
+        console.print("[bold green]WOW: [/]You do not have any unsolved problems.")
+        console.print("(This means any problem where you have submitted a solution but it was not accepted.)")
+        console.print("Obviously you will have several problems that you haven't tried.")
+        return
+
+    table = Table(title="Unsolved Problems", show_header=True, header_style="bold green", show_lines=True)
+    table.add_column("Problem ID", style="bright", justify="left", no_wrap=True)
+    table.add_column("Problem Name", style="bright", justify="left", no_wrap=True)
+    table.add_column("Last Submission", style="bright", justify="left", no_wrap=True)
+    for problem in problems:
+        data = problem.find_all('td')
+        _id = data[0].a.string.strip()
+        p_url = data[0].a['href'].strip()
+        name = data[1].a.string.strip()
+        sub_id = data[2].a.string.strip()
+
+        table.add_row(
+            f"[link=https://codeforces.com{p_url}]{_id}[/]", name,
+            f"[link=https://codeforces.com{data[2].a['href'].strip()}]{sub_id}[/]"
+        )
+
+    console.print(table)

{codeforces-0.0.6 → codeforces-0.0.8}/codeforces.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codeforces
-Version: 0.0.6
+Version: 0.0.8
 Summary: A simple command line tool to move your competitive programming workflow to your terminal.
 Home-page: https://github.com/Nirlep5252/codeforces-cli
 Author: Nirlep5252
@@ -47,9 +47,10 @@ $ cf --help
 `cf config` - save your username, password, problems-directory \
 `cf contests` - list all the current or upcoming contests \
 `cf contests {ID}` - view all the problems of an ongoing contest \
-`cf parse {Contest ID} {Problem ID | Optional}` - parse the problem and its test cases \
+`cf parse {Contest ID} {Problem ID | Optional} {--lang | Optional}` - parse the problem and its test cases \
 `cf run {FILE}` - check the test cases for the current problem (works based on current directory) \
-`cf submit {FILE}` - submit the problem (requires config) (works based on current directory)
+`cf submit {FILE}` - submit the problem (requires config) (works based on current directory) \
+`cf unsolved` - return the list of all your unsolved problems
 
 #### TODO commands:
 
@@ -59,7 +60,6 @@ $ cf --help
 #### TODO features:
 
 - [ ] Support all languages in run
-- [ ] Add a templating system
 - [ ] A problem recommendation system, maybe?
 
 

{codeforces-0.0.6 → codeforces-0.0.8}/codeforces.egg-info/SOURCES.txt

@@ -5,9 +5,11 @@ utils.py
 cf/__init__.py
 cf/config.py
 cf/contests.py
+cf/edit.py
 cf/parse.py
 cf/run.py
 cf/submit.py
+cf/unsolved.py
 codeforces.egg-info/PKG-INFO
 codeforces.egg-info/SOURCES.txt
 codeforces.egg-info/dependency_links.txt

{codeforces-0.0.6 → codeforces-0.0.8}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = codeforces
-version = 0.0.6
+version = 0.0.8
 url = https://github.com/Nirlep5252/codeforces-cli
 author = Nirlep5252
 description = A simple command line tool to move your competitive programming workflow to your terminal.

{codeforces-0.0.6 → codeforces-0.0.8}/utils.py

@@ -25,6 +25,19 @@ def get_config(console: Console) -> Optional[dict]:
     return data
 
 
+def get_bp(lang: str) -> Optional[str]:
+    slash = "/" if os.name == "posix" else "\\\\"
+    bp_dir = os.path.expanduser("~") + slash + "cf_boilerplates"
+
+    if not os.path.isdir(bp_dir):
+        return
+    if not os.path.isfile(bp_dir + slash + "template." + lang):
+        return
+
+    with open(bp_dir + slash + "template." + lang, "r") as f:
+        return f.read()
+
+
 class CFClient:
     def __init__(self, username: str, password: str):
         self.username = username