files: fix typo in filename method

Signed-off-by: pamfilos <[email protected]>
cern-sis · Apr 25, 2024 · 50ba8fb · 50ba8fb
1 parent 3af17a8
commit 50ba8fb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dags/common/scoap3_s3.py b/dags/common/scoap3_s3.py
@@ -11,14 +11,16 @@
 FILE_EXTENSIONS = {
     "pdf": ".pdf",
     "xml": ".xml",
-    "pdfa": ".a_2b.pdf"
+    "pdfa": ".pdf"
 }
 
 def update_filename_extension(filename, type):
     extension = FILE_EXTENSIONS.get(type, "")
-    if filename.endsWith(extension):
+    if filename.endswith(extension):
         return filename
     elif extension:
+        if type == "pdfa":
+            extension = f".a-2b.pdf"
         return f"{filename}{extension}"
 
 class Scoap3Repository(IRepository):