GO: allow package main;; closes #544

DMOJ · Nov 9, 2019 · 82f0a37 · 82f0a37
1 parent 9791337
commit 82f0a37
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dmoj/executors/GO.py b/dmoj/executors/GO.py
@@ -5,6 +5,7 @@
 
 reinline_comment = re.compile(br'//.*?(?=[\r\n])')
 recomment = re.compile(br'/\*.*?\*/', re.DOTALL)
+repackage = re.compile(br'\s*package\s+main\b')
 
 
 def decomment(x):
@@ -51,6 +52,6 @@ def get_version_flags(cls, command):
 
     def create_files(self, problem_id, source_code, *args, **kwargs):
         source_lines = decomment(source_code).strip().split(b'\n')
-        if source_lines[0].strip().split() != [b'package', b'main']:
+        if not repackage.match(source_lines[0]):
             raise CompileError(b'Your code must be defined in package main.\n')
         super().create_files(problem_id, source_code, *args, **kwargs)