kattis-cli 1.1.0__py3-none-any.whl → 1.1.2__py3-none-any.whl

kattis_cli/client.py

@@ -5,8 +5,8 @@ Object-oriented Kattis API client.
 This class wraps login, submit and status-checking functionality previously
 implemented as module-level functions in `kattis.py`.
 """
-from pdb import run
-from typing import List, Any, final
+
+from typing import List, Any
 import sys
 import os
 import re
@@ -20,8 +20,6 @@ from rich.console import Console
 from rich.align import Align
 from rich.live import Live
 from rich.prompt import Confirm
-import subprocess
-from pathlib import Path
 
 from kattis_cli.utils import languages
 from kattis_cli.utils import config

kattis_cli/download.py

@@ -42,7 +42,7 @@ class DownloadManager:
 
         return root_problem_folder
 
-    def download_html(self, problemid: str) -> str:
+    def download_html(self, problemid: str) -> Any:
         """Download the problem HTML page for a given problem id.
 
         Returns the HTML body as a stripped string when successful or
@@ -167,12 +167,11 @@ class DownloadManager:
         uri = f'{settings.KATTIS_PROBLEM_URL}{problemid}'
         uri += '/file/statement/samples.zip'
 
-        root_problem_folder = self.create_problem_folder(problemid)
-
-        data_folder = root_problem_folder.joinpath('data')
-        data_folder.mkdir(exist_ok=True)
         response = requests.get(uri, allow_redirects=True, timeout=10)
         if response.status_code == 200:
+            root_problem_folder = self.create_problem_folder(problemid)
+            data_folder = root_problem_folder.joinpath('data')
+            data_folder.mkdir(exist_ok=True)
             zip_path = data_folder.joinpath('samples.zip')
             with open(data_folder.joinpath('samples.zip'), 'wb') as f:
                 f.write(response.content)
@@ -197,7 +196,7 @@ def create_problem_folder(problemid: str) -> Path:
     return _manager.create_problem_folder(problemid)
 
 
-def download_html(problemid: str) -> str:
+def download_html(problemid: str) -> Any:
     """Download problem HTML using the module-level manager."""
 
     return _manager.download_html(problemid)

kattis_cli/fireworks.py

@@ -11,8 +11,6 @@ import sys
 import locale
 import time
 from random import randint, choice
-import subprocess
-from pathlib import Path
 
 
 def run_fireworks() -> None:
@@ -21,10 +19,9 @@ def run_fireworks() -> None:
     Draws random bursts in the terminal using ANSI colors. Works in CI/Docker
     and avoids external dependencies.
     """
-    frames = 40
+    frames = 5
     width = 60
     height = 20
-    message = None
     symbols = ['*', '•', '✶', '✸']
     colors = [
         '\x1b[37m',  # white
@@ -61,12 +58,6 @@ def run_fireworks() -> None:
         # Simple ASCII fallback symbols
         symbols = ['*', '+', 'o', '.']
         # Replace message with ASCII-only fallback if necessary
-        if message:
-            try:
-                message.encode('ascii')
-            except (UnicodeEncodeError, TypeError):
-                message = (message.encode('ascii', 'replace')
-                           .decode('ascii'))
         # Inform interactive users why we fell back (but avoid noisy logs
         # in captured/non-interactive runs).
         if sys.stdout.isatty():
@@ -108,7 +99,7 @@ def run_fireworks() -> None:
             time.sleep(0.12)
 
         # final message
-        msg = message or '🎉 Congratulations — Accepted! 🎉'
+        msg = '🎉 Congratulations — Accepted! 🎉'
         sys.stdout.write('\n')
         if use_color:
             sys.stdout.write('\x1b[1;33m')

kattis_cli/main.py

@@ -1,13 +1,13 @@
 """ Main module for the kattis_cli package.
 This is the main.py file for the kattis_cli package.
 
-Change the contents here instead of main.py.
+Change the contents in dev.py instead of in main.py.
 build.sh script copies the contents of this file to main.py.
 
 Change the __version__ to match in pyproject.toml
 Has to be higher than the pypi version.
 """
-__version__ = '1.1.0'
+__version__ = '1.1.2'
 
 from math import inf
 from typing import Tuple
@@ -45,12 +45,11 @@ def get(problemid: str) -> None:
         download.download_sample_data(problemid)
     except requests.exceptions.InvalidURL:
         console.print(
-            f"""Sample data for Problem ID: [bold blue]
-            {problemid}[/bold blue] not found.")
-            """)
-    console.print(
-        f"Downloading metadata: [bold blue]{problemid}[/bold blue]")
-    download.load_problem_metadata(problemid)
+            f"Problem ID: [bold red]{problemid}[/bold red] not found.")
+    else:
+        console.print(
+            f"Downloading metadata: [bold blue]{problemid}[/bold blue]")
+        download.load_problem_metadata(problemid)
 
 
 @main.command(help='Show problem metadata.')

kattis_cli/utils/languages.py

@@ -13,6 +13,8 @@ from kattis_cli.utils.utility import find_problem_root_folder
 
 LANGUAGE_GUESS = {
     '.c': 'c',
+    '.h': 'c',
+    '.hpp': 'cpp',
     '.c++': 'cpp',
     '.cc': 'cpp',
     '.c#': 'csharp',
@@ -129,6 +131,8 @@ def guess_language(ext: str, files: List[str]) -> str:
             return "c"
         else:
             return "cpp"
+    if ext == '.hpp':
+        return "cpp"
     if ext == ".py":
         return "python3"
     return LANGUAGE_GUESS.get(ext, '')
@@ -314,19 +318,21 @@ filename extension "{ext}"''')
     return problemid, loc_language, mainclass, files, root_folder, lang_config
 
 
-def get_coding_files() -> List[str]:
+def get_coding_files(base_path: Path = Path.cwd()) -> List[str]:
     """Get coding files from current directory.
 
     Returns:
         List[str]: List of coding files.
     """
-    cur_folder = str(Path.cwd())
+
+    if Path(base_path / "src").is_dir():
+        base_path = base_path / "src"
     console = Console()
-    files = [
-        f for f in os.listdir(cur_folder) if valid_extension(f)]
+    files = [str(f) for f in base_path.rglob(
+        '*') if f.is_file() and valid_extension(str(f))]
     if not files:
         console.print(
-            'No source file(s) found in the current folder!',
+            f'No source file(s) found in {base_path}!',
             style='bold red')
         exit(1)
     return files

kattis_cli/utils/utility.py

@@ -34,6 +34,8 @@ def find_problem_root_folder(
         Returns:
             bool: True if path exists, False otherwise
         """
+        if filename.startswith('/'):
+            filename = Path(filename).name
         for file in path.glob(filename):
             name, ext = os.path.splitext(file.name)
             folder_name = path.parts[-1]

