Add to_dict for Submission serialization to dict/json. Update decoding to handle some edge cases.

Author: fkdosilovic

src/judge0/clients.py

@@ -59,21 +59,11 @@ def get_statuses(self) -> list[dict]:
     def create_submission(self, submission: Submission):
         # TODO: check if client supports specified language_id
         params = {
-            "base64_encoded": "true",  # TODO: Make customizable later.
+            "base64_encoded": "true",
             "wait": str(self.wait).lower(),
         }
 
-        body = {
-            "source_code": submission.encode(submission.source_code),
-            "language_id": submission.language_id,
-        }
-
-        if submission.stdin:
-            body["stdin"] = submission.encode(submission.stdin)
-        if submission.expected_output:
-            body["expected_output"] = submission.encode(submission.expected_output)
-
-        submission.update_extra_request_fields(body)
+        body = submission.to_dict()
 
         resp = requests.post(
             f"{self.endpoint}/submissions",

src/judge0/submission.py

@@ -1,6 +1,5 @@
 from base64 import b64decode, b64encode
-from dataclasses import dataclass
-from typing import Union
+
 
 ENCODED_REQUEST_FIELDS = {
     "source_code",
@@ -44,7 +43,14 @@
 FIELDS = REQUEST_FIELDS | RESPONSE_FIELDS
 
 
-@dataclass
+def encode(text: str) -> str:
+    return b64encode(bytes(text, "utf-8")).decode()
+
+
+def decode(b64_encoded_str: str) -> str:
+    return b64decode(b64_encoded_str.encode()).decode(errors="backslashreplace")
+
+
 class Submission:
     """
     Stores a representation of a Submission to/from Judge0.
@@ -116,24 +122,30 @@ def __init__(
         self.wall_time = None
         self.memory = None
 
-    def encode(self, text: str) -> str:
-        return b64encode(bytes(text, "utf-8")).decode()
+    def set_attributes(self, attributes):
+        for attr, value in attributes.items():
+            if attr in ENCODED_FIELDS:
+                setattr(self, attr, decode(value) if value else None)
+            else:
+                setattr(self, attr, value)
+
+    def to_dict(self) -> dict:
+        body = {
+            "source_code": encode(self.source_code),
+            "language_id": self.language_id,
+        }
 
-    def decode(self, bytes_string: str) -> str:
-        return b64decode(bytes_string.encode()).decode()
+        if self.stdin is not None:
+            body["stdin"] = encode(self.stdin)
+        if self.expected_output is not None:
+            body["expected_output"] = encode(self.expected_output)
 
-    def update_extra_request_fields(self, body):
         for field in EXTRA_REQUEST_FIELDS:
             value = getattr(self, field)
             if value is not None:
                 body[field] = value
 
-    def set_attributes(self, attributes):
-        for attr, value in attributes.items():
-            if attr in ENCODED_FIELDS:
-                setattr(self, attr, self.decode(value) if value else None)
-            else:
-                setattr(self, attr, value)
+        return body
 
 
 class SingleFileSubmission(Submission):