codeforces 0.0.7__tar.gz → 0.0.8__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codeforces
-Version: 0.0.7
+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

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

@@ -5,6 +5,7 @@ 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
@@ -51,3 +52,4 @@ 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.7 → codeforces-0.0.8}/cf/submit.py

@@ -127,8 +127,8 @@ 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[/]"
@@ -148,4 +148,5 @@ Memory:     [bold]{memory} KB[/]
             live.refresh()
             if status != "TESTING":
                 live.stop()
+                sub_watcher.close()
                 break

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codeforces
-Version: 0.0.7
+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

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

@@ -5,6 +5,7 @@ utils.py
 cf/__init__.py
 cf/config.py
 cf/contests.py
+cf/edit.py
 cf/parse.py
 cf/run.py
 cf/submit.py

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

@@ -1,6 +1,6 @@
 [metadata]
 name = codeforces
-version = 0.0.7
+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.