kattis-cli 1.1.2__tar.gz → 1.2.0__tar.gz

{kattis_cli-1.1.2 → kattis_cli-1.2.0}/PKG-INFO

@@ -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.1.2 → kattis_cli-1.2.0}/README.md

@@ -110,6 +110,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.1.2 → kattis_cli-1.2.0}/pyproject.toml

@@ -1,31 +1,39 @@
-[tool.poetry]
+
+# PEP 621 metadata table for tools that expect [project].
+[project]
 name = "kattis-cli"
-version = "1.1.2"
-authors = ["Ram Basnet <rbasnet@coloradomesa.edu>"]
+version = "1.2.0"
 description = "A command-line tool for Kattis"
 readme = "README.md"
-repository = "https://github.com/rambasnet/kattis-cli"
+authors = [
+    { name = "Ram Basnet", email = "rbasnet@coloradomesa.edu" }
+]
+
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
 ]
 
-# PEP 621 metadata table for tools that expect [project].
-[project]
-name = "kattis-cli"
-version = "1.1.2"
-description = "A command-line tool for Kattis"
-readme = "README.md"
-authors = [
-    { name = "Ram Basnet", email = "rbasnet@coloradomesa.edu" }
+[tool.setuptools]
+include-package-data = true
+
+[tool.setuptools.package-data]
+# Include the kattis_templates package data inside the installed package
+# (files under src/kattis_cli/kattis_templates)
+kattis_cli = [
+    "kattis_templates/*",
+    "kattis_templates/**",
 ]
-repository = "https://github.com/rambasnet/kattis-cli"
 
+[tool.poetry]
 [project.urls]
 "Homepage" = "https://github.com/rambasnet/kattis-cli"
 "Bug Tracker" = "https://github.com/rambasnet/kattis-cli/issues"
 
+[tool.setuptools.packages.find]
+where = ["src"]
+
 [tool.poetry.dependencies]
 python = "^3.10.0"
 requests = "^2.31.0"
@@ -39,4 +47,3 @@ trogon = "^0.6.0"
 
 [tool.poetry.scripts]
 kattis = 'kattis_cli.main:main'
-

kattis_cli-1.2.0/src/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-1.2.0/src/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-1.2.0/src/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-1.2.0/src/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-1.2.0/src/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-1.1.2 → kattis_cli-1.2.0/src}/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-1.1.2 → kattis_cli-1.2.0/src}/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-1.2.0/src/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-1.1.2 → kattis_cli-1.2.0/src}/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-1.1.2 → kattis_cli-1.2.0/src}/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()