fix: Execute and Parse test for DuckDB

voltrondata · Jun 24, 2023 · dfeb7f6 · dfeb7f6
1 parent 2b68adf
commit dfeb7f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/backend/duckdb.py b/api/backend/duckdb.py
@@ -52,7 +52,7 @@ def check(self):
     def execute(self, data):
         try:
             if self.check()["db_health"] != "up and running":
-                self.connection = self.connect()
+                self.connect()
             for i in data:
                 self.connection.execute(query=i)
             return {"message": "DuckDB Operation successful"}
@@ -66,7 +66,7 @@ def execute(self, data):
     def parse(self, data):
         try:
             if self.check()["db_health"] != "up and running":
-                self.connection = self.connect()
+                self.connect()
             result = self.connection.get_substrait_json(data["query"]).fetchone()[0]
             return result
         except Exception as e: