Update database.py

database.py

@@ -1,82 +1,20 @@
-from sqlalchemy import (
-    create_engine,
-    MetaData,
-    Table,
-    Column,
-    String,
-    Integer,
-    Float,
-    insert,
-    text,
-)
+from sqlalchemy import create_engine, MetaData, inspect
 
-# Use a persistent SQLite database file
+# Initialize database engine (Persistent SQLite)
 engine = create_engine("sqlite:///database.db")
 metadata_obj = MetaData()
 
-# Define 'receipts' table
-receipts = Table(
-    "receipts",
-    metadata_obj,
-    Column("receipt_id", Integer, primary_key=True),
-    Column("customer_name", String(16)),
-    Column("price", Float),
-    Column("tip", Float),
-)
+# Function to check existing tables
+def get_existing_tables():
+    inspector = inspect(engine)
+    return inspector.get_table_names()
 
-# Create the table if it doesn't exist
-metadata_obj.create_all(engine)
+# Function to initialize database if empty
+def initialize_database():
+    tables = get_existing_tables()
+    if not tables:
+        print("No tables found in the database. Please upload an SQL file.")
+    else:
+        print(f"Database initialized with tables: {tables}")
 
-# Function to insert rows into the table
-def insert_rows_into_table(rows, table):
-    with engine.begin() as connection:
-        connection.execute(insert(table), rows)
-
-# Insert sample data if table is empty
-with engine.connect() as conn:
-    result = conn.execute(text("SELECT COUNT(*) FROM receipts"))
-    count = result.scalar()
-    if count == 0:
-        rows = [
-            {"receipt_id": 1, "customer_name": "Ezekiel Frost", "price": 10.50, "tip": 2.00},
-            {"receipt_id": 2, "customer_name": "Marla Dune", "price": 23.75, "tip": 4.50},
-            {"receipt_id": 3, "customer_name": "Thorne Blackwood", "price": 55.20, "tip": 6.25},
-            {"receipt_id": 4, "customer_name": "Selene Ashford", "price": 32.10, "tip": 3.50},
-            {"receipt_id": 5, "customer_name": "Vesper Hawthorne", "price": 19.80, "tip": 2.75},
-            {"receipt_id": 6, "customer_name": "Jaxon Raines", "price": 48.60, "tip": 5.00},
-            {"receipt_id": 7, "customer_name": "Liora Vale", "price": 27.45, "tip": 3.00},
-            {"receipt_id": 8, "customer_name": "Orion Finch", "price": 60.25, "tip": 7.50},
-            {"receipt_id": 9, "customer_name": "Calista Morrigan", "price": 38.90, "tip": 4.25},
-            {"receipt_id": 10, "customer_name": "Dorian Storm", "price": 50.75, "tip": 6.75},
-            {"receipt_id": 11, "customer_name": "Lucian Crowe", "price": 11.90, "tip": 1.50},
-            {"receipt_id": 12, "customer_name": "Eris Delacroix", "price": 29.95, "tip": 3.75},
-            {"receipt_id": 13, "customer_name": "Zane Evermore", "price": 45.80, "tip": 5.50},
-            {"receipt_id": 14, "customer_name": "Isolde Winter", "price": 31.20, "tip": 2.25},
-            {"receipt_id": 15, "customer_name": "Galen Hallow", "price": 18.60, "tip": 2.00},
-            {"receipt_id": 16, "customer_name": "Sable Whitmore", "price": 54.35, "tip": 6.00},
-            {"receipt_id": 17, "customer_name": "Cassian Vex", "price": 28.10, "tip": 3.25},
-            {"receipt_id": 18, "customer_name": "Selwyn Fox", "price": 61.45, "tip": 7.75},
-            {"receipt_id": 19, "customer_name": "Thalia Rune", "price": 37.20, "tip": 4.00},
-            {"receipt_id": 20, "customer_name": "Draven Nocturne", "price": 49.95, "tip": 6.25},
-            {"receipt_id": 21, "customer_name": "Kieran Wolfe", "price": 12.75, "tip": 1.75},
-            {"receipt_id": 22, "customer_name": "Ember Sterling", "price": 30.60, "tip": 3.50},
-            {"receipt_id": 23, "customer_name": "Valko Nightshade", "price": 47.90, "tip": 5.75},
-            {"receipt_id": 24, "customer_name": "Ronan Greaves", "price": 33.40, "tip": 3.00},
-            {"receipt_id": 25, "customer_name": "Lyra Voss", "price": 20.50, "tip": 2.25},
-            {"receipt_id": 26, "customer_name": "Caius Everdark", "price": 53.25, "tip": 6.50},
-            {"receipt_id": 27, "customer_name": "Eowyn Vale", "price": 26.80, "tip": 2.75},
-            {"receipt_id": 28, "customer_name": "Phoenix Ashen", "price": 59.30, "tip": 7.00},
-            {"receipt_id": 29, "customer_name": "Leif Hawthorne", "price": 36.15, "tip": 3.75},
-            {"receipt_id": 30, "customer_name": "Zephyr Sterling", "price": 48.75, "tip": 6.00},
-            {"receipt_id": 31, "customer_name": "Solene Ardent", "price": 13.90, "tip": 1.25},
-            {"receipt_id": 32, "customer_name": "Orpheus Dawn", "price": 31.95, "tip": 2.50},
-            {"receipt_id": 33, "customer_name": "Nyx Holloway", "price": 46.85, "tip": 5.25},
-            {"receipt_id": 34, "customer_name": "Rowan Nightshade", "price": 32.75, "tip": 3.00},
-            {"receipt_id": 35, "customer_name": "Dante Crowley", "price": 22.50, "tip": 2.00},
-            {"receipt_id": 36, "customer_name": "Sylas Vex", "price": 55.90, "tip": 6.75},
-            {"receipt_id": 37, "customer_name": "Isabeau Winters", "price": 27.85, "tip": 3.25},
-            {"receipt_id": 38, "customer_name": "Astra Everhart", "price": 62.10, "tip": 8.00},
-            {"receipt_id": 39, "customer_name": "Tiberius Dusk", "price": 39.60, "tip": 4.50},
-            {"receipt_id": 40, "customer_name": "Eldric Thorn", "price": 50.95, "tip": 6.25},
-        ]
-        insert_rows_into_table(rows, receipts)
+initialize_database()