simplified connection

validate_sql.py

@@ -1,31 +1,10 @@
 import sys
-import shutil
-import os
-import uuid
 import duckdb
-from duckdb import ParserException, SyntaxException, BinderException, CatalogException
-
-TMP_DIR = "tmp"
-class WithDuckDBConnectionInTmpDir(object):
-    def __init__(self):
-        self.tmp_dir = TMP_DIR + str(uuid.uuid1())
-        os.makedirs(self.tmp_dir)
-        self.original_wd = os.getcwd()
-
-    def __enter__(self):
-        os.chdir(self.tmp_dir)
-        self.con = duckdb.connect()
-        self.con.execute("SET enable_external_access=False")
-        return self.con
-
-    def __exit__(self, *args):
-        self.con.close()
-        os.chdir(self.original_wd)
-        shutil.rmtree(self.tmp_dir)
+from duckdb import ParserException, SyntaxException, BinderException
 
 def validate_query(query, schemas):
     try:
-        with WithDuckDBConnectionInTmpDir() as duckdb_conn:
+        with duckdb.connect(":memory:", config={"enable_external_access": False}) as duckdb_conn:
             # register schemas
             for schema in schemas.split(";"):
                 duckdb_conn.execute(schema)
@@ -47,8 +26,6 @@ def validate_query(query, schemas):
             raise e
         elif "Catalog Error: Copy Function" in message:
             raise e
-        else:
-            print(message)
 
 if __name__ == '__main__':
     if len(sys.argv) > 2: