From 77d6eedd24ba8112c2a598189cd733ab0af6a235 Mon Sep 17 00:00:00 2001
From: Pinapelz <yukais@pinapelz.com>
Date: Thu, 21 Aug 2025 15:14:07 -0700
Subject: add recovery script for filling backlog of missing datapoints

---
 sql/pg_handler.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'sql')

diff --git a/sql/pg_handler.py b/sql/pg_handler.py
index 68ce82c..a010407 100644
--- a/sql/pg_handler.py
+++ b/sql/pg_handler.py
@@ -143,8 +143,13 @@ class PostgresHandler:
                 cursor.execute(query)
             else:
                 cursor.execute(query, data)
-            result = cursor.fetchall()
-            return result
+            query_type = query.strip().upper().split()[0]
+            if query_type == 'SELECT':
+                result = cursor.fetchall()
+                return result
+            else:
+                self._connection.commit()
+                return True
         except Error as e:
             self._connection.rollback()
             print(f"Failed to execute query: {query}")
-- 
cgit v1.2.3