Format SQL in tests

arrowresearch · Oct 26, 2020 · eff242d · eff242d
1 parent 2b0067f
commit eff242d
Show file tree

Hide file tree

Showing 2 changed files with 470 additions and 174 deletions.
diff --git a/qc0/q.py b/qc0/q.py
@@ -38,20 +38,20 @@ def print_op(self):
         op = syn_to_op(self.syn, self.meta)
         print(op)
 
-    def sql(self):
+    def sql(self, format=False):
         op = syn_to_op(self.syn, self.meta)
         sql = op_to_sql(op)
         sql = sql.compile(self.engine, compile_kwargs={"literal_binds": True})
         sql = str(sql).strip()
         sql = "\n".join([line.strip() for line in sql.split("\n")])
-        return sql
+        if format:
+            import sqlparse
 
-    def print_sql(self):
-        import sqlparse
+            sql = sqlparse.format(sql, reindent=True, keyword_case="upper")
+        return sql
 
-        sql = self.sql()
-        sql = sqlparse.format(sql, reindent=True, keyword_case="upper")
-        print(sql)
+    def print_sql(self, format=True):
+        print(self.sql(format=format))
 
     def __getattr__(self, name):
         nav = Nav(name=name)