rbx.cp 0.5.0__py3-none-any.whl

rbx/config.py

@@ -0,0 +1,271 @@
+import functools
+import importlib
+import importlib.resources
+import os
+import pathlib
+import shutil
+import subprocess
+from typing import Any, Dict, List, Optional
+
+import requests
+import typer
+from pydantic import BaseModel
+
+from rbx import utils
+from rbx.console import console
+from rbx.grading.judge.storage import copyfileobj
+
+app = typer.Typer(no_args_is_help=True)
+
+_RESOURCES_PKG = 'rbx.resources'
+_CONFIG_FILE_NAME = 'default_config.json'
+
+
+def format_vars(template: str, **kwargs) -> str:
+    res = template
+    for key, value in kwargs.items():
+        key = key.replace('_', '-')
+        res = res.replace(f'%{{{key}}}', value)
+    return res
+
+
+class Artifact(BaseModel):
+    filename: Optional[str] = None
+    executable: bool = False
+    optional: bool = False
+
+
+class Language(BaseModel):
+    template: str
+    file: str
+    submitFile: Optional[str] = None
+    preprocess: Optional[List[str]] = None
+    exec: str
+    artifacts: Dict[str, Optional[Artifact]] = {}
+    submitor: Optional[str] = None
+
+    def get_file(self, basename: str) -> str:
+        return format_vars(self.file, problem_code=basename)
+
+    def has_submit_file(self) -> bool:
+        return self.submitFile is not None
+
+    def get_submit_file(self, basename: str) -> str:
+        if not self.submitFile:
+            return self.get_file(basename)
+        return format_vars(
+            self.submitFile, file=self.get_file(basename), problem_code=basename
+        )
+
+    def get_template(self) -> str:
+        return get_app_file(pathlib.Path('templates') / self.template).read_text()
+
+
+SubmitorConfig = Dict[str, Any]
+Credentials = Dict[str, Any]
+
+
+class Config(BaseModel):
+    defaultLanguage: str
+    languages: Dict[str, Language]
+    editor: Optional[str] = None
+    submitor: Dict[str, SubmitorConfig]
+    credentials: Credentials
+    boxEnvironment: str = 'default'
+
+    def get_default_language(self) -> Optional[Language]:
+        return self.languages.get(self.defaultLanguage)
+
+    def get_language(self, name: Optional[str] = None) -> Optional[Language]:
+        return self.languages.get(name or self.defaultLanguage)
+
+
+def get_app_path() -> pathlib.Path:
+    return utils.get_app_path()
+
+
+def get_empty_app_persist_path() -> pathlib.Path:
+    app_dir = get_app_path() / 'persist'
+    shutil.rmtree(str(app_dir), ignore_errors=True)
+    app_dir.mkdir(parents=True, exist_ok=True)
+    return app_dir
+
+
+def get_app_file(path: pathlib.Path) -> pathlib.Path:
+    file_path = get_app_path() / path
+    if file_path.is_file():
+        return file_path
+
+    with importlib.resources.as_file(
+        importlib.resources.files('rbx') / 'resources' / path  # type: ignore
+    ) as file:
+        if file.is_file():
+            file_path.parent.mkdir(parents=True, exist_ok=True)
+            with file.open('rb') as fr:
+                with file_path.open('wb') as fw:
+                    copyfileobj(fr, fw)
+    return file_path
+
+
+def _download_checker(name: str, save_at: pathlib.Path):
+    console.print(f'Downloading checker {name}...')
+    r = requests.get(
+        f'https://raw.githubusercontent.com/MikeMirzayanov/testlib/master/checkers/{name}'
+    )
+
+    if r.ok:
+        save_at.parent.mkdir(parents=True, exist_ok=True)
+        with save_at.open('wb') as f:
+            f.write(r.content)
+
+
+def _download_testlib(save_at: pathlib.Path):
+    console.print('Downloading testlib.h...')
+    r = requests.get(
+        'https://raw.githubusercontent.com/MikeMirzayanov/testlib/master/testlib.h'
+    )
+
+    if r.ok:
+        save_at.parent.mkdir(parents=True, exist_ok=True)
+        with save_at.open('wb') as f:
+            f.write(r.content)
+    else:
+        console.print('[error]Failed to download testlib.h.[/error]')
+        raise typer.Exit(1)
+
+
+def _download_jngen(save_at: pathlib.Path):
+    console.print('Downloading jngen.h...')
+    r = requests.get('https://raw.githubusercontent.com/ifsmirnov/jngen/master/jngen.h')
+
+    if r.ok:
+        save_at.parent.mkdir(parents=True, exist_ok=True)
+        with save_at.open('wb') as f:
+            f.write(r.content)
+    else:
+        console.print('[error]Failed to download jngen.h.[/error]')
+        raise typer.Exit(1)
+
+
+def _download_bits_stdcpp(save_at: pathlib.Path):
+    console.print('Downloading bits/stdc++.h...')
+    r = requests.get(
+        'https://raw.githubusercontent.com/tekfyl/bits-stdc-.h-for-mac/master/stdc%2B%2B.h'
+    )
+
+    if r.ok:
+        save_at.parent.mkdir(parents=True, exist_ok=True)
+        with save_at.open('wb') as f:
+            f.write(r.content)
+    else:
+        console.print('[error]Failed to download bits/stdc++.h.[/error]')
+        raise typer.Exit(1)
+
+
+def get_builtin_checker(name: str) -> pathlib.Path:
+    app_file = get_app_file(pathlib.Path('checkers') / name)
+    if not app_file.exists():
+        _download_checker(name, app_file)
+    return app_file
+
+
+def get_testlib() -> pathlib.Path:
+    app_file = get_app_file(pathlib.Path('testlib.h'))
+    if not app_file.exists():
+        _download_testlib(app_file)
+    return app_file
+
+
+def get_jngen() -> pathlib.Path:
+    app_file = get_app_file(pathlib.Path('jngen.h'))
+    if not app_file.exists():
+        _download_jngen(app_file)
+    return app_file
+
+
+def get_bits_stdcpp() -> pathlib.Path:
+    app_file = get_app_file(pathlib.Path('stdc++.h'))
+    if not app_file.exists():
+        _download_bits_stdcpp(app_file)
+    return app_file
+
+
+def get_default_config_path() -> pathlib.Path:
+    with importlib.resources.as_file(
+        importlib.resources.files('rbx') / 'resources' / _CONFIG_FILE_NAME
+    ) as file:
+        return file
+
+
+def get_default_app_path() -> pathlib.Path:
+    return get_default_config_path().parent
+
+
+def get_default_config() -> Config:
+    return Config.model_validate_json(get_default_config_path().read_text())
+
+
+def get_config_path() -> pathlib.Path:
+    return get_app_path() / 'config.json'
+
+
+def get_editor():
+    return get_config().editor or os.environ.get('EDITOR', None)
+
+
+def open_editor(path: pathlib.Path, *args):
+    editor = get_editor()
+    if editor is None:
+        raise Exception('No editor found. Please set the EDITOR environment variable.')
+    subprocess.run([editor, str(path), *[str(arg) for arg in args]])
+
+
+@functools.cache
+def get_config() -> Config:
+    config_path = get_config_path()
+    if not config_path.is_file():
+        utils.create_and_write(config_path, utils.model_json(get_default_config()))
+    return Config.model_validate_json(config_path.read_text())
+
+
+def save_config(cfg: Config):
+    cfg_path = get_config_path()
+    cfg_path.write_text(utils.model_json(cfg))
+    get_config.cache_clear()
+
+
+@app.command()
+def path():
+    """
+    Show the absolute path of the config file.
+    """
+    get_config()  # Ensure config is created.
+    console.print(get_config_path())
+
+
+@app.command('list, ls')
+def list():
+    """
+    Pretty print the config file.
+    """
+    console.print_json(utils.model_json(get_config()))
+
+
+@app.command()
+def reset():
+    """
+    Reset the config file to the default one.
+    """
+    if not typer.confirm('Do you really want to reset your config to the default one?'):
+        return
+    cfg_path = get_config_path()
+    cfg_path.unlink(missing_ok=True)
+    get_config()  # Reset the config.
+
+
+@app.command('edit, e')
+def edit():
+    """
+    Open the config in an editor.
+    """
+    open_editor(get_config_path())

