Fix pk in tg_p1 curves

Som-Energia · Jan 15, 2024 · 028a3c5 · 028a3c5
1 parent cc9950a
commit 028a3c5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cchloader/backends/timescaledb.py b/cchloader/backends/timescaledb.py
@@ -96,9 +96,13 @@ def insert_cch_batch_chunk(self, collection, batch):
         fields_to_update.append('write_date=EXCLUDED.create_date')
         fields_to_update.append('write_uid=EXCLUDED.create_uid')
 
-        sql = "INSERT INTO {} ({}) VALUES %s ON CONFLICT (name, utc_timestamp) DO UPDATE SET {};".format(
+        on_conflict = 'name, utc_timestamp'
+        if collection == 'tg_p1':
+            on_conflict = 'name, utc_timestamp, type'
+        sql = "INSERT INTO {} ({}) VALUES %s ON CONFLICT ({}) DO UPDATE SET {};".format(
             collection,
             ','.join(field_names),
+            on_conflict,
             ', '.join(fields_to_update)
         )