diff --git a/dags/OSV.py b/dags/OSV.py index d3fde88..bc7cc75 100644 --- a/dags/OSV.py +++ b/dags/OSV.py @@ -100,6 +100,7 @@ def read_data_1C(**kwargs): df = pd.DataFrame(data_from_1c['data']) engine = get_db_engine() table_name = 'oborotno_salbdovaya_vedomostb' + temp_table_name = f'temp_{table_name}' field_mapping = { # Основные поля @@ -135,7 +136,8 @@ def read_data_1C(**kwargs): df = df.rename(columns=field_mapping) with engine.begin() as conn: if not df.empty: - conn.execute(f"CREATE TEMP TABLE temp_{table_name} AS SELECT * FROM public.{table_name} WHERE 1 = 0") + conn.execute(f"DROP TABLE IF EXISTS {temp_table_name}") + conn.execute(f"CREATE TEMP TABLE {temp_table_name} (LIKE public.{table_name})") df.to_sql( f'temp_{table_name}', con=conn,