diff --git a/main.py b/main.py index 5ef7e45..f10928f 100644 --- a/main.py +++ b/main.py @@ -44,23 +44,23 @@ class DBConn: print(f"create sql = {create_tbl_sql}") self.cur.execute(create_tbl_sql) - def write_to_database(self, tbl, cols): - keys = cols.keys() # OrderedDict + def write_to_database(self, tbl, column_data): + col_names = column_data.keys() # OrderedDict col_names = ', '.join( - [f"\"{x}\"" for x in keys] + [f"\"{x}\"" for x in col_names] ) value_placeholders = ', '.join( - ["?" for _ in keys] + ["?" for _ in col_names] ) # trim values of spaces - values = tuple([str(cols[k]).strip() for k in keys]) + values = tuple([str(column_data[k]).strip() for k in col_names]) # there might be a scenario that new cols might appear at a later stage, if so, # we shall accommodate by adding it to the database tbl_cols = self.table_mapping[tbl] not_found = [] - for col in cols: + for col in col_names: if col not in tbl_cols: not_found.append(col) @@ -69,7 +69,7 @@ class DBConn: self.cur.execute(f"alter table \"{tbl}\" add column \"{new_col}\" text") print(f"added {not_found} cols to {tbl}") - self.table_mapping[tbl] = cols + self.table_mapping[tbl] = column_data sql = f"insert into \"{tbl}\" ({col_names}) values({value_placeholders})" self.cur.execute(sql, values)