rbx.cp 0.17.4__py3-none-any.whl → 0.17.6__py3-none-any.whl

rbx/__version__.py

@@ -1 +1 @@
-__version__ = '0.17.4'
+__version__ = '0.17.6'

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/tooling/boca/scraper.py

@@ -359,7 +359,8 @@ class BocaScraper:
         form['problemnumber'] = f'{problem_index + 1}'
         form['problemname'] = problem_shortname
         form['confirmation'] = 'confirm'
-        form['autojudge_new_sel'] = ['all']
+        if form.find_control('autojudge_new_sel'):
+            form['autojudge_new_sel'] = ['all']
         form['Submit3'] = 'Send'
 
         with file.open('rb') as f:

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,9 @@ class StupidSandbox(SandboxBase):
             else:
                 raise RuntimeError(f'Unknown pid: {pid}')
 
+        for p in (interactor, program, 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.4"
+min_version: "0.17.6"
 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
@@ -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.4.dist-info → rbx_cp-0.17.6.dist-info}/METADATA

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

{rbx_cp-0.17.4.dist-info → rbx_cp-0.17.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 rbx/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rbx/__version__.py,sha256=v9gIUnhG3PIYaQ8cxH8mz-h-4xf3QwSMMhvperd2jrk,23
+rbx/__version__.py,sha256=659MIXoktqvdpSoUwgo79H2UvpA7j13iYjBPkBKDdjQ,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,7 +60,7 @@ 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/sanitizers/warning_stack.py,sha256=3-5quni5NWMs2k2NI6MWVFR_ec8BeGFVZ29xyifZ_dY,2882
 rbx/box/schema.py,sha256=3rkRkBiyZw0Dq3GMKX1zh5BE4-aSURFZogAQGoYqVck,20006
 rbx/box/setter_config.py,sha256=vyZbsbgkjelZOblMLX5A1j4CLqPNQkrv83IZr_3-46Y,5301
 rbx/box/solutions.py,sha256=Id5X9711p5VX0kZfVXpj_klADPz8k5GvQsV3oXvhgrU,52848
@@ -94,7 +94,7 @@ rbx/box/tooling/boca/debug_utils.py,sha256=Rs1ptMWsy32jl_DJZASeOfyqh8HCJdhIXufdb
 rbx/box/tooling/boca/main.py,sha256=knl1rpaHIwA63KkzMJMZQrejzMpbTPBhYqGx1IpuNm4,289
 rbx/box/tooling/boca/manual_scrape.py,sha256=iuNtOHrWczdaWNNRMKGYkgr-wSwgtjZedpz5-sX5IRQ,724
 rbx/box/tooling/boca/scrape.py,sha256=q1BoVstfLbw5CwB9e2hQnseBv6HQMnXq2z_tk18ui6M,1024
-rbx/box/tooling/boca/scraper.py,sha256=6B3aI62YdzD7o1zykh9qavl6Qm6cTD--mEvVIjPvEaI,33701
+rbx/box/tooling/boca/scraper.py,sha256=ShckY5M-IyMY0o7aA5IWoFVSA5-N7Tadnx_-9J9s7UQ,33756
 rbx/box/tooling/converter.py,sha256=cTRUuZ7GToMuAUo6VE7tOtaeNkXHJZE29me0yopvvsY,2544
 rbx/box/tooling/main.py,sha256=uj6XHjJMF3nyb4enbbTK8XbUAuiRLT_RP1kVwSMqV5E,2003
 rbx/box/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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=aKJyIVhnxisOZRy3NMwQqSXTpC-WW8GJuEi06SKbmSk,11693
 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=IrSo1KIBPbwSxQvKUihj6OUdw-q-P5vBohVWQTT8AjU,520
+rbx/resources/presets/default/preset.rbx.yml,sha256=uyZz6ro_cHHxGDQe22zmpOfpb_-zHcp1fx7aGecWd5A,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.4.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-rbx_cp-0.17.4.dist-info/METADATA,sha256=2t-I-lOYfk3vuXaByEueUFyqMaY2LTkJOqcfIboboUw,4799
-rbx_cp-0.17.4.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-rbx_cp-0.17.4.dist-info/entry_points.txt,sha256=Gw2_BZ5Jon61biaH_ETbAQGXy8fR5On9gw2U4A1erpo,40
-rbx_cp-0.17.4.dist-info/RECORD,,
+rbx/utils.py,sha256=VXAqDewr2UvLtC-JxLPMd3dt26oFu1vtf5bLvbFTYEI,11362
+rbx_cp-0.17.6.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+rbx_cp-0.17.6.dist-info/METADATA,sha256=tuH4FwzQfe95iMusJLIElbUDuL2VCkJYjhmIKg6jagY,4799
+rbx_cp-0.17.6.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+rbx_cp-0.17.6.dist-info/entry_points.txt,sha256=Gw2_BZ5Jon61biaH_ETbAQGXy8fR5On9gw2U4A1erpo,40
+rbx_cp-0.17.6.dist-info/RECORD,,