diff options
Diffstat (limited to 'megapixels/app/models/sql_factory.py')
| -rw-r--r-- | megapixels/app/models/sql_factory.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/megapixels/app/models/sql_factory.py b/megapixels/app/models/sql_factory.py index a89d89bf..5cdaa889 100644 --- a/megapixels/app/models/sql_factory.py +++ b/megapixels/app/models/sql_factory.py @@ -62,7 +62,9 @@ def load_sql_dataset(path, replace=False, engine=None, base_model=None): df = pd.read_csv(fn) # fix columns that are named "index", a sql reserved word df.reindex_axis(sorted(df.columns), axis=1) - # df.columns = sorted(table.__table__.columns).keys() + columns = [column.name for column in table.__table__.columns] + # print(columns) + df.columns = sorted(columns) df.to_sql(name=table.__tablename__, con=engine, if_exists='replace', index=False) return dataset |