Add dunder repr to FileSystem and Submission. Expose async_run and sync_run as public api functions.

fkdosilovic · fkdosilovic · commit d62b456dabaf · 2024-11-23T23:27:46.000+01:00
diff --git a/src/judge0/__init__.py b/src/judge0/__init__.py
@@ -1,6 +1,15 @@
 import os
 
-from .api import async_execute, execute, get_client, run, sync_execute, wait
+from .api import (
+    async_execute,
+    async_run,
+    execute,
+    get_client,
+    run,
+    sync_execute,
+    sync_run,
+    wait,
+)
 from .base_types import Flavor, Language, LanguageAlias, Status, TestCase
 from .clients import (
     ATD,
@@ -42,6 +51,8 @@
     "async_execute",
     "execute",
     "get_client",
+    "async_run",
+    "sync_run",
     "run",
     "sync_execute",
     "wait",
diff --git a/src/judge0/api.py b/src/judge0/api.py
@@ -258,7 +258,6 @@ def sync_execute(
 
 
 execute = sync_execute
-
 run = sync_execute
 sync_run = sync_execute
 async_run = async_execute
diff --git a/src/judge0/filesystem.py b/src/judge0/filesystem.py
@@ -49,6 +49,10 @@ def __init__(
         elif isinstance(content, Filesystem):
             self.files = copy.deepcopy(content.files)
 
+    def __repr__(self) -> str:
+        content_encoded = b64encode(self.encode()).decode()
+        return f"{self.__class__.__name__}(content={content_encoded!r})"
+
     def encode(self) -> bytes:
         zip_buffer = io.BytesIO()
         with zipfile.ZipFile(zip_buffer, "w") as zip_file:
diff --git a/src/judge0/submission.py b/src/judge0/submission.py
@@ -182,3 +182,13 @@ def is_done(self) -> bool:
 
     def copy(self) -> "Submission":
         return copy.deepcopy(self)
+
+    def __repr__(self) -> str:
+        arguments = ", ".join(f"{field}={getattr(self, field)!r}" for field in FIELDS)
+        return f"{self.__class__.__name__}({arguments})"
+
+    def __iter__(self):
+        if self.post_execution_filesystem is None:
+            return iter([])
+        else:
+            return iter(self.post_execution_filesystem)
