rbx.cp 0.17.5__py3-none-any.whl → 0.17.7__py3-none-any.whl

rbx/__version__.py

@@ -1 +1 @@
-__version__ = '0.17.5'
+__version__ = '0.17.7'

rbx/box/sanitizers/warning_stack.py

@@ -29,9 +29,9 @@ class WarningStack:
         f = reference.get_file(cacher)
         if f is None:
             return
-        with dest_path.open('wb') as fout:
-            shutil.copyfileobj(f, fout)
-        f.close()
+        with f:
+            with dest_path.open('wb') as fout:
+                shutil.copyfileobj(f, fout)
         self.sanitizer_warnings[code.path] = dest_path
 
     def clear(self):

rbx/box/schema.py

@@ -106,6 +106,11 @@ class ExpectedOutcome(AutoEnum):
 
     Especially useful for environments where TLE and RTE are indistinguishable."""
 
+    JUDGE_FAILED = alias('judge failed', 'jf')  # type: ignore
+    """Expected outcome for solutions that finish with a judge failed verdict.
+    
+    Only useful for checker tests."""
+
     def style(self) -> str:
         if self == ExpectedOutcome.ANY:
             return 'orange'
@@ -161,6 +166,8 @@ class ExpectedOutcome(AutoEnum):
             return outcome in {Outcome.RUNTIME_ERROR} or outcome.is_slow()
         if self == ExpectedOutcome.OUTPUT_LIMIT_EXCEEDED:
             return outcome == Outcome.OUTPUT_LIMIT_EXCEEDED
+        if self == ExpectedOutcome.JUDGE_FAILED:
+            return outcome == Outcome.JUDGE_FAILED
         return False
 
     def get_matches(self) -> List[Outcome]:

rbx/grading/judge/program.py

@@ -20,8 +20,12 @@ FileLike = Union[PathOrStr, IO[bytes], int]
 def _maybe_close_files(files):
     for fobj in files:
         if isinstance(fobj, int):
-            continue
-        fobj.close()
+            try:
+                os.close(fobj)
+            except OSError:
+                pass
+        else:
+            fobj.close()
 
 
 def _is_pathlike(obj: Any) -> bool:
@@ -250,6 +254,20 @@ class Program:
         threading.Thread(target=self._handle_wall, daemon=True).start()
         threading.Thread(target=self._handle_alarm, daemon=True).start()
 
+    def close_pipes(self):
+        if self.popen is None:
+            return
+        if self.params.io.input == subprocess.PIPE and self.pipes.input is not None:
+            self.pipes.input.close()
+        if self.params.io.output == subprocess.PIPE and self.pipes.output is not None:
+            self.pipes.output.close()
+        if self.params.io.stderr == subprocess.PIPE and self.pipes.stderr is not None:
+            self.pipes.stderr.close()
+
+    def close(self):
+        self.close_pipes()
+        _maybe_close_files(self._files)
+
     def process_exit(self, exitstatus, ru) -> ProgramResult:
         _maybe_close_files(self._files)
 
@@ -307,4 +325,6 @@ class Program:
     def wait(self):
         assert self.popen is not None
         _, exitstatus, ru = os.wait4(self.pid, 0)
-        return self.process_exit(exitstatus, ru)
+        res = self.process_exit(exitstatus, ru)
+        self.close_pipes()
+        return res

rbx/grading/judge/sandboxes/stupid_sandbox.py

@@ -335,6 +335,12 @@ class StupidSandbox(SandboxBase):
             else:
                 raise RuntimeError(f'Unknown pid: {pid}')
 
+        for p in (interactor, program):
+            p.close()
+        if should_tee:
+            for p in (solution_tee, interactor_tee):
+                p.close()
+
         return typing.cast(Tuple[SandboxLog, SandboxLog], tuple(results))
 
     def cleanup(self, delete=False):

rbx/resources/presets/default/preset.rbx.yml

@@ -1,7 +1,7 @@
 ---
 # yaml-language-server: $schema=https://rsalesc.github.io/rbx/schemas/Preset.json
 name: "default"
-min_version: "0.17.5"
+min_version: "0.17.7"
 uri: "rsalesc/rbx/rbx/resources/presets/default"
 problem: "problem"
 contest: "contest"

rbx/utils.py

@@ -1,13 +1,11 @@
 import contextlib
 import enum
-import fcntl
 import functools
 import json
 import os
 import os.path
 import pathlib
 import re
-import resource
 import shutil
 import subprocess
 import sys
@@ -54,7 +52,7 @@ def get_upgrade_command(
     parsed_version = (
         semver.VersionInfo.parse(version) if isinstance(version, str) else version
     ) or get_semver()
-    return f'pipx install --upgrade {PIP_NAME}@{parsed_version.major}'
+    return f'uv tool install {PIP_NAME}@{parsed_version.major}'
 
 
 def check_version_compatibility_between(
@@ -75,19 +73,41 @@ def check_version_compatibility(required: str) -> SemVerCompatibility:
     return check_version_compatibility_between(installed, required)
 
 
-def print_open_fd_count() -> int:
+def print_open_fd_count(id: Optional[str] = None) -> int:
     import psutil
 
     try:
-        current_process = psutil.Process()
-        open_fds = len(current_process.open_files())
-        print(f'Number of opened file descriptors: {open_fds}')
+        open_fds = get_open_fds()
+        print(f'Number of opened file descriptors for {id or "..."}: {open_fds}')
     except psutil.AccessDenied:
         print('Access denied. Run with appropriate permissions (e.g., sudo) if needed.')
     except Exception as e:
         print(f'An error occurred: {e}')
 
 
+class FdLeakDetector:
+    def __init__(self, id: Optional[str] = None, diff: bool = False):
+        self.open_fds = 0
+        self.id = id
+        self.diff = diff
+
+    def __enter__(self):
+        self.open_fds = get_open_fds()
+        return self
+
+    def __exit__(self, *args, **kwargs):
+        open_fds = get_open_fds()
+        if open_fds > self.open_fds:
+            print(
+                f'File descriptor leak detected for {self.id or "..."}: {open_fds - self.open_fds} new file descriptors opened'
+            )
+        elif self.diff and open_fds < self.open_fds:
+            print(
+                f'File descriptor diff detected for {self.id or "..."}: {self.open_fds - open_fds} file descriptors closed'
+            )
+        self.open_fds = open_fds
+
+
 def create_and_write(path: pathlib.Path, *args, **kwargs):
     path.parent.mkdir(parents=True, exist_ok=True)
     path.write_text(*args, **kwargs)
@@ -256,16 +276,28 @@ def confirm_on_status(status: Optional[rich.status.Status], *args, **kwargs) ->
     return res
 
 
-def get_open_fds():
-    fds = []
-    soft, hard = resource.getrlimit(resource.RLIMIT_NOFILE)
-    for fd in range(0, soft):
-        try:
-            fcntl.fcntl(fd, fcntl.F_GETFD)
-        except IOError:
-            continue
-        fds.append(fd)
-    return fds
+def get_open_fds() -> int:
+    import psutil
+
+    try:
+        current_process = psutil.Process()
+        open_fds = current_process.num_fds()
+        return open_fds
+    except psutil.AccessDenied:
+        return 0
+    except Exception as e:
+        print(f'An error occurred: {e}')
+        return 0
+
+    # fds = []
+    # soft, hard = resource.getrlimit(resource.RLIMIT_NOFILE)
+    # for fd in range(0, soft):
+    #     try:
+    #         fcntl.fcntl(fd, fcntl.F_GETFD)
+    #     except IOError:
+    #         continue
+    #     fds.append(fd)
+    # return fds
 
 
 def command_exists(command):

{rbx_cp-0.17.5.dist-info → rbx_cp-0.17.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rbx.cp
-Version: 0.17.5
+Version: 0.17.7
 Summary: 
 Author: Roberto Sales
 Requires-Python: >=3.9.1,<4.0.0

{rbx_cp-0.17.5.dist-info → rbx_cp-0.17.7.dist-info}/RECORD

@@ -1,5 +1,5 @@
 rbx/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rbx/__version__.py,sha256=rZYsXJU98B0OEia0ZbSa3uP1L8XPNy2IvM2kzP3BE4s,23
+rbx/__version__.py,sha256=LhXO4wjASxNtbb9ULVbPs3Z2KD8fcjHapdbGc9c2q90,23
 rbx/annotations.py,sha256=_TkLhgZWiUyon3bonHwUo03ls1jY8LwgcR4bVgtgnc0,3519
 rbx/autoenum.py,sha256=cusv8ClXRlDVvhZ8eDrtYcL_2peXlHugAey_ht8roXk,12025
 rbx/box/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -60,8 +60,8 @@ rbx/box/presets/schema.py,sha256=5Nw72QrSsA39cUCCLlVfSYMImK_-TfJEGu5TSIDv3W8,291
 rbx/box/remote.py,sha256=hCdiZSmh6Y1vnSEwLZ_gtAMYV-WLVlU3iVnvcObRpRc,5274
 rbx/box/retries.py,sha256=BZsi4sYBjm3VK5zb_pBQSYQuKo3ZntmtEFoVPZHg4QI,4982
 rbx/box/sanitizers/issue_stack.py,sha256=nHtF8bksTIbvOlmDEy72SoXnXfQyIDBHU_3NOn-X1Ss,3534
-rbx/box/sanitizers/warning_stack.py,sha256=6-rr3dkMq6MpfjrVZ8lSQjF4RZ5YzZSAPMzHCfm-6h4,2876
-rbx/box/schema.py,sha256=3rkRkBiyZw0Dq3GMKX1zh5BE4-aSURFZogAQGoYqVck,20006
+rbx/box/sanitizers/warning_stack.py,sha256=3-5quni5NWMs2k2NI6MWVFR_ec8BeGFVZ29xyifZ_dY,2882
+rbx/box/schema.py,sha256=Q4rDXssTVsmf8kc_0tvZ1SnyXauYi-XubV2WiLBHIBY,20292
 rbx/box/setter_config.py,sha256=vyZbsbgkjelZOblMLX5A1j4CLqPNQkrv83IZr_3-46Y,5301
 rbx/box/solutions.py,sha256=Id5X9711p5VX0kZfVXpj_klADPz8k5GvQsV3oXvhgrU,52848
 rbx/box/state.py,sha256=MMf3DvfQji0jKEliCHct2Tpp_0epL1tvP8HbHNArQIc,166
@@ -132,10 +132,10 @@ rbx/grading/grading_context.py,sha256=TaRyLwPkkxvspQIFUFk8Ok0T8EST2pHMMNoVDx9lbF
 rbx/grading/judge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rbx/grading/judge/cacher.py,sha256=zPyOqd7hP_zjuAGAauckFs6EvQE8pP7Mrug7fylIZvY,21929
 rbx/grading/judge/digester.py,sha256=gtOIe_iL4PEWA7OKelW1gjSI-nBvbOpDPJGV8VQyjSg,912
-rbx/grading/judge/program.py,sha256=8hVme1IbyZlrgg8j-po3KZJsYqUbgqfpax-vSZbobRA,9993
+rbx/grading/judge/program.py,sha256=VJ-ItHGsumAvR7YWYtnmAFairfAIu_ASHfyhu7bqvRw,10671
 rbx/grading/judge/sandbox.py,sha256=e9Nstf9EoFGE0OBQgqcz5BD5hYDegIeGP5dWVZSA29Q,21342
 rbx/grading/judge/sandboxes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rbx/grading/judge/sandboxes/stupid_sandbox.py,sha256=QuIP93lBOm7bnkwB3NlhHWxRTs4DTosa_EYPqHq2QCY,11600
+rbx/grading/judge/sandboxes/stupid_sandbox.py,sha256=0Pw-Jvh2A7Jbm5mpx3r4mojP9HgFkDhSREWFuS3ixKs,11765
 rbx/grading/judge/sandboxes/tee.py,sha256=fLulB8fV7k5cO-ikRgURpsETrvr6LoUSjGxFM3GZs5U,672
 rbx/grading/judge/storage.py,sha256=8o3pz0Ansanc5t23XGtshJHCD_q4dIR_XorpeVEvg1g,15621
 rbx/grading/limits.py,sha256=zCyRTEtUQFd1RKL_fFKl3qSiRSIDCalBaSLlt1yhi0Y,896
@@ -201,7 +201,7 @@ rbx/resources/presets/default/contest/statement/info.rbx.tex,sha256=Wz8Tbmi2lPWM
 rbx/resources/presets/default/contest/statement/instructions.tex,sha256=JG_eR13ukZgEahrrmrbg40H8cUzpoUE8QLocihN-fZ8,2414
 rbx/resources/presets/default/contest/statement/logo.png,sha256=RLNYmZoc-BR6AZKkmr4UEg3h01YeFzvy604jMAQC7aA,414485
 rbx/resources/presets/default/env.rbx.yml,sha256=quSPG5Xs9KroYLATNLPNtORLGRWtrLLt2Fx81T1enAM,1692
-rbx/resources/presets/default/preset.rbx.yml,sha256=e5kETGeZhY5r9QkFB9Wgn3s2NmSOOKt2i_dEc5L9_vw,520
+rbx/resources/presets/default/preset.rbx.yml,sha256=wItvMn_dh0Gyd7o_fJd6MtaafVQ7Jwihk6moGYYrETA,520
 rbx/resources/presets/default/problem/.gitignore,sha256=1rt95y9Q7ZHIQn28JyZQUdD5zkpRosjAl9ZqoQmX2cE,149
 rbx/resources/presets/default/problem/gens/gen.cpp,sha256=rn6sGRjZ1sFE1Rq02r6488iquY9xTrutcvLv4d1sohA,178
 rbx/resources/presets/default/problem/manual_tests/samples/000.in,sha256=w66OEtCJGqjUNj8cJrqgImgGVm8W_OlIUtF255ds-ow,4
@@ -222,9 +222,9 @@ rbx/resources/presets/default/shared/problem_template.rbx.tex,sha256=KDibf4LtUio
 rbx/resources/templates/rbx.h,sha256=0AZds9R0PmuPgnlTENb33Y81LW0LlnmOJFaoN8oG3Yo,3638
 rbx/resources/templates/template.cpp,sha256=xXWpWo7fa7HfmPNqkmHcmv3i46Wm0ZL-gPmkRfGvLn4,317
 rbx/testing_utils.py,sha256=vNNdaytowJfuopszVHeFzVtHWlPfipPW4zpqCOvdZKU,2908
-rbx/utils.py,sha256=yBIp6vwY2hhY2s7bJtvlP5U5LEnZLfG8Y044VDKy6B4,10343
-rbx_cp-0.17.5.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-rbx_cp-0.17.5.dist-info/METADATA,sha256=YFq-MJGMCLVJH7JC3l11zpuxlb6Ky2DcSzqY-N8bfnU,4799
-rbx_cp-0.17.5.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-rbx_cp-0.17.5.dist-info/entry_points.txt,sha256=Gw2_BZ5Jon61biaH_ETbAQGXy8fR5On9gw2U4A1erpo,40
-rbx_cp-0.17.5.dist-info/RECORD,,
+rbx/utils.py,sha256=ua7qzt5rrpTKgEZXC6cKVDVklOReIYWL6sNsSWkrWE4,11355
+rbx_cp-0.17.7.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+rbx_cp-0.17.7.dist-info/METADATA,sha256=9RJ8D--x5pfsbzJ7dm7TaJdhAJYc2AXarRMHazUQuRw,4799
+rbx_cp-0.17.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+rbx_cp-0.17.7.dist-info/entry_points.txt,sha256=Gw2_BZ5Jon61biaH_ETbAQGXy8fR5On9gw2U4A1erpo,40
+rbx_cp-0.17.7.dist-info/RECORD,,

{rbx_cp-0.17.5.dist-info → rbx_cp-0.17.7.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any