Modification puor utilisation dans flask

main.py

@@ -178,28 +178,18 @@ def zip_directory(directory, zip_file):
                 file_path = os.path.join(root, file)
                 zipf.write(file_path, os.path.relpath(file_path, directory))
 
-
-def main():
-    # Creation du répertoire sources pour mettre les documents
-    create_original_files_directory()
-
-    # Configure le parseur d'arguments
-    parser = argparse.ArgumentParser(description='read arguments')
-    parser.add_argument('--filigramme', type=str, help='Valeur du filigramm comme "agence (juillet 2024)"')
-
-    # Analyse les arguments
-    args = parser.parse_args()
-
+def traitement(paramFiligramme):
     # Lecture de la configuration
     config = load_config()
 
     # Utilise la valeur du filigramme fournie ou celle par défaut de la configuration
-    filigramme = args.filigramme if args.filigramme else config.get('filigrame', 'Protection par filigrame')
+    filigramme = paramFiligramme if paramFiligramme else config.get('filigrame', 'Protection par filigrame')
 
     source_directory = config["source_directory"]
     destination_directory = os.path.join(source_directory, filigramme.replace(" ", "_"))
     temp_directory = os.path.join(source_directory, "temp")
 
+
     # Extrait les pages des pdf
     try:
         print("Extrait les pages des pdf")
@@ -230,6 +220,20 @@ def main():
     except:
         print(f"Aucune image à taguer")
         return {"Aucune image à taguer"}
+def main():
+    # Creation du répertoire sources pour mettre les documents
+    create_original_files_directory()
+
+    # Configure le parseur d'arguments
+    parser = argparse.ArgumentParser(description='read arguments')
+    parser.add_argument('--filigramme', type=str, help='Valeur du filigram comme "agence (juillet 2024)"')
+
+    # Analyse les arguments
+    args = parser.parse_args()
+
+    traitement(args.filigramme)
+
+
 
 
 if __name__ == '__main__':