pdf_auralia/pdf_oralia/scripts.py

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 <dossier_source> --dest <dossier_destination>


    """
    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")

    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}")