Merge pull request EvgSkv#6 from EvgSkv/dev

Dev
xeirniomJ · Oct 16, 2020 · 08f540a · 08f540a
2 parents c4ae569 + 064f26a
commit 08f540a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/colab_logica.py b/colab_logica.py
@@ -84,8 +84,13 @@ def RunSQL(sql, engine):
     return client.query(sql).to_dataframe()
   elif engine == 'psql':
     return pandas.read_sql(sql, DB_CONNECTION)
+  elif engine == 'sqlite':
+    statements = parse.SplitRaw(sql, ';')
+    for s in statements[:-2]:
+      cursor = DB_CONNECTION.execute(s)
+    return pandas.read_sql(statements[-2], DB_CONNECTION)
   else:
-    raise Exception('Logica colab only supports BigQuery and PostgreSQL '
+    raise Exception('Logica only supports BigQuery, PostgreSQL and SQLite '
                     'for now.')