Add pre_execution_copy of Submission.

Author: fkdosilovic

pyproject.toml

@@ -33,9 +33,9 @@ requires = ["setuptools"]
 build-backend = "setuptools.build_meta"
 
 [project.urls]
-Homepage = "https://github.com/judge0/judge0-py"
-Repository = "https://github.com/judge0/judge0-py.git"
-Issues = "https://github.com/judge0/judge0-py/issues"
+Homepage = "https://github.com/judge0/judge0-python"
+Repository = "https://github.com/judge0/judge0-python.git"
+Issues = "https://github.com/judge0/judge0-python/issues"
 
 [project.optional-dependencies]
 test = ["pytest", "mkdocs"]

src/judge0/api.py

@@ -175,7 +175,7 @@ def create_submissions_from_test_cases(
     all_submissions = []
     for submission in submissions_list:
         for test_case in test_cases_list:
-            submission_copy = submission.copy()
+            submission_copy = submission.pre_execution_copy()
             if test_case is not None:
                 submission_copy.stdin = test_case.input
                 submission_copy.expected_output = test_case.expected_output

src/judge0/submission.py

@@ -1,6 +1,6 @@
 import copy
 from datetime import datetime
-from typing import Union
+from typing import Optional, Union
 
 from judge0.filesystem import Filesystem
 
@@ -73,9 +73,9 @@ class Submission:
 
     def __init__(
         self,
-        source_code: str,
-        language: Union[LanguageAlias, int] = LanguageAlias.PYTHON,
         *,
+        source_code: Optional[str] = None,
+        language: Union[LanguageAlias, int] = LanguageAlias.PYTHON,
         additional_files=None,
         compiler_options=None,
         command_line_arguments=None,
@@ -180,8 +180,11 @@ def is_done(self) -> bool:
         else:
             return self.status not in (Status.IN_QUEUE, Status.PROCESSING)
 
-    def copy(self) -> "Submission":
-        return copy.deepcopy(self)
+    def pre_execution_copy(self) -> "Submission":
+        new_submission = Submission()
+        for attr in REQUEST_FIELDS:
+            setattr(new_submission, attr, copy.deepcopy(getattr(self, attr)))
+        return new_submission
 
     def __repr__(self) -> str:
         arguments = ", ".join(f"{field}={getattr(self, field)!r}" for field in FIELDS)

tests/test_api_test_cases.py

@@ -9,10 +9,10 @@
 @pytest.mark.parametrize(
     "submissions,test_cases,expected_type",
     [
-        [Submission(""), TestCase(), Submission],
-        [[Submission("")], TestCase(), list],
-        [Submission(""), [TestCase()], list],
-        [[Submission("")], [TestCase()], list],
+        [Submission(source_code=""), TestCase(), Submission],
+        [[Submission(source_code="")], TestCase(), list],
+        [Submission(source_code=""), [TestCase()], list],
+        [[Submission(source_code="")], [TestCase()], list],
     ],
 )
 def test_create_submissions_from_test_cases_return_type(
@@ -39,14 +39,14 @@ def test_create_submissions_from_test_cases_return_type(
             [Status.ACCEPTED, Status.ACCEPTED],
         ],
         [
-            Submission("print(f'Hello, {input()}')"),
+            Submission(source_code="print(f'Hello, {input()}')"),
             [
                 TestCase("Judge0", "Hello, Judge0"),
             ],
             [Status.ACCEPTED],
         ],
         [
-            Submission("print(f'Hello, {input()}')"),
+            Submission(source_code="print(f'Hello, {input()}')"),
             [
                 TestCase("Judge0", "Hello, Judge0"),
                 TestCase("pytest", "Hi, pytest"),
@@ -55,8 +55,8 @@ def test_create_submissions_from_test_cases_return_type(
         ],
         [
             [
-                Submission("print(f'Hello, {input()}')"),
-                Submission("print(f'Hello,  {input()}')"),
+                Submission(source_code="print(f'Hello, {input()}')"),
+                Submission(source_code="print(f'Hello,  {input()}')"),
             ],
             [
                 TestCase("Judge0", "Hello, Judge0"),
@@ -97,7 +97,7 @@ def test_test_cases_from_run(
     [
         [
             Submission(
-                "print(f'Hello, {input()}')",
+                source_code="print(f'Hello, {input()}')",
                 stdin="Judge0",
                 expected_output="Hello, Judge0",
             ),
@@ -106,12 +106,12 @@ def test_test_cases_from_run(
         [
             [
                 Submission(
-                    "print(f'Hello, {input()}')",
+                    source_code="print(f'Hello, {input()}')",
                     stdin="Judge0",
                     expected_output="Hello, Judge0",
                 ),
                 Submission(
-                    "print(f'Hello, {input()}')",
+                    source_code="print(f'Hello, {input()}')",
                     stdin="pytest",
                     expected_output="Hello, pytest",
                 ),

tests/test_submission.py

@@ -3,7 +3,7 @@
 
 def test_status_before_and_after_submission(request):
     client = request.getfixturevalue("judge0_ce_client")
-    submission = Submission('print("Hello World!")')
+    submission = Submission(source_code='print("Hello World!")')
 
     assert submission.status is None
 
@@ -16,7 +16,7 @@ def test_status_before_and_after_submission(request):
 
 def test_is_done(request):
     client = request.getfixturevalue("judge0_ce_client")
-    submission = Submission('print("Hello World!")')
+    submission = Submission(source_code='print("Hello World!")')
 
     assert submission.status is None