{kattis_cli-1.1.0.dist-info → kattis_cli-1.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kattis-cli
-Version: 1.1.0
+Version: 1.1.2
 Summary: A command-line tool for Kattis
 License-File: LICENSE
 Author: Ram Basnet

{kattis_cli-1.1.0.dist-info → kattis_cli-1.1.2.dist-info}/RECORD

@@ -1,21 +1,21 @@
 kattis_cli/.kattis-cli.toml,sha256=kGh0gmpFAnivyUC9hR8Y1ZqYJeTBWNyBQ9zOg4AInZ0,898
 kattis_cli/__init__.py,sha256=afN92pog2fGyicY6KNBofYbCBYbYj4Fpi_INUSpsc-E,402
-kattis_cli/client.py,sha256=sg6nPkLLFchnGeReR0x6fjazyDfRj78wh48e-6ng5ZA,17007
-kattis_cli/download.py,sha256=IFjebvvxJyPKfsK8Voo06llSsybBqqylUWEv5lF2J7o,8651
-kattis_cli/fireworks.py,sha256=eBeDZuSKEpAwX2Z2RIzIQKKoaIItepFhAnV5juJ2GXg,4444
+kattis_cli/client.py,sha256=5ef-pq737q-VBBxY1F-bEbP9Mf8145YppL4m9NlVCVY,16938
+kattis_cli/download.py,sha256=BATQskOUFyc286vDYY7S4nmBWPBrYsgHwCE1VBsltx0,8662
+kattis_cli/fireworks.py,sha256=yKFXIvDDdNrzBhpEt2Pdy5zgfYjJffoOEf2yUFlu2TQ,4134
 kattis_cli/kattis.py,sha256=wpnM6sC0a32kJYzEux27izug59bicexNnOFxcx0qWM8,2621
 kattis_cli/kattis_setup.py,sha256=mic79GjOI_rC6Lfl5D5ORDMWdrcxR71ksYUM51yezu8,5659
-kattis_cli/main.py,sha256=EqLNAcDBug6atUOnKavlXhBMtmcy5fUQ7Ti3-JWLrps,4511
+kattis_cli/main.py,sha256=jRANFixHzJdnACp1Mzi5p5wxFrSF6FXFrPxLhn8olSc,4491
 kattis_cli/settings.py,sha256=d5q4dYj9VqDSqPalleh2oZWtND-1bPB0T2IwdajFrBg,591
 kattis_cli/solution_tester.py,sha256=hTXtryYs8lkO9CkSPtQNjhrAPPLaC4au5ueCQ3ioZWQ,7577
 kattis_cli/ui.py,sha256=2qFWz9QB7cgjLfZKHrWUrKFVYLRLwl0c37ZTNljwmAI,4580
 kattis_cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kattis_cli/utils/config.py,sha256=6SIuiXA9sFADo8RZUj8A6rXb3uqmrWrmmArfZrqNQdw,2293
-kattis_cli/utils/languages.py,sha256=wlhl1Zng0gQAtsWW6FLxdkdzqeeY_qQ94wqa3NLypEU,8758
+kattis_cli/utils/languages.py,sha256=gcJJJwkPS1Zpg76atwoo2O2Vd7ETKBQV-KD_cwclrmE,8928
 kattis_cli/utils/run_program.py,sha256=NWQ6vtTeWgkaW75r91FIHGXR5cAbeu8yMb5hwzpYFsg,2613
-kattis_cli/utils/utility.py,sha256=iPmfsHi-DwYxP9mKU_VurNnLJXf0dngwYPxUdqckgYo,2907
-kattis_cli-1.1.0.dist-info/METADATA,sha256=Og_q8RDL3hD-jUSfzZjlxhLvUD75avhxfqxRL2Qs1JU,6962
-kattis_cli-1.1.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-kattis_cli-1.1.0.dist-info/entry_points.txt,sha256=kyzGN20VqUPR_H0J_jJUKT-10-cAMFLVegQ6C7tbHss,47
-kattis_cli-1.1.0.dist-info/licenses/LICENSE,sha256=JmBa4SEKBCDWEgiOZcISU4tUCpli6xSpVlSYgkBXSNQ,1067
-kattis_cli-1.1.0.dist-info/RECORD,,
+kattis_cli/utils/utility.py,sha256=oUbydPbcC2wXvfku21YAWpaJjuBKaBRVybMAWCSfM9Y,2987
+kattis_cli-1.1.2.dist-info/METADATA,sha256=SM1Ah1m_9WvKo-glbw1Xx-BnEwUSFFhxU6oCViGjUxc,6962
+kattis_cli-1.1.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+kattis_cli-1.1.2.dist-info/entry_points.txt,sha256=kyzGN20VqUPR_H0J_jJUKT-10-cAMFLVegQ6C7tbHss,47
+kattis_cli-1.1.2.dist-info/licenses/LICENSE,sha256=JmBa4SEKBCDWEgiOZcISU4tUCpli6xSpVlSYgkBXSNQ,1067
+kattis_cli-1.1.2.dist-info/RECORD,,