1 files changed, 27 insertions, 0 deletions
diff --git a/python/db_csvheader.py b/python/db_csvheader.py
new file mode 100644
index 00000000..4f286584
--- /dev/null
+++ b/python/db_csvheader.py
@@ -0,0 +1,27 @@
+from serenitas.utils.db import dbconn
+import csv
+import argparse
+
+if __name__ == "__main__":
+    # First run --save, then run --accept if it looks good
+    parser = argparse.ArgumentParser(description="Save, accept within tolerances")
+    parser.add_argument(
+        "fname",
+        nargs="?",
+    )
+    args = parser.parse_args()
+    conn = dbconn("dawndb")
+    with open(f"/home/serenitas/flint/{args.fname}.csv") as csvfile:
+        csvreader = csv.reader(csvfile)
+        for row in csvreader:
+            sql_str = "INSERT INTO csv_templates values (%s, %s) ON CONFLICT DO NOTHING"
+            with conn.cursor() as c:
+                c.execute(sql_str, (args.fname, row))
+                c.execute(
+                    "SELECT headers FROM csv_templates where template_name=%s",
+                    (args.fname,),
+                )
+                for row in c:
+                    print(row.headers)
+            conn.commit()
+            break