Add get_client public function.

fkdosilovic · fkdosilovic · commit 204bcd628a31 · 2024-11-21T17:22:36.000+01:00
diff --git a/src/judge0/__init__.py b/src/judge0/__init__.py
@@ -1,6 +1,6 @@
 import os
 
-from .api import async_execute, execute, run, sync_execute, wait
+from .api import async_execute, execute, get_client, run, sync_execute, wait
 from .base_types import Flavor, Language, LanguageAlias, Status, TestCase
 from .clients import (
     ATD,
@@ -41,6 +41,7 @@
     "TestCase",
     "async_execute",
     "execute",
+    "get_client",
     "run",
     "sync_execute",
     "wait",
diff --git a/src/judge0/api.py b/src/judge0/api.py
@@ -7,6 +7,18 @@
 from .submission import Submission, Submissions
 
 
+def get_client(flavor: Flavor = Flavor.CE) -> Client:
+    from . import _get_implicit_client
+
+    if isinstance(flavor, Flavor):
+        return _get_implicit_client(flavor=flavor)
+    else:
+        raise ValueError(
+            "Expected argument flavor to be of of type enum Flavor, "
+            f"got {type(flavor)}."
+        )
+
+
 def resolve_client(
     client: Optional[Union[Client, Flavor]] = None,
     submissions: Optional[Union[Submission, Submissions]] = None,
@@ -15,11 +27,8 @@ def resolve_client(
     if isinstance(client, Client):
         return client
 
-    from . import _get_implicit_client
-
-    # User explicitly choose the flavor of the client.
     if isinstance(client, Flavor):
-        return _get_implicit_client(flavor=client)
+        return get_client(flavor=client)
 
     if client is None and isinstance(submissions, list) and len(submissions) == 0:
         raise ValueError("Client cannot be determined from empty submissions.")
@@ -33,7 +42,7 @@ def resolve_client(
     languages = [submission.language for submission in submissions]
 
     for flavor in Flavor:
-        client = _get_implicit_client(flavor)
+        client = get_client(flavor)
         if client is not None and all(
             (client.is_language_supported(lang) for lang in languages)
         ):
diff --git a/tests/test_submission.py b/tests/test_submission.py
@@ -1,5 +1,3 @@
-import pytest
-
 from judge0 import Status, Submission, wait
 
 
@@ -23,6 +21,6 @@ def test_is_done(request):
     assert submission.status is None
 
     client.create_submission(submission)
-    wait(client, submission)
+    wait(client=client, submissions=submission)
 
     assert submission.is_done()
