Skip to content

Commit 5b780b8

Browse files
committed
Implement dunder str for Status enum. Refactor get_submissions and create_submissions since batched returns tuple.
1 parent d8eb378 commit 5b780b8

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

src/judge0/api.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -69,11 +69,10 @@ def create_submissions(
6969
batch_size = client.EFFECTIVE_SUBMISSION_BATCH_SIZE
7070
result_submissions = []
7171
for submission_batch in batched(submissions, batch_size):
72-
submissions_list = list(submission_batch)
7372
if batch_size > 1:
74-
result_submissions.extend(client.create_submissions(submissions_list))
73+
result_submissions.extend(client.create_submissions(submission_batch))
7574
else:
76-
result_submissions.append(client.create_submission(submissions_list[0]))
75+
result_submissions.append(client.create_submission(submission_batch[0]))
7776

7877
return result_submissions
7978

@@ -93,14 +92,14 @@ def get_submissions(
9392
batch_size = client.EFFECTIVE_SUBMISSION_BATCH_SIZE
9493
result_submissions = []
9594
for submission_batch in batched(submissions, batch_size):
96-
submissions_list = list(submission_batch)
95+
9796
if batch_size > 1:
9897
result_submissions.extend(
99-
client.get_submissions(submissions_list, fields=fields)
98+
client.get_submissions(submission_batch, fields=fields)
10099
)
101100
else:
102101
result_submissions.append(
103-
client.get_submission(submissions_list[0], fields=fields)
102+
client.get_submission(submission_batch[0], fields=fields)
104103
)
105104

106105
return result_submissions

src/judge0/base_types.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,3 +57,6 @@ class Status(IntEnum):
5757
RUNTIME_ERROR_OTHER = 12
5858
INTERNAL_ERROR = 13
5959
EXEC_FORMAT_ERROR = 14
60+
61+
def __str__(self):
62+
return self.name.lower().replace("_", " ").title()

0 commit comments

Comments
 (0)