diff --git a/lib/models/Drink_Orders.py b/lib/models/Drink_Orders.py index 967e418b7..5d5722a4c 100644 --- a/lib/models/Drink_Orders.py +++ b/lib/models/Drink_Orders.py @@ -1,15 +1,25 @@ -from models.Drinks import Drinks -from models.Customer import Customer +from __init__ import CONN, CURSOR -class DrinkOrder: + +class Drink_Orders(): def __init__(self): pass - def age_verification(): - #Test for age requirements here, build on it later - if customer.age < drinks.age_requirement: - print("Can't Drink") - raise ValueError("Not Old Enough For Alcohol") - else: - print("Party On!") + @classmethod + def customer_drinks(cls): + sql = ''' + customer_name TEXT, + customer_id INTEGER, + drink_name TEXT, + drink_id INTEGER, + PRIMARY KEY (customer_name, customer_id, drink_name, drink_id), + FOREIGN KEY (customer_name) REFERENCES customer(name), + FOREIGN KEY (customer_id) REFERENCES customer(id), + FOREIGN KEY (drink_name) REFERENCES drink(name), + FOREIGN KEY (drink_id) REFERENCES drinks(id); + ''' + CURSOR.execute(sql) + CONN.commit() + + diff --git a/lib/models/seed.py b/lib/models/seed.py index 22d0ea773..868694881 100644 --- a/lib/models/seed.py +++ b/lib/models/seed.py @@ -9,6 +9,7 @@ # Drop the tables if they exist cursor.execute('DROP TABLE IF EXISTS customer') cursor.execute('DROP TABLE IF EXISTS drinks') +cursor.execute('DROP TABLE IF EXISTS drink_orders') # Create tables cursor.execute(''' @@ -26,6 +27,20 @@ ) ''') +cursor.execute(''' +CREATE TABLE IF NOT EXISTS drink_orders ( + customer_name TEXT, + customer_id INTEGER, + drink_name TEXT, + drink_id INTEGER, + PRIMARY KEY (customer_name, customer_id, drink_name, drink_id), + FOREIGN KEY (customer_name) REFERENCES customer(name), + FOREIGN KEY (customer_id) REFERENCES customer(id), + FOREIGN KEY (drink_name) REFERENCES drink(name), + FOREIGN KEY (drink_id) REFERENCES drinks(id) +) +''') + # Function to insert data into the customer and drinks tables def seed_db(): # Sample user data @@ -44,6 +59,16 @@ def seed_db(): ('Bees Knees',) ] + drink_orders = [ + ('Jay', 1, 'Cosmo', 1) # Jay ordered Cosmo + # (1, 3), # Jay ordered Tequila Sunrise + # (2, 2), # Victoria ordered Manhattan + # (3, 5), # Kerissa ordered Bees Knees + # (4, 4), # Madison ordered Rum Runner + # (5, 1), # Johnathan ordered Cosmo + # (5, 2) # Johnathan ordered Manhattan +] + # Insert sample customers into the database cursor.executemany(''' INSERT INTO customer (name, age) @@ -55,12 +80,18 @@ def seed_db(): INSERT INTO drinks (name) VALUES (?) ''', drinks) + + cursor.executemany(''' + INSERT INTO drink_orders (customer_name, customer_id, drink_name, drink_id) + VALUES (?, ?, ?, ?) + ''', drink_orders) # Commit the transaction conn.commit() print(f"{len(customers)} customers added to the database.") print(f"{len(drinks)} drinks added to the database.") + print(f"{len(customers)} customers, {len(drinks)} drinks, and {len(drink_orders)} drink-orders associations were added to the database.") # Run the seed function if __name__ == "__main__":