bn_user
/
sigma
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Обновить dags/upload_dicts.py

This commit is contained in:
bn_user 2025-11-13 16:02:28 +00:00
parent 893e36809e
commit 4667122ca3
1 changed files with 27 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
dags/upload_dicts.py
View File

@ -77,16 +77,24 @@ def take_dogovor(**kwargs):
index=False, index=False,
method='multi' method='multi'
) )
conn.execute(f"DELETE FROM public.{table_name}") # conn.execute(f"DELETE FROM public.{table_name}")
conn.execute(f""" conn.execute(f"""
INSERT INTO public.{table_name} (name, nomer, date_begin, date_end, uid_dogovor) INSERT INTO public.{table_name} (id, name, nomer, date_begin, date_end, uid_dogovor)
SELECT SELECT
row_number() over (order by uid_dogovor) id,
name, name,
nomer, nomer,
date_begin, date_begin,
date_end, date_end,
uid_dogovor uid_dogovor
FROM temp_{table_name} FROM temp_{table_name}
ON CONFLICT (uid_dogovor)
DO UPDATE SET
id = EXCLUDED.id,
name = EXCLUDED.name,
nomer = EXCLUDED.nomer,
date_begin = EXCLUDED.date_begin,
date_end = EXCLUDED.date_end;
""" """
) )
@ -134,15 +142,22 @@ def take_bank(**kwargs):
index=False, index=False,
method='multi' method='multi'
) )
conn.execute(f"DELETE FROM public.{table_name}") # conn.execute(f"DELETE FROM public.{table_name}")
conn.execute(f""" conn.execute(f"""
INSERT INTO public.{table_name} (name, inn, kpp, uid_bank) INSERT INTO public.{table_name} (name, inn, kpp, uid_bank)
SELECT SELECT
row_number() over (order by uid_bank) id,
name, name,
inn, inn,
kpp, kpp,
uid_bank uid_bank
FROM temp_{table_name} FROM temp_{table_name}
ON CONFLICT (uid_bank)
DO UPDATE SET
id = EXCLUDED.id,
name = EXCLUDED.name,
inn = EXCLUDED.inn,
kpp = EXCLUDED.date_begin;
""" """
) )
@ -189,7 +204,7 @@ def take_organizaciya(**kwargs):
index=False, index=False,
method='multi' method='multi'
) )
conn.execute(f"DELETE FROM public.{table_name}") # conn.execute(f"DELETE FROM public.{table_name}")
conn.execute(f""" conn.execute(f"""
INSERT INTO public.{table_name} (name, inn, kpp, uid_organizaciya) INSERT INTO public.{table_name} (name, inn, kpp, uid_organizaciya)
SELECT SELECT
@ -198,6 +213,12 @@ def take_organizaciya(**kwargs):
kpp, kpp,
uid_organizaciya uid_organizaciya
FROM temp_{table_name} FROM temp_{table_name}
ON CONFLICT (uid_organizaciya)
DO UPDATE SET
id = EXCLUDED.id,
name = EXCLUDED.name,
inn = EXCLUDED.inn,
kpp = EXCLUDED.date_begin;
""" """
) )