From 34cfc5895941f34cb2a6425d8491b8cefc01c25f Mon Sep 17 00:00:00 2001 From: Lorenz Stechauner Date: Wed, 19 Jul 2023 10:56:13 +0200 Subject: [PATCH] migrate.py: sort payment_delivery the right way --- wgmaster/migrate.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/wgmaster/migrate.py b/wgmaster/migrate.py index 8e8b94a..0737c97 100755 --- a/wgmaster/migrate.py +++ b/wgmaster/migrate.py @@ -1521,10 +1521,9 @@ def migrate_payments(in_dir: str, out_dir: str) -> None: with utils.csv_open(f'{out_dir}/payment_delivery.csv') as f_del_pay: f_del_pay.header('year', 'did', 'dpnr', 'avnr', 'bucket_1', 'bucket_2', 'bucket_3', 'amount') - for p in utils.csv_parse_dict(f'{in_dir}/TLieferungen.csv'): - if p['LINR'] not in DELIVERY_MAP: - continue - y, did, dpnr = DELIVERY_MAP[p['LINR']] + deliveries = {d['LINR']: d for d in utils.csv_parse_dict(f'{in_dir}/TLieferungen.csv')} + for linr, (y, did, dpnr) in DELIVERY_MAP.items(): + p = deliveries[linr] if y not in variant_year_map: continue for aznr, avnr, tznr in variant_year_map[y]: