summaryrefslogtreecommitdiff
path: root/megapixels/app/models
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2019-01-12 17:38:56 +0100
committerJules Laplace <julescarbon@gmail.com>2019-01-12 17:38:56 +0100
commit59f692719bb6b4163594243e4c11262dc88466b4 (patch)
tree2cc361b8ddba46996ca9eb732fbde315ea6e4e16 /megapixels/app/models
parent0fa4c5b7d1e16859f282bdda73bb6af4f4f78b6e (diff)
fixing imports
Diffstat (limited to 'megapixels/app/models')
-rw-r--r--megapixels/app/models/sql_factory.py4
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