-
Notifications
You must be signed in to change notification settings - Fork 1
/
insert_food_baveragse
44 lines (33 loc) · 1.24 KB
/
insert_food_baveragse
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
from app import app
from models import db, FoodAndBeverage
# db_uri = 'mysql://root:@127.0.0.1/restaurant'
# app.config['SQLALCHEMY_DATABASE_URI'] = db_uri
# app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db.init_app(app)
def insert_food_data():
foods = [
'Pizza', 'Burger', 'Salad', 'Pasta', 'Sandwich',
'Steak', 'Sushi', 'Tacos', 'Fried Chicken', 'Ramen',
'Lasagna', 'Curry', 'Burrito', 'Grilled Cheese', 'Pad Thai'
]
for food in foods:
new_food = FoodAndBeverage(name=food, type='food')
db.session.add(new_food)
db.session.commit()
print("Sample food data inserted successfully!")
def insert_beverage_data():
beverages = [
'Coffee', 'Tea', 'Soda', 'Juice', 'Smoothie',
'Beer', 'Wine', 'Cocktail', 'Lemonade', 'Milkshake',
'Iced Tea', 'Hot Chocolate', 'Limeade', 'Iced Coffee', 'Margarita'
]
for beverage in beverages:
new_beverage = FoodAndBeverage(name=beverage, type='beverage')
db.session.add(new_beverage)
db.session.commit()
print("Sample beverage data inserted successfully!")
if __name__ == '__main__':
with app.app_context():
# Insert sample data
insert_food_data()
insert_beverage_data()