rbx.cp 0.5.0__py3-none-any.whl

rbx/resources/presets/default/problem/statement/olymp.sty

@@ -0,0 +1,250 @@
+% Copyright (c) 2005-2013 Olymp.sty Developers Group.
+% See https://github.com/GassaFM/olymp.sty/blob/master/AUTHORS
+% for the complete list of authors.
+
+%\ProvidesPackage{olymp}
+\usepackage{verbatim}
+\usepackage{xstring}
+
+% \makeatletter
+% \newcommand\FirstWord[1]{\@firstword#1 \@nil}%
+% \def\@firstword#1 #2\@nil{\@removecomma#1,\@nil}%
+% \def\@removecomma#1,#2\@nil{\@removeperiod#1.\@nil}
+% \def\@removeperiod#1.#2\@nil{\@removesemicolon#1;\@nil}
+% \def\@removesemicolon#1;#2\@nil{#1}
+% \makeatother
+\newcommand\FirstWord[1]{#1}
+
+\ProcessOptions\relax
+
+\setlength{\parindent}{0pt}
+\setlength{\parskip}{0.8em}
+
+% contest keywords definitions
+\def\conProblemTutorial{Problem Tutorial}
+\def\conProblem{Problem}
+\def\conProblemAuthor{Author:}
+\def\conProblemDeveloper{Developer:}
+\def\conProblemOrigin{Origin:}
+\def\conInputFileName{Input file:}
+\def\conOutputFileName{Output file:}
+\def\conTimeLimit{Time limit:}
+\def\conMemoryLimit{Memory limit:}
+\def\conFeedback{Feedback:}
+\def\constdin{standard input}
+\def\constdout{standard output}
+\def\conSpecification{Specification}
+\def\conInteraction{Interaction Protocol}
+\def\conInput{Input}
+\def\conOutput{Output}
+\def\conExample{Example}
+\def\conExamples{Examples}
+\def\conExampleNotes{Notes}
+\def\conExplanation{Explanation}
+\def\conExplanations{Explanations}
+\def\conIllustration{Illustration}
+\def\conScoring{Scoring}
+\def\conNote{Note}
+\def\conNotes{Notes}
+\def\conConstraints{Constraints}
+\def\conversion{version}
+\def\conrevision{revision}
+\def\conSubtaskOne{Subtask 1}
+\def\conSubtaskTwo{Subtask 2}
+\def\conSubtaskThree{Subtask 3}
+\def\conSubtaskFour{Subtask 4}
+\def\conSubtaskFive{Subtask 5}
+\def\conSubtaskSix{Subtask 6}
+\def\conSubtask{Subtask}
+\def\conpoints{points}
+\def\conPage{Page}
+\def\conof{of}
+\def\connotstated{not stated}
+\def\conIntentionallyBlankPage{This page is intentionally left blank}
+\def\conTimeSuffix{s}
+\def\conMemorySufix{mb}
+
+% fancyhdr
+
+% font
+
+\newcommand{\afterfrontpage}{
+	\vspace*{\fill}
+	\begin{center}
+		\problemheadfont\conIntentionallyBlankPage
+	\end{center}
+	\vspace*{\fill}
+	\clearpage
+}
+\newcommand{\problemheadfont}{\LARGE}
+\newcommand{\problemsectionfont}{\Large}
+\newcommand{\problemtextfont}{\normalsize}
+\newcommand{\beforeproblemsectioncaption}{\smallbreak\smallskip}
+\newcommand{\afterproblemsectioncaption}{\smallskip}
+\newcommand{\problemend}{
+	\clearpage
+	\ifintentionallyblankpages
+		\ifodd\value{page}
+		\else
+			\afterfrontpage
+		\fi
+	\fi
+}
+\def\presectionspacing{\hspace{3mm}}
+
+% section commands from olymp
+
+\newcommand{\createsection}{\@newsection}
+
+\def\@newsection#1#2{\DeclareRobustCommand{#1}{
+		{\beforeproblemsectioncaption\bf\problemsectionfont\presectionspacing{#2}}
+		\nopagebreak\par\afterproblemsectioncaption}
+}
+
+\newcommand{\createsectionpar}{\@newsectionpar}
+
+\def\@newsectionpar#1#2{\DeclareRobustCommand{#1}[1]{
+		{\beforeproblemsectioncaption\bf\problemsectionfont{#2~##1}}
+		\nopagebreak\par\afterproblemsectioncaption}
+}
+
+\newcommand{\createsectionpartwo}{\@newsectionpartwo}
+
+\def\@newsectionpartwo#1#2#3{\DeclareRobustCommand{#1}[2]{
+		{\beforeproblemsectioncaption\problemsectionfont{\textbf{#2}~\textbf{##1}~(##2~#3)}}
+		\nopagebreak\par\afterproblemsectioncaption}
+}
+
+\createsection{\Instructions}{\conInstructions}
+\createsection{\Specification}{\conSpecification}
+\createsection{\Interaction}{\conInteraction}
+\createsection{\InputFile}{\conInput}
+\createsection{\OutputFile}{\conOutput}
+\createsection{\Example}{\conExample}
+\createsection{\Examples}{\conExamples}
+\createsection{\Explanation}{\conExplanation}
+\createsection{\Explanations}{\conExplanations}
+\createsection{\Illustration}{\conIllustration}
+\createsection{\Scoring}{\conScoring}
+\createsection{\Note}{\conNote}
+\createsection{\Notes}{\conNotes}
+\createsection{\Constraints}{\conConstraints}
+\createsection{\SubtaskOne}{\conSubtaskOne}
+\createsection{\SubtaskTwo}{\conSubtaskTwo}
+\createsection{\SubtaskThree}{\conSubtaskThree}
+\createsection{\SubtaskFour}{\conSubtaskFour}
+\createsection{\SubtaskFive}{\conSubtaskFive}
+\createsection{\SubtaskSix}{\conSubtaskSix}
+\createsectionpar{\Subtask}{\conSubtask}
+\createsectionpartwo{\SubtaskWithCost}{\conSubtask}{\conpoints}
+\createsection{\Editorial}{\conEditorial}
+
+% frontpage
+\newenvironment{frontpage}[3]{
+	\thispagestyle{empty}
+	{\centering
+		\noindent\rule{18cm}{0.4pt}
+
+		\vspace{1cm}
+		%\includegraphics[width=0.15\textwidth]{example-image-1x1}\par\vspace{1cm}
+		{\LARGE #1\par}
+		%\vspace{1cm}
+		{\Large #2\par}
+		\vspace{1cm}}
+	\def\thisFrontPageDate{#3}
+	\Instructions
+}{
+	\vspace*{\fill}
+	\centering{\Large \thisFrontPageDate\par}
+	\centering\noindent\rule{18cm}{0.4pt}
+	\newpage
+}
+
+% problem
+\newcounter{ProblemCounter}
+
+\newenvironment{problem}[6][]{
+
+	\def\ProblemLetter{#1}
+	\stepcounter{ProblemCounter}
+	\begin{center}
+		\ifx\ProblemLetter\empty
+		\else
+			\problemtextfont\textbf{\conProblem\ \ProblemLetter} \\
+		\fi
+		\problemheadfont\textsf{#2} \\
+		\ifx#5\empty
+		\else
+			\problemtextfont{\conTimeLimit\ #5} \\
+		\fi
+		\ifx#6\empty
+		\else
+			\problemtextfont{\conMemoryLimit\ #6} \\
+		\fi
+		%\problemtextfont{\conSourceFileName\ \ProblemLetter\ \conSourceExts} \\
+		\vspace{0.3cm}
+		\noindent\rule{16cm}{0.4pt}
+	\end{center}
+	\problemtextfont
+	\newcommand{\InputFileName}{#3}
+	\newcommand{\OutputFileName}{#4}
+}{\problemend}
+
+% examples
+% -- Setup sizes --
+\newlength{\thelinewidth}
+\newlength{\exmpwidinf}
+\newlength{\exmpwidouf}
+\newlength{\exmpwidewid}
+\newlength{\exmpthreewidinf}
+\newlength{\exmpthreewidouf}
+\newlength{\exmpthreewidnote}
+
+\thelinewidth=\textwidth
+\exmpwidinf=0.43\thelinewidth
+\exmpwidouf=0.43\thelinewidth
+\exmpwidewid=0.9\thelinewidth
+\exmpthreewidinf=0.28\thelinewidth
+\exmpthreewidouf=0.28\thelinewidth
+\exmpthreewidnote=0.30\thelinewidth
+
+\def\s@tm@cr@s{
+	\def\widthin##1{\exmpwidinf=##1\relax}
+	\def\widthout##1{\exmpwidouf=##1\relax}
+	\def\stretchin##1{\advance\exmpwidinf by ##1\relax}
+	\def\stretchout##1{\advance\exmpwidouf by ##1\relax}
+	\@ifstar{
+		\error Star must not be used in example environment any more
+	}
+}
+
+% This is magic, which delete space after verbatiminput
+\addto@hook{\every@verbatim}{\topsep=0pt\relax}
+
+\newenvironment{example}[1][]{
+	\s@tm@cr@s#1
+	\ttfamily\obeyspaces\obeylines\frenchspacing
+	\newcommand{\exmp}[2]{
+		\begin{minipage}[t]{\exmpwidinf}\rightskip=0pt plus 1fill\relax##1\medskip\end{minipage}&
+		\begin{minipage}[t]{\exmpwidouf}\rightskip=0pt plus 1fill\relax##2\medskip\end{minipage}\\
+		\hline
+	}
+
+	\newcommand{\exmpfile}[2]{
+		\exmp{
+			\verbatiminput{##1}
+		}{
+			\verbatiminput{##2}
+		}%
+	}
+
+	\begin{center}
+		\begin{tabular}{|l|l|}
+			\hline
+			\multicolumn{1}{|c|}{\bf\texttt{\textbf{\InputFileName}}} &
+			\multicolumn{1}{|c|}{\bf\texttt{\textbf{\OutputFileName}}}  \\
+			\hline
+			}{
+		\end{tabular}
+	\end{center}
+}

rbx/resources/presets/default/problem/statement/statement.rbx.tex

@@ -0,0 +1,18 @@
+%- block legend
+Dado dois números inteiros $A$ e $B$, determine o valor de $A + B$.
+
+\includegraphics[width=6cm]{projecao.png}
+%- endblock
+
+%- block input
+A entrada é composta por uma única linha contendo dois números
+inteiros $A$ e $B$ ($1 \leq A, B \leq \VAR{vars.MAX_N | sci}$).
+%- endblock
+
+%- block output
+A saída deve conter somente um único inteiro, a soma de $A$ e $B$.
+%- endblock
+
+%- block notes
+Sem muitas notas pra este problema.
+%- endblock

rbx/resources/presets/default/problem/statement/template.rbx.tex

@@ -0,0 +1,89 @@
+
+\documentclass[titlepage, oneside, a4paper]{article}
+
+\usepackage {import}
+\usepackage[margin=25mm, left=17mm, right=17mm]{geometry}
+\usepackage[utf8]{inputenc}
+\usepackage[english]{babel}
+\usepackage{amsmath}
+\usepackage{amssymb}
+\usepackage{fancyhdr}
+\usepackage{comment}
+\usepackage{olymp}
+\usepackage{epigraph}
+\usepackage{expdlist}
+\usepackage{graphicx}
+\usepackage{ulem}
+\usepackage{lastpage}
+
+%- if problem.blocks.preamble is defined
+\VAR{problem.blocks.preamble}
+%- endif
+
+\def\showSourceFileName{1}
+\newif\ifintentionallyblankpages % comment this out to add blank pages
+
+\pagestyle{fancy}
+\fancyhf{}
+\renewcommand{\footrulewidth}{0.4pt}
+
+\title{}
+\author{}
+\date{}
+
+\makeatletter
+\let\newtitle\@title
+\let\newauthor\@author
+\let\newdate\@date
+\makeatother
+
+\rhead{\newtitle}
+\lhead{\newauthor}
+\lfoot{\newdate}
+\rfoot{\thepage}
+
+\begin{document}
+
+\begin{problem}
+%- if problem.short_name is defined
+[\VAR{problem.short_name}]
+%- endif
+{\VAR{problem.title | escape}}
+{stdin}
+{stdout}
+{\VAR{problem.package.timeLimit / 1000 | round(1, 'floor')} s}
+{\VAR{problem.package.memoryLimit} MB}
+
+\VAR{problem.blocks.legend}
+
+%- if problem.blocks.input is defined
+\InputFile
+\VAR{problem.blocks.input}
+%- endif
+
+%- if problem.blocks.output is defined
+\OutputFile
+\VAR{problem.blocks.output}
+%- endif
+
+\Examples
+
+%- for sample in problem.samples
+\begin{example}
+	\exmpfile{\VAR{sample.inputPath}}{\VAR{sample.outputPath if sample.outputPath is not none else ''}}%
+\end{example}
+%- endfor
+
+%- if problem.blocks.notes is defined
+\Notes
+\VAR{problem.blocks.notes}
+%- endif
+
+%- if problem.blocks.editorial is defined
+\Editorial
+\VAR{problem.blocks.editorial}
+%- endif
+
+\end{problem}
+
+\end{document}

rbx/resources/presets/default/problem/tests/samples/000.in

@@ -0,0 +1 @@
+3 7

rbx/resources/presets/default/problem/tests/samples/001.in

@@ -0,0 +1 @@
+123456789 123456789

rbx/resources/presets/default/problem/validator.cpp

@@ -0,0 +1,16 @@
+#include "testlib.h"
+
+using namespace std;
+
+int main(int argc, char *argv[]) {
+    registerValidation(argc, argv);
+    prepareOpts(argc, argv);
+
+    int MAX_N = opt<int>("MAX_N"); // Read from package vars.
+
+    inf.readInt(1, MAX_N, "A");
+    inf.readSpace();
+    inf.readInt(1, MAX_N, "B");
+    inf.readEoln();
+    inf.readEof();
+}

rbx/resources/presets/default/problem/wcmp.cpp

@@ -0,0 +1,34 @@
+#include "testlib.h"
+
+using namespace std;
+
+int main(int argc, char *argv[]) {
+    setName("compare sequences of tokens");
+    registerTestlibCmd(argc, argv);
+
+    int n = 0;
+    string j, p;
+
+    while (!ans.seekEof() && !ouf.seekEof()) {
+        n++;
+
+        ans.readWordTo(j);
+        ouf.readWordTo(p);
+
+        if (j != p)
+            quitf(_wa, "%d%s words differ - expected: '%s', found: '%s'", n, englishEnding(n).c_str(),
+                  compress(j).c_str(), compress(p).c_str());
+    }
+
+    if (ans.seekEof() && ouf.seekEof()) {
+        if (n == 1)
+            quitf(_ok, "\"%s\"", compress(j).c_str());
+        else
+            quitf(_ok, "%d tokens", n);
+    } else {
+        if (ans.seekEof())
+            quitf(_wa, "Participant output contains extra tokens");
+        else
+            quitf(_wa, "Unexpected EOF in the participants output");
+    }
+}

rbx/resources/templates/template.cpp

@@ -0,0 +1,19 @@
+#include <bits/stdc++.h>
+
+using namespace std;
+{% if problem_test_type == 'multiNumber' %}
+void solve(int tn) {
+
+}
+{% endif %}
+int32_t main()
+{
+  ios::sync_with_stdio(false);
+  cin.tie(0);
+  {% if problem_test_type == 'multiNumber' %}
+  int T;
+  cin >> T;
+  for (int i = 1; i <= T; i++)
+    solve(i);
+  {% endif %}
+}

rbx/run.py

@@ -0,0 +1,45 @@
+import atexit
+import os
+import shlex
+
+from rbx import annotations, grading_utils, metadata
+from rbx.config import get_config
+from rbx.console import stderr_console
+from rbx.grading import steps
+from rbx.grading.judge.sandboxes import stupid_sandbox
+
+
+def main(
+    problem: annotations.Problem,
+    language: annotations.LanguageWithDefault = None,
+    keep_sandbox: bool = False,
+):
+    dumped_problem = metadata.find_problem_by_anything(problem)
+    if not dumped_problem:
+        stderr_console.print(
+            f'[error]Problem with identifier [item]{problem}[/item] not found.[/error]'
+        )
+        return
+
+    lang = get_config().get_language(language)
+    if not lang:
+        stderr_console.print(
+            f'[error]Language {language or get_config().defaultLanguage} not found in config. Please check your configuration.[/error]'
+        )
+        return
+
+    box = stupid_sandbox.StupidSandbox()
+    atexit.register(lambda: box.cleanup(delete=not keep_sandbox))
+
+    preprocess_cmds = grading_utils.build_preprocess_commands(dumped_problem, lang)
+    sandbox_params = grading_utils.build_preprocess_sandbox_params()
+    artifacts = grading_utils.build_compile_grading_artifacts(dumped_problem, lang)
+
+    if not steps.compile(preprocess_cmds, sandbox_params, box, artifacts):
+        stderr_console.print(
+            f'[error]Failed to preprocess problem [item]{dumped_problem.pretty_name()}[/item].[/error]'
+        )
+        return
+
+    cmd = shlex.split(lang.exec)
+    os.execv(cmd[0], cmd)

rbx/schema.py

@@ -0,0 +1,64 @@
+import uuid
+from typing import Dict, List, Optional
+
+from pydantic import BaseModel
+
+from rbx import utils
+
+
+class Testcase(BaseModel):
+    input: str
+    output: str
+
+
+class Batch(BaseModel):
+    id: str
+    size: int
+
+    @staticmethod
+    def create():
+        return Batch(id=str(uuid.uuid4()), size=1)
+
+
+class Problem(BaseModel):
+    name: str
+    group: str = ''
+    url: str = ''
+    interactive: bool = False
+    memoryLimit: int
+    timeLimit: int
+    tests: List[Testcase] = []
+    testType: str = 'single'
+    batch: Batch
+
+    def get_code(self):
+        return self.get_normalized_name()
+
+    def get_normalized_name(self) -> str:
+        return utils.normalize_with_underscores(self.name)
+
+
+class DumpedProblem(Problem):
+    code: str
+    aliases: List[str]
+    checker: Optional[str] = None
+
+    @staticmethod
+    def from_problem(problem: Problem, **kwargs) -> 'DumpedProblem':
+        return DumpedProblem(**problem.model_dump(), **kwargs)
+
+    def pretty_name(self) -> str:
+        if self.name == self.code:
+            return self.name
+        return f'{self.name} ({self.code})'
+
+    def get_vars(self) -> Dict[str, str]:
+        return {
+            'problem_name': self.name,
+            'problem_code': self.code,
+            'problem_url': self.url,
+            'problem_contest': self.group,
+            'problem_time_limit': f'{self.timeLimit}ms',
+            'problem_memory_limit': f'{self.memoryLimit}MB',
+            'problem_test_type': self.testType,
+        }

rbx/submit.py

@@ -0,0 +1,61 @@
+import atexit
+from pathlib import PosixPath
+from typing import Optional
+
+from rbx import annotations, grading_utils, metadata, submitors, utils
+from rbx.config import get_config
+from rbx.console import console
+from rbx.grading import steps
+from rbx.grading.judge.sandboxes import stupid_sandbox
+
+
+def main(
+    problem: annotations.Problem,
+    language: Optional[annotations.LanguageWithDefault] = None,
+    keep_sandbox: bool = False,
+):
+    dumped_problem = metadata.find_problem_by_anything(problem)
+    if not dumped_problem:
+        console.print(
+            f'[error]Problem with identifier [item]{problem}[/item] not found.[/error]'
+        )
+        return
+
+    lang = get_config().get_language(language)
+    if not lang:
+        console.print(
+            f'[error]Language {language or get_config().defaultLanguage} not found in config. Please check your configuration.[/error]'
+        )
+        return
+
+    box = stupid_sandbox.StupidSandbox()
+    atexit.register(lambda: box.cleanup(delete=not keep_sandbox))
+
+    with console.status(
+        f'Preprocessing problem [item]{dumped_problem.pretty_name()}[/item]...'
+    ):
+        preprocess_cmds = grading_utils.build_preprocess_commands(dumped_problem, lang)
+        sandbox_params = grading_utils.build_preprocess_sandbox_params()
+        artifacts = grading_utils.build_compile_grading_artifacts(dumped_problem, lang)
+        if not steps.compile(preprocess_cmds, sandbox_params, box, artifacts):
+            console.print(
+                f'[error]Failed to preprocess problem [item]{dumped_problem.pretty_name()}[/item].[/error]'
+            )
+            return
+
+    submit_file = PosixPath(lang.get_submit_file(dumped_problem.code))
+    console.print(
+        f'Problem to be submitted: [item]{dumped_problem.pretty_name()}[/item]'
+    )
+    console.print(f'Submission file: {submit_file.absolute()}')
+
+    if not utils.confirm_on_status(
+        None, 'Do you want to submit this problem?', default=False
+    ):
+        console.print('Skipping submission.')
+        return
+
+    if submitors.handle_submit(submit_file, dumped_problem, lang):
+        console.print('[green]Submission successful.[/green]')
+    else:
+        console.print('[error]Submission failed.[/error]')

rbx/submitors/__init__.py

@@ -0,0 +1,18 @@
+import pathlib
+
+from rbx.config import Language, get_config
+from rbx.schema import Problem
+from rbx.submitors.codeforces import CodeforcesSubmitor
+
+_SUBMITORS = [CodeforcesSubmitor()]
+
+
+def handle_submit(file: pathlib.Path, problem: Problem, lang: Language) -> bool:
+    for submitor in _SUBMITORS:
+        if submitor.should_handle(problem):
+            submitor_config = get_config().submitor[lang.submitor]
+            credentials = get_config().credentials[submitor.key()]
+            return submitor.submit(
+                file, problem, submitor_config[submitor.key()], credentials
+            )
+    return False