Add Config dataclass for storing client's info from get_config_info route. Use client.config.max_submission_batch_size in get_submissions and create_submissions.

fkdosilovic · fkdosilovic · commit c98e077db039 · 2024-11-25T20:53:52.000+01:00
diff --git a/src/judge0/api.py b/src/judge0/api.py
@@ -65,10 +65,9 @@ def create_submissions(
     if isinstance(submissions, Submission):
         return client.create_submission(submissions)
 
-    # TODO: Use result from get_config.
     result_submissions = []
     for submission_batch in batched(
-        submissions, client.EFFECTIVE_SUBMISSION_BATCH_SIZE
+        submissions, client.config.max_submission_batch_size
     ):
         if len(submission_batch) > 1:
             result_submissions.extend(client.create_submissions(submission_batch))
@@ -89,10 +88,9 @@ def get_submissions(
     if isinstance(submissions, Submission):
         return client.get_submission(submissions, fields=fields)
 
-    # TODO: Use result from get_config.
     result_submissions = []
     for submission_batch in batched(
-        submissions, client.EFFECTIVE_SUBMISSION_BATCH_SIZE
+        submissions, client.config.max_submission_batch_size
     ):
         if len(submission_batch) > 1:
             result_submissions.extend(
diff --git a/src/judge0/base_types.py b/src/judge0/base_types.py
@@ -89,3 +89,46 @@ class Status(IntEnum):
 
     def __str__(self):
         return self.name.lower().replace("_", " ").title()
+
+
+@dataclass(frozen=True)
+class Config:
+    allow_enable_network: bool
+    allow_enable_per_process_and_thread_memory_limit: bool
+    allow_enable_per_process_and_thread_time_limit: bool
+    allowed_languages_for_compile_options: list[str]
+    callbacks_max_tries: int
+    callbacks_timeout: float
+    cpu_extra_time: float
+    cpu_time_limit: float
+    enable_additional_files: bool
+    enable_batched_submissions: bool
+    enable_callbacks: bool
+    enable_command_line_arguments: bool
+    enable_compiler_options: bool
+    enable_network: bool
+    enable_per_process_and_thread_memory_limit: bool
+    enable_per_process_and_thread_time_limit: bool
+    enable_submission_delete: bool
+    enable_wait_result: bool
+    maintenance_mode: bool
+    max_cpu_extra_time: float
+    max_cpu_time_limit: float
+    max_extract_size: int
+    max_file_size: int
+    max_max_file_size: int
+    max_max_processes_and_or_threads: int
+    max_memory_limit: int
+    max_number_of_runs: int
+    max_processes_and_or_threads: int
+    max_queue_size: int
+    max_stack_limit: int
+    max_submission_batch_size: int
+    max_wall_time_limit: float
+    memory_limit: int
+    number_of_runs: int
+    redirect_stderr_to_stdout: bool
+    stack_limit: int
+    submission_cache_duration: float
+    use_docs_as_homepage: bool
+    wall_time_limit: float
diff --git a/src/judge0/clients.py b/src/judge0/clients.py
@@ -2,7 +2,7 @@
 
 import requests
 
-from .base_types import Language, LanguageAlias
+from .base_types import Config, Language, LanguageAlias
 from .data import LANGUAGE_TO_LANGUAGE_ID
 from .submission import Submission, Submissions
 
@@ -20,10 +20,8 @@ def __init__(self, endpoint, auth_headers) -> None:
         self.auth_headers = auth_headers
 
         try:
-            self.languages = [
-                Language(id=lang["id"], name=lang["name"])
-                for lang in self.get_languages()
-            ]
+            self.languages = [Language(**lang) for lang in self.get_languages()]
+            self.config = Config(**self.get_config_info())
         except Exception as e:
             raise RuntimeError(
                 f"Authentication failed. Visit {self.HOME_URL} to get or review your authentication credentials."