rbx/conftest.py

@@ -0,0 +1,38 @@
+import os
+import pathlib
+import shutil
+import tempfile
+from collections.abc import Iterator
+
+import pytest
+
+from rbx.testing_utils import get_testdata_path
+
+
+@pytest.fixture
+def testdata_path() -> pathlib.Path:
+    return get_testdata_path()
+
+
+@pytest.fixture
+def cleandir() -> Iterator[pathlib.Path]:
+    with tempfile.TemporaryDirectory() as newpath:
+        abspath = pathlib.Path(newpath).absolute()
+        old_cwd = pathlib.Path.cwd()
+        os.chdir(newpath)
+        try:
+            yield abspath
+        finally:
+            os.chdir(str(old_cwd))
+
+
+@pytest.fixture
+def cleandir_with_testdata(
+    request, testdata_path: pathlib.Path, cleandir: pathlib.Path
+) -> Iterator[pathlib.Path]:
+    marker = request.node.get_closest_marker('test_pkg')
+    if marker is None:
+        raise ValueError('test_pkg marker not found')
+    testdata = testdata_path / marker.args[0]
+    shutil.copytree(str(testdata), str(cleandir), dirs_exist_ok=True)
+    yield cleandir

rbx/console.py

@@ -0,0 +1,27 @@
+import sys
+
+from rich.console import Console
+from rich.theme import Theme
+
+theme = Theme(
+    {
+        'default': 'bright_white',
+        'rbx': 'bold italic yellow',
+        'info': 'bright_black',
+        'status': 'bright_white',
+        'item': 'bold blue',
+        'error': 'bold red',
+        'success': 'bold green',
+        'lnumber': 'dim cyan',
+        'warning': 'bold yellow',
+    }
+)
+console = Console(theme=theme, style='info', highlight=False)
+stderr_console = Console(theme=theme, style='info', highlight=False, stderr=True)
+
+
+def multiline_prompt(text: str) -> str:
+    console.print(f'{text} (Ctrl-D to finish):\n')
+    lines = sys.stdin.readlines()
+    console.print()
+    return ''.join(lines)

