From 8c96998c9d1d073b9194d6d5573fdd68ff1f41c9 Mon Sep 17 00:00:00 2001 From: gowthaman Date: Sat, 17 Jun 2023 23:26:24 +0530 Subject: [PATCH] quote column names, so if/any wierd json keys are coming, then it won't fail --- main.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index 9e580f7..649796e 100644 --- a/main.py +++ b/main.py @@ -34,7 +34,7 @@ class DBConn: def make_table(self, tbl_name, cols): syspk = "syspk integer primary key autoincrement" - other_cols = ', '.join([f"{f} TEXT" for f in cols]) + other_cols = ', '.join([f"\"{f}\" TEXT" for f in cols]) create_tbl_sql = f"create table if not exists {tbl_name} ({syspk}, {other_cols})" self.cur.execute(create_tbl_sql) @@ -42,7 +42,7 @@ class DBConn: keys = cols.keys() # OrderedDict col_names = ', '.join( - [x for x in keys] + [f"\"{x}\"" for x in keys] ) value_placeholders = ', '.join( ["?" for _ in keys] @@ -51,7 +51,6 @@ class DBConn: values = tuple([str(cols[k]).strip() for k in keys]) sql = f"insert into {tbl} ({col_names}) values({value_placeholders})" - self.cur.execute(sql, values) def make_csv_from_tables(self, prefix=''):