Add kwargs in sync_execute and async_execute

judge0 · fkdosilovic · Nov 12, 2024 · Nov 11, 2024 · Nov 11, 2024 · Nov 12, 2024
commit b2ba9dac8e16a1ac7b117125c078b1cafb48d14a
diff --git a/examples/0004_hello_world.py b/examples/0004_hello_world.py
@@ -0,0 +1,3 @@
+import judge0
+result = judge0.run(source_code="print('Hello Judge0')")
+print(result.stdout)
diff --git a/src/judge0/api.py b/src/judge0/api.py
@@ -95,7 +95,17 @@ def async_execute(
     *,
     client: Optional[Union[Client, Flavor]] = None,
     submissions: Optional[Union[Submission, list[Submission]]] = None,
+    source_code: Optional[str] = None,
+    **kwargs,
 ) -> Union[Submission, list[Submission]]:
+    if submissions is not None and source_code is not None:
+        raise ValueError(
+            "source_code argument cannot be provided if submissions argument is provided."
+        )
+
+    if source_code is not None:
+        submissions = Submission(source_code=source_code, **kwargs)
+
     # Check the edge cases if client is not provided.
     if client is None:
         if submissions is None:
@@ -119,7 +129,17 @@ def sync_execute(
     *,
     client: Optional[Union[Client, Flavor]] = None,
     submissions: Optional[Union[Submission, list[Submission]]] = None,
+    source_code: Optional[str] = None,
+    **kwargs,
 ) -> Union[Submission, list[Submission]]:
+    if submissions is not None and source_code is not None:
+        raise ValueError(
+            "source_code argument cannot be provided if submissions argument is provided."
+        )
+
+    if source_code is not None:
+        submissions = Submission(source_code=source_code, **kwargs)
+
     # Check the edge cases if client is not provided.
     if client is None:
         if submissions is None: