take table name from file name
Browse files- persistence.py +4 -5
persistence.py
CHANGED
|
@@ -69,11 +69,10 @@ def save_data_to_table(table_name, df):
|
|
| 69 |
|
| 70 |
|
| 71 |
if __name__ == "__main__":
|
| 72 |
-
filename
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
table_name = 'test1'
|
| 77 |
|
| 78 |
create_table_from_schema(table_name, schema)
|
| 79 |
save_data_to_table(table_name, df)
|
|
|
|
| 69 |
|
| 70 |
|
| 71 |
if __name__ == "__main__":
|
| 72 |
+
filename = 'data.csv'
|
| 73 |
+
df = get_df_from_csv(filename)
|
| 74 |
+
schema = get_schema_from_df(df)
|
| 75 |
+
table_name = filename.split('.')[0]
|
|
|
|
| 76 |
|
| 77 |
create_table_from_schema(table_name, schema)
|
| 78 |
save_data_to_table(table_name, df)
|