整理项目

tmp/tmp_dwd_tasks.py

@@ -0,0 +1,19 @@
+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()