fix #320: skip building pdfs for zip --kattis

RagnarGrootKoerkamp · Oct 31, 2023 · aa7e6ee · aa7e6ee
1 parent 455f0aa
commit aa7e6ee
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 11 deletions.
diff --git a/bin/export.py b/bin/export.py
@@ -38,6 +38,8 @@ def revert():
 
 # Write any .lang.pdf files to .pdf.
 def remove_language_suffix(fname, statement_language):
+    if not statement_language:
+        return fname
     out = Path(fname)
     if out.suffixes == ['.' + statement_language, '.pdf']:
         out = out.with_suffix('').with_suffix('.pdf')
@@ -48,13 +50,16 @@ def build_samples_zip(problems, statement_language):
     zf = zipfile.ZipFile(
         'samples.zip', mode="w", compression=zipfile.ZIP_DEFLATED, allowZip64=False
     )
-    for fname in glob(Path('.'), f'contest*.{statement_language}.pdf'):
-        if Path(fname).is_file():
-            zf.write(
-                fname,
-                remove_language_suffix(fname, statement_language),
-                compress_type=zipfile.ZIP_DEFLATED,
-            )
+
+    # Do not include contest PDF for kattis.
+    if not config.args.kattis:
+        for fname in glob(Path('.'), f'contest*.{statement_language}.pdf'):
+            if Path(fname).is_file():
+                zf.write(
+                    fname,
+                    remove_language_suffix(fname, statement_language),
+                    compress_type=zipfile.ZIP_DEFLATED,
+                )
 
     for problem in problems:
         outputdir = Path(problem.label)

diff --git a/bin/tools.py b/bin/tools.py
@@ -951,14 +951,17 @@ def run_parsed_arguments(args):
                 success &= generate.generate(problem)
                 config.args = old_args
 
-                success &= latex.build_problem_pdfs(problem)
+                if not config.args.kattis:
+                    success &= latex.build_problem_pdfs(problem)
+                    # Add problem PDF for only one language to the zip file
+                    statement_language = force_single_language(problems)
+                else:
+                    statement_language = None
+
                 if not config.args.force:
                     success &= problem.validate_format('input_format', constraints={})
                     success &= problem.validate_format('output_format', constraints={})
 
-                # Add problem PDF for only one language to the zip file
-                statement_language = force_single_language(problems)
-
                 # Write to problemname.zip, where we strip all non-alphanumeric from the
                 # problem directory name.
                 success &= export.build_problem_zip(problem, output, statement_language)