20 lines
924 B
Python
20 lines
924 B
Python
import os
|
|
import psycopg2
|
|
|
|
DSN = os.environ.get('PG_DSN')
|
|
store_id = int(os.environ.get('STORE_ID','2790685415443269'))
|
|
conn = psycopg2.connect(DSN)
|
|
conn.autocommit = True
|
|
cur = conn.cursor()
|
|
rows = []
|
|
for code in ('INIT_DWD_SCHEMA','DWD_LOAD_FROM_ODS','DWD_QUALITY_CHECK'):
|
|
cur.execute("SELECT task_id FROM etl_admin.etl_task WHERE task_code=%s AND store_id=%s", (code, store_id))
|
|
if cur.fetchone():
|
|
cur.execute("UPDATE etl_admin.etl_task SET enabled=TRUE, updated_at=now() WHERE task_code=%s AND store_id=%s", (code, store_id))
|
|
rows.append((code, 'updated'))
|
|
else:
|
|
cur.execute("INSERT INTO etl_admin.etl_task(task_code,store_id,enabled,cursor_field,window_minutes_default,overlap_seconds,page_size,params) VALUES (%s,%s,TRUE,NULL,60,120,1000,'{}') RETURNING task_id", (code, store_id))
|
|
rows.append((code, 'inserted', cur.fetchone()[0]))
|
|
print(rows)
|
|
cur.close(); conn.close()
|