bugfix: loading test-cases in txt will not produce a first "ghost" en…

…try using a folder as input source.
DMOJ · Dec 24, 2023 · eb0a63e · eb0a63e
1 parent 75a7381
commit eb0a63e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/judge/models/problem_data.py b/judge/models/problem_data.py
@@ -183,8 +183,8 @@ def load_test_cases_from_zip(self, *args, **kwargs):
             tc.delete()
 
         files = sorted(ZipFile(self.zipfile).namelist())
-        input = [x for x in files if '.in' in x or 'input' in x]
-        output = [x for x in files if '.out' in x or 'output' in x]
+        input = [x for x in files if '.in' in x or ('input' in x and '.' in x)]
+        output = [x for x in files if '.out' in x or ('output' in x and '.' in x)]
 
         cases = []
         for i in range(len(input)):

diff --git a/judge/views/problem_data.py b/judge/views/problem_data.py
@@ -212,9 +212,8 @@ def post(self, request, *args, **kwargs):
                     case.save()
 
             # Once loaded all the test cases (from zip or manually), the content can be setup
-            if data.problem.include_test_cases:
-                data.setup_test_cases_content()
-                data.problem.save()
+            data.setup_test_cases_content()
+            data.problem.save()
 
             ProblemDataCompiler.generate(problem, data, problem.cases.order_by('order'), valid_files)
             return HttpResponseRedirect(request.get_full_path())