archive creator update

now properly makes artifacts
NSPC911 · Dec 13, 2024 · 7ca710a · 7ca710a
1 parent 928b44a
commit 7ca710a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 8 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -13,11 +13,20 @@ jobs:
 
       - name: Run Archive Creator script
         run: |
-          python archive_creator.py
+          python archive_creator.py --workflow
       
-      - name: Upload as artifact
+      - name: Upload bash artifact
         uses: actions/upload-artifact@v4
         with:
-          path: archives/
+          name: bash
+          path: archives/bash
+          if-no-files-found: warn
+          compression-level: 9
+
+      - name: Upload bat artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: bat
+          path: archives/bat
           if-no-files-found: warn
           compression-level: 9
diff --git a/archive_creator.py b/archive_creator.py
@@ -1,7 +1,10 @@
-import zipfile, os, hashlib
+import os, hashlib, argparse
+from zipfile import ZipFile, ZIP_DEFLATED
 from colorama import *
 init(strip=False, convert=False, autoreset=True)
-
+parser = argparse.ArgumentParser(description=None)
+parser.add_argument("--workflow", action="store_true", default=False, help="Enable workflow mode")
+args = parser.parse_args()
 
 root_dir = os.path.dirname(os.path.abspath(__file__))
 archives_dir = os.path.join(root_dir, 'archives')
@@ -20,7 +23,7 @@
 
 for archive_name, files_to_include in archive_specs.items():
     archive_path = os.path.join(archives_dir, archive_name)
-    with zipfile.ZipFile(archive_path, 'w', zipfile.ZIP_DEFLATED) as archive:
+    with ZipFile(archive_path, 'w', ZIP_DEFLATED) as archive:
         for file in files_to_include:
             archive.write(os.path.join(root_dir, file), file)
 
@@ -34,5 +37,13 @@
     for byte_block in iter(lambda: f.read(4096), b""):
         hasher.update(byte_block)
     bashhash = hasher.hexdigest()
-print(f"{Fore.YELLOW}SHA256{Fore.WHITE} for {Fore.BLUE}bat.zip{Fore.WHITE}  is {Fore.RED}{bathash}")
-print(f"{Fore.YELLOW}SHA256{Fore.WHITE} for {Fore.BLUE}bash.zip{Fore.WHITE} is {Fore.RED}{bashhash}")
+print(f"{Fore.YELLOW}SHA256{Fore.WHITE} for {Fore.LIGHTBLUE_EX}bat.zip{Fore.WHITE}  is {Fore.RED}{bathash}")
+print(f"{Fore.YELLOW}SHA256{Fore.WHITE} for {Fore.LIGHTBLUE_EX}bash.zip{Fore.WHITE} is {Fore.RED}{bashhash}")
+
+if args.workflow:
+    with ZipFile(f'{archives_dir}/bat.zip', 'r', ZIP_DEFLATED) as archive:
+        archive.extractall(f'{archives_dir}/bat')
+    os.remove(f'{archives_dir}/bat.zip')
+    with ZipFile(f'{archives_dir}/bash.zip', 'r', ZIP_DEFLATED) as archive:
+        archive.extractall(f'{archives_dir}/bash')
+    os.remove(f'{archives_dir}/bash.zip')