title case case study

Author: zackkrida

hardware_store/README.md

@@ -8,13 +8,14 @@ This sample dataset represents a chain of hardware stores managing their invento
 erDiagram
 %% https://mermaid.js.org/syntax/entityRelationshipDiagram.html
 
-    store_locations {
+  erDiagram
+    "Store Locations" {
         BIGINT id PK
         string name
         string address
     }
 
-    customers {
+    "Customers" {
         BIGINT id PK
         string first_name
         string last_name
@@ -23,7 +24,7 @@ erDiagram
         string address
     }
 
-    assets {
+    "Assets" {
         BIGINT id PK
         string name
         string serial_number
@@ -34,7 +35,7 @@ erDiagram
         BIGINT store_id FK
     }
 
-    transactions {
+    "Transactions" {
         BIGINT id PK
         BIGINT asset_id FK
         BIGINT customer_id FK
@@ -44,7 +45,7 @@ erDiagram
         string note
     }
 
-    rentals {
+    "Rentals" {
         BIGINT id PK
         BIGINT transaction_id FK
         TIMESTAMP rental_start
@@ -56,11 +57,10 @@ erDiagram
 
 %% Relationships
 %%  See: https://mermaid.js.org/syntax/entityRelationshipDiagram.html#relationship-syntax
-    assets ||--|{ store_locations : "store_id"
-    transactions ||--|| assets : "asset_id"
-    transactions ||--|{ customers : "customer_id"
-    rentals ||--|| transactions : "transaction_id"
-
+    "Assets" ||--|{ "Store Locations" : "store_id"
+    "Transactions" ||--|| "Assets" : "asset_id"
+    "Transactions" ||--|{ "Customers" : "customer_id"
+    "Rentals" ||--|| "Transactions" : "transaction_id"
 ```
 
 

hardware_store/generate_data.py

@@ -17,7 +17,7 @@
 # Helper function to clean values for COPY
 def clean_value(value):
     if value is None:
-        return r"\N"  # PostgreSQL NULL
+        return r"\N"
     if isinstance(value, str):
         return value.replace("\t", " ").replace("\n", " ")
     return str(value)
@@ -81,11 +81,11 @@ def generate_rentals(transaction_ids):
 transaction_ids = list(range(1, NUM_TRANSACTIONS + 1))
 
 tables = {
-    "store_locations": generate_store_locations(),
-    "customers": generate_customers(),
-    "assets": generate_assets(store_ids),
-    "transactions": generate_transactions(asset_ids, customer_ids),
-    "rentals": generate_rentals(transaction_ids),
+    "Store Locations": generate_store_locations(),
+    "Customers": generate_customers(),
+    "Assets": generate_assets(store_ids),
+    "Transactions": generate_transactions(asset_ids, customer_ids),
+    "Rentals": generate_rentals(transaction_ids),
 }
 
 # Write to SQL file
@@ -95,7 +95,8 @@ def generate_rentals(transaction_ids):
     f.write('SET search_path="Hardware Store";\n\n')
 
     for table_name, generator in tables.items():
-        f.write(f"COPY {table_name} FROM stdin;\n")
+        # Add quotes around table name since it contains spaces
+        f.write(f'COPY "{table_name}" FROM stdin;\n')
         for row in generator:
             cleaned_row = "\t".join(map(clean_value, row))
             f.write(f"{cleaned_row}\n")