rbx/create.py

@@ -0,0 +1,37 @@
+import pathlib
+
+from rbx import annotations
+from rbx.clone import create_problem_structure
+from rbx.config import get_config
+from rbx.console import console
+from rbx.schema import Batch, Problem
+
+
+def main(
+    name: str,
+    lang: annotations.Language,
+    timelimit: annotations.Timelimit = 1000,
+    memorylimit: annotations.Memorylimit = 256,
+    multitest: annotations.Multitest = False,
+):
+    language = get_config().get_language(lang)
+    if language is None:
+        console.print(
+            f'[error]Language {lang or get_config().defaultLanguage} not found in config. Please check your configuration.[/error]'
+        )
+        return
+
+    problem = Problem(
+        name=name,
+        timeLimit=timelimit,
+        memoryLimit=memorylimit,
+        testType='multiNumber' if multitest else 'single',
+        batch=Batch.create(),
+    )
+    create_problem_structure(
+        pathlib.Path(),
+        problem,
+        language,
+        status=None,
+        verbose=True,
+    )

rbx/edit.py

@@ -0,0 +1,24 @@
+import pathlib
+from typing import Optional
+
+from rbx import annotations, console, metadata
+from rbx.config import get_config, open_editor
+
+
+def main(problem: str, language: Optional[annotations.LanguageWithDefault] = None):
+    lang = get_config().get_language(language)
+    if lang is None:
+        console.console.print(
+            f'[error]Language {language or get_config().defaultLanguage} not found in config. Please check your configuration.[/error]'
+        )
+        return
+
+    dumped_problem = metadata.find_problem_by_anything(problem)
+    if not dumped_problem:
+        console.console.print(
+            f'[error]Problem with identifier {problem} not found.[/error]'
+        )
+        return
+
+    filename = lang.get_file(dumped_problem.code)
+    open_editor(pathlib.Path(filename))