diff --git a/main.py b/main.py index 4b24fd8..8cadc05 100644 --- a/main.py +++ b/main.py @@ -1,17 +1,19 @@ -import os import argparse import concurrent.futures -import mail as ml -from datetime import date, datetime -import dotenv -import ovh -import fetcher as ft -from urllib.request import urlretrieve import logging -from logging.handlers import TimedRotatingFileHandler -import traceback +import os import sqlite3 import time as tm +import traceback +from datetime import date, datetime +from logging.handlers import TimedRotatingFileHandler +from urllib.request import urlretrieve + +import dotenv +import ovh + +import fetcher as ft +import mail as ml def init(): @@ -259,7 +261,8 @@ def save_pdf(bill: dict) -> None: Télécharge le PDF d’une facture dans un sous-dossier par année. Noms de fichiers : .pdf """ - year_dir = os.path.join(PATH_OVH, str(datetime.now().year)) + fmt_year = str(datetime.now().year) + "/ovh/" + year_dir = os.path.join(PATH_OVH, fmt_year) os.makedirs(year_dir, exist_ok=True) dest = os.path.join(year_dir, f"{bill['billId']}.pdf") @@ -296,7 +299,11 @@ if __name__ == "__main__": init() start = tm.time() logger.info("Démarrage du traitement des factures OVH pour %s", YEAR) - os.makedirs(os.path.join(PATH_OVH, str(YEAR)), exist_ok=True) + final_path = str(YEAR) + "/ovh/" + os.makedirs( + os.path.join(PATH_OVH, final_path), + exist_ok=True, + ) ids_candidats = indexer(get_ids()) bills_json = []