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

kattis_cli/kattis_templates/main/c.txt

@@ -0,0 +1,8 @@
+#include <stdio.h>
+#include <stdlib.h>
+
+int main() {
+    // Your solution here
+
+    return 0;
+}

kattis_cli/kattis_templates/main/cpp.txt

@@ -0,0 +1,15 @@
+#include <iostream>
+#include <vector>
+#include <string>
+#include <algorithm>
+
+using namespace std;
+
+int main() {
+    ios_base::sync_with_stdio(false);
+    cin.tie(NULL);
+
+    // Your solution here
+
+    return 0;
+}

kattis_cli/kattis_templates/main/java.txt

@@ -0,0 +1,14 @@
+import java.util.Scanner;
+
+public class {class_name} {
+    public static void main(String[] args) {
+        Scanner scanner = new Scanner(System.in);
+
+        // Your solution here
+        // if (scanner.hasNext()) {
+        //     String line = scanner.next();
+        // }
+
+        scanner.close();
+    }
+}

kattis_cli/kattis_templates/main/python.txt

@@ -0,0 +1,21 @@
+"""Python template
+"""
+import sys
+
+
+def main() -> None:
+    """Main function
+    """
+    # Read all lines from stdin
+    input_data = sys.stdin.read().split()
+    if not input_data:
+        return
+
+    # Example: reading an integer
+    # n = int(input_data[0])
+
+    # Your solution here
+
+
+if __name__ == "__main__":
+    main()

kattis_cli/kattis_templates/main/python3.txt

@@ -0,0 +1,22 @@
+"""Python template
+"""
+import sys
+
+
+def main() -> None:
+    """Main function
+    """
+    # Read all lines from stdin
+    input_data = sys.stdin.read().split()
+    if not input_data:
+        return
+
+    # Example: reading an integer
+    # n = int(input_data[0])
+
+    # Your solution here
+
+
+if __name__ == "__main__":
+    name()
+    

kattis_cli/main.py

