chirpy123 commited on
Commit
3f5f223
·
1 Parent(s): 2908fbc

take table name from file name

Browse files
Files changed (1) hide show
  1. 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, df = get_df_from_csv('data/data.csv')
73
- schema = get_schema_from_df(df)
74
-
75
- schema = {'fields': [{'name': 'index', 'type': 'integer'}, {'name': 'Duration', 'type': 'integer'}, {'name': 'Pulse', 'type': 'integer'}, {'name': 'Maxpulse', 'type': 'integer'}, {'name': 'Calories', 'type': 'number'}], 'primaryKey': ['index'], 'pandas_version': '0.20.0'}
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)