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")
    file_name, header_name = args.fname.split(".")
    with open(f"/home/serenitas/flint/{header_name}.csv") as csvfile:
        csvreader = csv.reader(csvfile)
        for row in csvreader:
            row = [header.replace("\n", "") for header in row]
            try:
                with open(f"headers/{file_name}.py", "a") as template_file:
                    template_file.write(f"\n{header_name}={row}")
            except:
                with open(f"headers/{file_name}.py", "w") as template_file:
                    template_file.write(f"{header_name}={row}")
            csvfile.close()
            break