updated validation

app.py

@@ -66,8 +66,12 @@ def validate_sql(query, schema):
         )
         # Get output and potential parser, and binder error message
         stdout, stderr = process.communicate(timeout=0.5)
-        if stdout != "":
-            return False, stdout.decode('utf8')
+        if stderr:
+            error_message = stderr.decode('utf8').split("\n")
+            # skip traceback
+            if len(error_message) > 3:
+                error_message = "\n".join(error_message[3:])
+            return False, error_message
         return True, ""
     except subprocess.TimeoutExpired:
         process.kill()

validate_sql.py

@@ -14,21 +14,23 @@ def validate_query(query, schemas):
             cursor = duckdb_conn.cursor()
             cursor.execute(query)
     except ParserException as e:
-        print(str(e))
+        return str(e)
     except SyntaxException as e:
-        print(str(e))
+        return str(e)
     except BinderException as e:
-        print(str(e))
+        return str(e)
     except CatalogException as e:
         if not ("but it exists" in str(e) and "extension" in str(e)):
-            print(str(e))
+            return str(e)
     except Exception as e:
-        return True
-    return True
+        return None
+    return None
 
 
 if __name__ == "__main__":
     if len(sys.argv) > 2:
-        validate_query(sys.argv[1], sys.argv[2])
+        error = validate_query(sys.argv[1], sys.argv[2])
+        if error:
+            raise Exception(error)
     else:
         print("No query provided.")