Feat: adapt to new excel format

This commit is contained in:
Bertrand Benjamin 2024-04-15 11:59:45 +02:00
parent b60fa3be17
commit 215e26b84f

View File

@ -12,9 +12,9 @@ logger.setLevel(logging.DEBUG)
def extract_cat(cat: pd.DataFrame): def extract_cat(cat: pd.DataFrame):
cat_drop = list(cat[cat["Nouvelles"] == "NE PAS IMPORTER"]["Anciennes"]) cat_drop = list(cat[cat["Nouvelles"] == "NE PAS IMPORTER"])
# cat_drop = list(cat[cat["Nouvelles"] == "NE PAS IMPORTER"]["Anciennes"])
cat_trans = cat[cat["Nouvelles"] != "NE PAS IMPORTER"] cat_trans = cat[cat["Nouvelles"] != "NE PAS IMPORTER"]
trans = {} trans = {}
for _, (old, new) in cat_trans.iterrows(): for _, (old, new) in cat_trans.iterrows():
trans[old] = new trans[old] = new
@ -140,7 +140,7 @@ FLUXES_CRG = {
), ),
"2022 - charge.xlsx": Flux( "2022 - charge.xlsx": Flux(
sources=[ sources=[
ExcelSource(filename="2022 - charge.xlsx", sheet_name="Sheet1"), ExcelSource(filename="2022 - charge.xlsx", sheet_name="DB CRG"),
], ],
transformation=Transformation( transformation=Transformation(
function=trans_2022_charge, function=trans_2022_charge,