Create database.py

database.py

@@ -0,0 +1,43 @@
+from sqlalchemy import (
+    create_engine,
+    MetaData,
+    Table,
+    Column,
+    String,
+    Integer,
+    Float,
+    insert,
+)
+
+# Initialize in-memory SQLite database
+engine = create_engine("sqlite:///:memory:")
+metadata_obj = MetaData()
+
+# Define 'receipts' table
+receipts = Table(
+    "receipts",
+    metadata_obj,
+    Column("receipt_id", Integer, primary_key=True),
+    Column("customer_name", String(16), primary_key=True),
+    Column("price", Float),
+    Column("tip", Float),
+)
+
+# Create the table
+metadata_obj.create_all(engine)
+
+# 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
+rows = [
+    {"receipt_id": 1, "customer_name": "Alan Payne", "price": 12.06, "tip": 1.20},
+    {"receipt_id": 2, "customer_name": "Alex Mason", "price": 23.86, "tip": 0.24},
+    {"receipt_id": 3, "customer_name": "Woodrow Wilson", "price": 53.43, "tip": 5.43},
+    {"receipt_id": 4, "customer_name": "Margaret James", "price": 21.11, "tip": 1.00},
+]
+
+# Populate the database
+insert_rows_into_table(rows, receipts)