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