Create output schema if not exists

elwinsnijders · Feb 4, 2019 · bbbd6f4 · bbbd6f4
1 parent 34ec63a
commit bbbd6f4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bag3d/importer.py b/bag3d/importer.py
@@ -114,6 +114,8 @@ def csv2db(conn, cfg, out_paths):
 
     table_idx = sql.Identifier(cfg['output']['schema'] + "_id_idx")
     a = create_heights_table(conn, cfg['output']['schema'], cfg['output']['table'])
+
+    conn.sendQuery(sql.SQL("CREATE SCHEMA IF NOT EXISTS {schema};").format(schema=schema_out_q))
 
     if a:
         tbl = ".".join([cfg['output']['schema'], cfg['output']['table']])