This commit is contained in:
Alex 2024-07-25 12:50:14 +00:00
commit 84b08f0d34

33
main.py
View File

@ -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")
@ -228,8 +218,21 @@ def main():
return {"result": "processed data with filigramme: " + filigramme}
except:
print(f"Aucune image à taguer")
return {"Aucune image à taguer"}
return {"result": "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__':