import logging from logging.config import dictConfig from pathlib import Path import click from .extract import extract_save from .join import join_excel @click.group() @click.option("--debug/--no-debug", default=False) def main(debug): if debug: logging_level = logging.DEBUG else: logging_level = logging.INFO logging_config = dict( version=1, formatters={"f": {"format": "%(levelname)-8s %(name)-12s %(message)s"}}, handlers={ "h": { "class": "logging.StreamHandler", "formatter": "f", "level": logging_level, } }, root={ "handlers": ["h"], "level": logging_level, }, ) dictConfig(logging_config) @main.group() def extract(): pass @extract.command() @click.argument("pdf_file", required=1) @click.option("--dest", help="Où mettre les fichiers produits", default="") def on(pdf_file, dest): if not dest: pdf_path = Path(pdf_file) dest = pdf_path.parent extract_save(pdf_file, dest) @extract.command() @click.option("--src", help="Tous les fichiers dans folder", default="./") @click.option("--dest", help="Où mettre les fichiers produits", default="./") def all(src, dest): p = Path(src) d = Path(dest) d.mkdir(exist_ok=True) pdf_files = [x for x in p.iterdir() if ".pdf" in str(x)] for pdf_file in pdf_files: logging.info(f"Found {pdf_file}") extract_save(pdf_file, d) @main.command() @click.option("--src", help="Tous les fichiers dans src", default="./") @click.option("--dest", help="Où mettre les fichiers produits", default="") @click.option( "--force", help="Ecraser si le ficher destination existe.", default=False, is_flag=True, ) def join(src, dest, force): """Join tous les fichiers excel charge (resp locataire) de src dans un seul fichier charge.xlsx dans dist. Exemple: pdf-oralia join --src --dest """ dest_charge = f"{dest}/charge.xlsx" if not force and Path(dest_charge).exists(): raise ValueError(f"The file {dest_charge} already exists") dest_locataire = f"{dest}/locataire.xlsx" if not force and Path(dest_locataire).exists(): raise ValueError(f"The file {dest_locataire} already exists") if not Path(src).exists(): raise ValueError(f"The source directory ({src}) does not exists.") join_excel(src, dest_charge, "*_charge.xlsx") logging.info(f"Les données charges ont été concaténées dans {dest_charge}") join_excel(src, dest_locataire, "*_locataire.xlsx") logging.info(f"Les données locataires ont été concaténées dans {dest_locataire}")