diff options
Diffstat (limited to 'python/db_csvheader.py')
| -rw-r--r-- | python/db_csvheader.py | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/python/db_csvheader.py b/python/db_csvheader.py index 4f286584..6b9d4f66 100644 --- a/python/db_csvheader.py +++ b/python/db_csvheader.py @@ -11,17 +11,16 @@ if __name__ == "__main__": ) args = parser.parse_args() conn = dbconn("dawndb") - with open(f"/home/serenitas/flint/{args.fname}.csv") as csvfile: + 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: - 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() + 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 |