@@ -1,15 +1,11 @@
 """ Main module for the kattis_cli package.
-This is the main.py file for the kattis_cli package.
-
-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.2'
+__version__ = '1.2.0'
 
 from math import inf
+import os
 from typing import Tuple
 from rich.console import Console
 import click
@@ -21,6 +17,7 @@ import kattis_cli.solution_tester as solution_tester
 import kattis_cli.kattis as kattis
 import kattis_cli.utils.languages as languages
 import kattis_cli.kattis_setup as kattis_setup
+import kattis_cli.template as template
 
 
 @tui()
@@ -43,13 +40,16 @@ def get(problemid: str) -> None:
         f"Downloading samples: [bold blue]{problemid}[/bold blue]")
     try:
         download.download_sample_data(problemid)
+        console.print(
+            f"Downloading metadata: [bold blue]{problemid}[/bold blue]")
+        download.load_problem_metadata(problemid)
     except requests.exceptions.InvalidURL:
         console.print(
             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)
+            f"Downloaded samples and metadata: \
+[bold blue]{problemid}[/bold blue]")
 
 
 @main.command(help='Show problem metadata.')
@@ -115,7 +115,6 @@ def submit(problemid: str, language: str,
         languages.update_args(
             problemid, language, mainclass, list(files))
     # Finally, submit the solution
-    # print(f'{problemid=} {language=} {mainclass=} {tag=} {force=} {_files=}')
     if not mainclass:
         mainclass = languages.guess_mainfile(
             language, _files, problemid, lang_config)
@@ -132,5 +131,21 @@ def setup() -> None:
     kattis_setup.setup()
 
 
+@main.command(help='Create a template file for a problem.')
+@click.option('-l', '--language', default='python3',
+              help='Programming language (default: python3)')
+@click.option('-p', '--problemid', default=None,
+              help='Problem ID (default: parent directory name)')
+@click.option('-s', '--src', is_flag=True, default=False,
+              help='Create src layout structure')
+def template_cmd(language: str, problemid: str, src: bool) -> None:
+    """Create a template file.
+    """
+    if not problemid:
+        problemid = os.path.basename(os.getcwd())
+
+    template.create_template(language, problemid, src)
+
+
 if __name__ == '__main__':
     main()

kattis_cli/solution_tester.py

@@ -91,6 +91,10 @@ class SolutionTester:
         total = len(in_files)
         console.clear()
         title = f"[not italic bold blue]👷‍ Testing {mainclass} "
+        main_src_file = next((f for f in files if f.endswith(mainclass)), None)
+        if not main_src_file:
+            main_src_file = mainclass
+
         title += f" using {loc_language} 👷‍[/]"
         table.title = title
         with Live(table_centered, console=console,
@@ -145,7 +149,7 @@ class SolutionTester:
                         expected = b"No .ans or .out file found!"
                 code, ans, error = run_program.run(
                     lang_config,
-                    mainclass,
+                    main_src_file,
                     in_file,
                 )
                 if code != 0:
@@ -159,7 +163,10 @@ class SolutionTester:
                     result = "[bold red]❌[/bold red]"
 
                 in_filename = Path(in_file).parts[-1]
-                out_filename = Path(out_file).parts[-1]
+                if b"No .ans or .out file found!" == expected:
+                    out_filename = "N/A"
+                else:
+                    out_filename = Path(out_file).parts[-1]
                 time.sleep(0.1)
                 table.add_row(in_filename,
                               input_content.decode('utf-8'),

kattis_cli/template.py

@@ -0,0 +1,161 @@
+"""Module for generating project templates."""
+
+import os
+import shutil
+from pathlib import Path
+import importlib.resources as pkg_resources
+from rich.console import Console
+
+
+LANGUAGE_EXTENSIONS = {
+    'c': '.c',
+    'cpp': '.cpp',
+    'csharp': '.cs',
+    'cobol': '.cob',
+    'fsharp': '.fs',
+    'go': '.go',
+    'haskell': '.hs',
+    'java': '.java',
+    'nodejs': '.js',
+    'typescript': '.ts',
+    'kotlin': '.kt',
+    'lisp': '.lisp',
+    'objective-c': '.m',
+    'ocaml': '.ml',
+    'pascal': '.pas',
+    'php': '.php',
+    'prolog': '.pl',
+    'python': '.py',
+    'python3': '.py',
+    'ruby': '.rb',
+    'rust': '.rs',
+    'scala': '.scala',
+    'fortran': '.f90',
+    'bash': '.sh',
+    'apl': '.apl',
+    'gerbil': '.ss',
+    'julia': '.jl',
+    'vb': '.vb',
+    'dart': '.dart',
+    'zig': '.zig',
+    'swift': '.swift',
+    'nim': '.nim',
+}
+
+
+def create_template(
+        language: str,
+        problemid: str,
+        src_layout: bool = False) -> None:
+    """Create a template file for the specified language and problem ID.
+
+    Args:
+        language (str): The programming language (e.g., 'python3', 'cpp').
+        problemid (str): The problem ID, used for the filename.
+        src_layout (bool): If True, create a src/ directory structure.
+    """
+    language = language.lower()
+    home_templates = os.path.expanduser('~/kattis_templates')
+
+    def _check_kattis_templates() -> bool:
+        # Always use ~/kattis_templates as the user-facing template directory
+        # On first use, if not present, copy all default templates from
+        # package data
+        if not os.path.isdir(home_templates):
+            try:
+                # Use importlib.resources to access kattis_templates as a
+                # package resource
+                src = pkg_resources.files("kattis_cli") / "kattis_templates"
+                shutil.copytree(str(src), home_templates)
+            except Exception as e:
+                console.print(
+                    f"[bold red]❌ Error:[/bold red] Could not \
+copy templates: {e}")
+                return False
+        return True
+
+    def _get_main_template(language: str) -> str:
+        main_templates = Path(home_templates).joinpath('main')
+        for file in main_templates.iterdir():
+            # print(file.stem.lower(), language)
+            if file.is_file() and file.stem.lower() == language:
+                return str(file)
+        return ''
+
+    console = Console()
+    if not _check_kattis_templates():
+        console.print(
+            f"[bold red]❌ Error:[/bold red] Templates '{language}' \
+not found in {home_templates}!")
+        return
+
+    def _copytree_with_problemid(src: str,
+                                 dst: str,
+                                 problemid: str) -> None:
+        for root, _, files in os.walk(src):
+            rel = os.path.relpath(root, src)
+            rel = rel.replace('problemid', problemid)
+            target_root = os.path.join(dst, rel) if rel != '.' else dst
+            os.makedirs(target_root, exist_ok=True)
+            for f in files:
+                src_file = os.path.join(root, f)
+                dst_file = os.path.join(
+                    target_root, f.replace('problemid', problemid))
+                shutil.copy2(src_file, dst_file)
+
+    # If src_layout, copy the project structure for the language
+    if src_layout:
+        proj_struct_root = Path(home_templates).joinpath('project_structure')
+        lang_src_struct = proj_struct_root.joinpath(language)
+        if os.path.isdir(lang_src_struct):
+            _copytree_with_problemid(
+                str(lang_src_struct), os.getcwd(), problemid)
+            console.print(
+                f"[bold blue]✅ [/bold blue] Created project \
+structure for {language}.")
+        target_dir = os.path.join("src", problemid)
+    else:
+        target_dir = "."
+
+    main_template = _get_main_template(language=language)
+    extension = LANGUAGE_EXTENSIONS.get(language, None)
+
+    try:
+        with open(str(main_template), 'r', encoding='utf-8') as tf:
+            template_content = tf.read()
+    except OSError:
+        console.print(
+            f"[bold red] ❌ [/bold red] Could not read \
+template file: {main_template}")
+        return
+
+    if not main_template:
+        return
+    # Handle Java class naming convention (capitalize first letter)
+    if language == 'java':
+        class_name = problemid[0].upper(
+        ) + problemid[1:] if problemid else "Main"
+        filename = f"{class_name}{extension}"
+        content = template_content.replace("{class_name}", class_name)
+    else:
+        filename = f"{problemid}{extension}"
+        content = template_content
+
+    filepath = os.path.join(target_dir, filename)
+
+    if os.path.exists(filepath) and os.path.getsize(filepath) > 0:
+        console.print(
+            f"[bold yellow]Warning:[/bold yellow] File '{filepath}' \
+exists with content. Skipping.")
+        return
+
+    try:
+        with open(filepath, 'w', encoding='utf-8') as f:
+            f.write(content)
+        console.print(
+            f"[bold green]✅ [/bold green] Created template \
+'{filepath}' for {language}.")
+    except OSError as e:
+        console.print(
+            f"[bold red]❌ [/bold red] Failed to create template \
+'{filepath}': {e}")

kattis_cli/utils/config.py

@@ -11,7 +11,7 @@ import yaml
 
 from kattis_cli.utils.utility import find_problem_root_folder
 
-_DEFAULT_CONFIG = Path.home().joinpath('.kattisrc')
+_KATTIS_RC = Path.home().joinpath('.kattisrc')
 
 
 class ConfigError(Exception):
@@ -42,8 +42,8 @@ def get_kattisrc() -> configparser.ConfigParser:
     """Returns a ConfigParser object for the .kattisrc file(s)
     """
     cfg = configparser.ConfigParser()
-    if os.path.exists(_DEFAULT_CONFIG):
-        cfg.read(_DEFAULT_CONFIG)
+    if os.path.exists(_KATTIS_RC):
+        cfg.read(_KATTIS_RC)
 
     if not cfg.read([os.path.join(os.path.expanduser("~"), '.kattisrc'),
                      os.path.join(os.path.dirname(sys.argv[0]), '.kattisrc')]):

kattis_cli/utils/languages.py

@@ -271,7 +271,7 @@ def update_args(problemid: str,
 
     console = Console()
     if not files:
-        files = get_coding_files()
+        files = get_coding_files(Path.cwd())
     # check if problemid is given
     # if not problemid:
     cur_folder = Path.cwd()
@@ -318,13 +318,14 @@ filename extension "{ext}"''')
     return problemid, loc_language, mainclass, files, root_folder, lang_config
 
 
-def get_coding_files(base_path: Path = Path.cwd()) -> List[str]:
+def get_coding_files(base_path: Path) -> List[str]:
     """Get coding files from current directory.
 
     Returns:
         List[str]: List of coding files.
     """
-
+    if not base_path:
+        base_path = Path.cwd()
     if Path(base_path / "src").is_dir():
         base_path = base_path / "src"
     console = Console()

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

@@ -1,19 +1,14 @@
 Metadata-Version: 2.4
 Name: kattis-cli
-Version: 1.1.2
+Version: 1.2.0
 Summary: A command-line tool for Kattis
 License-File: LICENSE
 Author: Ram Basnet
 Author-email: rbasnet@coloradomesa.edu
 Requires-Python: >=3.10.0,<4.0.0
+Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: beautifulsoup4 (>=4.12.2,<5.0.0)
 Requires-Dist: click (>=8.1.7,<9.0.0)
 Requires-Dist: lxml (>=6.0.1,<7.0.0)
@@ -138,6 +133,24 @@ kattis info
 
 ![Problem info](./images/kattis_info.png)
 
+### Creating project template
+
+- create project template files for a given programming language
+- src style default templates are created in the home directory under **.kattis-templates** folder
+- you can overwrite/create new templates following the structure provided `kattis_templates`
+- local languages codes supported are:
+  - python3
+  - c++
+  - java
+  - c
+  - nodejs
+
+```bash
+kattis template --help
+# create python3 project template for cold problem
+kattis template -l python3 -p cold -s
+```
+
 ### Test a solution locally
 
 ![Test](images/kattis_test.png)

kattis_cli-1.2.0.dist-info/RECORD

@@ -0,0 +1,38 @@
+kattis_cli/.kattis-cli.toml,sha256=kGh0gmpFAnivyUC9hR8Y1ZqYJeTBWNyBQ9zOg4AInZ0,898
+kattis_cli/__init__.py,sha256=afN92pog2fGyicY6KNBofYbCBYbYj4Fpi_INUSpsc-E,402
+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/kattis_templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/main/c.txt,sha256=HIJF1XMi40XbaC63o7kqfInvK_WvO6pPkST_xjmPPw0,96
+kattis_cli/kattis_templates/main/cpp.txt,sha256=TmVm0vX1YZyzG3P0p2doix-47bofS0xiZeKiPbmNcWM,214
+kattis_cli/kattis_templates/main/java.txt,sha256=4-485QJ0sV1RczRV-VxKYc6El2xD16y2oUAUEQpu9Fg,309
+kattis_cli/kattis_templates/main/python.txt,sha256=RTDUEN2sSp6u1NuI1j371hgii3ouRfnfXI2Yb55D2zM,327
+kattis_cli/kattis_templates/main/python3.txt,sha256=bYcDVlq0cGhjVF1wXHlrN42cQkMD47qmmogzMfCpXtY,331
+kattis_cli/kattis_templates/project_structure/c/Makefile,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/c/src/problemid/problemid.c,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/c/tests/test_problemid.c,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/cpp/Makefile,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/cpp/src/problemid/problemid.cpp,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/cpp/tests/test_problemid.cpp,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/python3/Makefile,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/python3/src/problemid/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/python3/src/problemid/problemid.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/kattis_templates/project_structure/python3/tests/test_problemid.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/main.py,sha256=d8-2R4r1_kHb52nfpydS_5dWgf32MVUiJwVAjljk8TA,4996
+kattis_cli/settings.py,sha256=d5q4dYj9VqDSqPalleh2oZWtND-1bPB0T2IwdajFrBg,591
+kattis_cli/solution_tester.py,sha256=SQUq1Da06-_b9tTi0-CiTogscWje1Ur7KuaiK1Bxsgs,7860
+kattis_cli/template.py,sha256=QAbaQt9H9lPh8aOUMdl0B_vf2tZKL3mh9qdw7G-kw3g,5274
+kattis_cli/ui.py,sha256=2qFWz9QB7cgjLfZKHrWUrKFVYLRLwl0c37ZTNljwmAI,4580
+kattis_cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kattis_cli/utils/config.py,sha256=sR6lbRl9KiXRKIV1zRP5s6KJ9jed0znilsHSBS-_G2k,2278
+kattis_cli/utils/languages.py,sha256=GQjos4VKkyEBySPaOK1SFsWTqIIWdld0O_VHG-YEgi4,8977
+kattis_cli/utils/run_program.py,sha256=NWQ6vtTeWgkaW75r91FIHGXR5cAbeu8yMb5hwzpYFsg,2613
+kattis_cli/utils/utility.py,sha256=oUbydPbcC2wXvfku21YAWpaJjuBKaBRVybMAWCSfM9Y,2987
+kattis_cli-1.2.0.dist-info/METADATA,sha256=EuWgm_1VbuCG0OCUR2qzd9xbBEuGNcWBOrmEGwCqHtI,7206
+kattis_cli-1.2.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+kattis_cli-1.2.0.dist-info/entry_points.txt,sha256=kyzGN20VqUPR_H0J_jJUKT-10-cAMFLVegQ6C7tbHss,47
+kattis_cli-1.2.0.dist-info/licenses/LICENSE,sha256=JmBa4SEKBCDWEgiOZcISU4tUCpli6xSpVlSYgkBXSNQ,1067
+kattis_cli-1.2.0.dist-info/RECORD,,

kattis_cli-1.1.2.dist-info/RECORD

@@ -1,21 +0,0 @@
-kattis_cli/.kattis-cli.toml,sha256=kGh0gmpFAnivyUC9hR8Y1ZqYJeTBWNyBQ9zOg4AInZ0,898
-kattis_cli/__init__.py,sha256=afN92pog2fGyicY6KNBofYbCBYbYj4Fpi_INUSpsc-E,402
-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=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=gcJJJwkPS1Zpg76atwoo2O2Vd7ETKBQV-KD_cwclrmE,8928
-kattis_cli/utils/run_program.py,sha256=NWQ6vtTeWgkaW75r91FIHGXR5cAbeu8yMb5hwzpYFsg,2613
-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,,