lafrite/pdf_auralia
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects 1 Releases Wiki Activity

Feat: add dest to on command

Browse Source
This commit is contained in:
Bertrand Benjamin 2022-09-28 05:33:09 +02:00
parent b4e70f00fc
commit 572f58c0a9
2 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pdf_oralia/extract.py
View File

@ -41,8 +41,8 @@ def extract_from_pdf(pdf, charge_dest, location_dest):
def extract_save(pdf_file, dest): def extract_save(pdf_file, dest):
"""Extract charge and locataire for pdf_file and put xlsx file in dest""" """Extract charge and locataire for pdf_file and put xlsx file in dest"""
pdf_file = Path(pdf_file) pdf_file = Path(pdf_file)
xls_charge = dest / f"{pdf_file.stem.replace(' ', '_')}_charge.xlsx" xls_charge = Path(dest) / f"{pdf_file.stem.replace(' ', '_')}_charge.xlsx"
xls_locataire = dest / f"{pdf_file.stem.replace(' ', '_')}_locataire.xlsx" xls_locataire = Path(dest) / f"{pdf_file.stem.replace(' ', '_')}_locataire.xlsx"
pdf = pdfplumber.open(pdf_file) pdf = pdfplumber.open(pdf_file)
extract_from_pdf(pdf, xls_charge, xls_locataire) extract_from_pdf(pdf, xls_charge, xls_locataire)

12
pdf_oralia/scripts.py
View File

@ -17,11 +17,13 @@ def extract():
@extract.command() @extract.command()
@click.argument("pdf_file", required=1) @click.argument("pdf_file", required=1)
def on(pdf_file): @click.option("--dest", help="Où mettre les fichiers produits", default="")
pdf_path = Path(pdf_file) def on(pdf_file, dest):
pdf_filename = pdf_path.name if not dest:
pdf_path = pdf_path.parent pdf_path = Path(pdf_file)
extract_save(pdf_file, pdf_path) dest = pdf_path.parent
extract_save(pdf_file, dest)
@extract.command() @extract.command()