quote column names, so if/any wierd json keys are coming, then it won't fail
This commit is contained in:
parent
abf2e9490f
commit
8c96998c9d
5
main.py
5
main.py
@ -34,7 +34,7 @@ class DBConn:
|
|||||||
|
|
||||||
def make_table(self, tbl_name, cols):
|
def make_table(self, tbl_name, cols):
|
||||||
syspk = "syspk integer primary key autoincrement"
|
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})"
|
create_tbl_sql = f"create table if not exists {tbl_name} ({syspk}, {other_cols})"
|
||||||
self.cur.execute(create_tbl_sql)
|
self.cur.execute(create_tbl_sql)
|
||||||
|
|
||||||
@ -42,7 +42,7 @@ class DBConn:
|
|||||||
keys = cols.keys() # OrderedDict
|
keys = cols.keys() # OrderedDict
|
||||||
|
|
||||||
col_names = ', '.join(
|
col_names = ', '.join(
|
||||||
[x for x in keys]
|
[f"\"{x}\"" for x in keys]
|
||||||
)
|
)
|
||||||
value_placeholders = ', '.join(
|
value_placeholders = ', '.join(
|
||||||
["?" for _ in keys]
|
["?" for _ in keys]
|
||||||
@ -51,7 +51,6 @@ class DBConn:
|
|||||||
values = tuple([str(cols[k]).strip() for k in keys])
|
values = tuple([str(cols[k]).strip() for k in keys])
|
||||||
|
|
||||||
sql = f"insert into {tbl} ({col_names}) values({value_placeholders})"
|
sql = f"insert into {tbl} ({col_names}) values({value_placeholders})"
|
||||||
|
|
||||||
self.cur.execute(sql, values)
|
self.cur.execute(sql, values)
|
||||||
|
|
||||||
def make_csv_from_tables(self, prefix=''):
|
def make_csv_from_tables(self, prefix=''):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user