def main(): if not engine.dialect.has_table(engine, 'wine_attributes'): Base.metadata.create_all(engine) try: file_name = "wine_attributes.csv" data = load_data(file_name) for i in data: record = WineAttribute( **{ 'japanese_title': i[1], 'english_title': i[2], 'step': float(i[3]), 'max_value': float(i[4]), 'min_value': float(i[5]) }) session.add(record) session.commit() except Exception as e: print(e) session.rollback() finally: session.close()
def get_all(): wine_attributes = [] try: wine_attributes = session.query(WineAttribute).all() except Exception as e: print(e) finally: session.close() return WineAttributeSchema().dump(wine_attributes, many=True)