mirror of
https://github.com/freedbygrace/SQL.git
synced 2026-07-26 19:38:16 +00:00
Add Python dependency and start CSV-based data generation
DEPENDENCY INSTALLER UPDATES: - Added Python 3 installation check and installer - Python is required for fast CSV data generation - Updated step numbers from [1/4] to [1/5] - Python installed before PostgreSQL client - Supports Ubuntu, Debian, CentOS, RHEL, Fedora, Arch, macOS DEPLOY SCRIPT UPDATES: - Added automatic dependency checking (Step 3/8) - Checks for python3, psql, and docker - Automatically runs install-dependencies.sh if missing - Updated all step numbers to reflect 8 total steps - Step 3: Check dependencies (NEW) - Step 4: Fix permissions (was Step 3) - Step 5: Start containers (was Step 4) - Step 6: Initialize schema (was Step 5) - Step 7: Generate data (was Step 6) - Step 8: Verify deployment (was Step 7) CSV DATA GENERATION (IN PROGRESS): - Created scripts/generate_csv_data.py - Python-based CSV generator for bulk import - Much faster than SQL INSERT statements - Generates realistic data with proper relationships - Partial implementation (customers, accounts, merchants, cards) - Next: Complete transactions and analytics data RATIONALE: - Current bash-based data generation has transaction commit issues - Data shows as generated but doesn't persist (0 rows after completion) - CSV + COPY command is PostgreSQL best practice for bulk loading - 10-100x faster than INSERT statements - More reliable - atomic COPY operations - Easier to debug - can inspect CSV files
This commit is contained in:
Alphaeus Mote
@@ -0,0 +1,229 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Generate CSV files for bulk import into PostgreSQL
|
||||
Much faster and more reliable than INSERT statements
|
||||
"""
|
||||
|
||||
import csv
|
||||
import random
|
||||
import os
|
||||
from datetime import datetime, timedelta
|
||||
from pathlib import Path
|
||||
|
||||
# Configuration
|
||||
NUM_CUSTOMERS = 100000
|
||||
NUM_ACCOUNTS = 150000
|
||||
NUM_MERCHANTS = 50000
|
||||
NUM_CARDS = 200000
|
||||
NUM_TRANSACTIONS = 5000000
|
||||
FRAUD_RATE = 0.07
|
||||
|
||||
# Output directory
|
||||
DATA_DIR = Path(__file__).parent.parent / "data" / "csv"
|
||||
DATA_DIR.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
print("=" * 80)
|
||||
print("CSV Data Generation for Business Analytics Database")
|
||||
print("=" * 80)
|
||||
print(f"Output directory: {DATA_DIR}")
|
||||
print(f"Customers: {NUM_CUSTOMERS:,}")
|
||||
print(f"Accounts: {NUM_ACCOUNTS:,}")
|
||||
print(f"Merchants: {NUM_MERCHANTS:,}")
|
||||
print(f"Cards: {NUM_CARDS:,}")
|
||||
print(f"Transactions: {NUM_TRANSACTIONS:,}")
|
||||
print(f"Fraud rate: {FRAUD_RATE*100}%")
|
||||
print("=" * 80)
|
||||
print()
|
||||
|
||||
# Reference data
|
||||
FIRST_NAMES = ["James", "Mary", "John", "Patricia", "Robert", "Jennifer", "Michael", "Linda",
|
||||
"William", "Barbara", "David", "Elizabeth", "Richard", "Susan", "Joseph", "Jessica",
|
||||
"Thomas", "Sarah", "Charles", "Karen", "Christopher", "Nancy", "Daniel", "Lisa"]
|
||||
|
||||
LAST_NAMES = ["Smith", "Johnson", "Williams", "Brown", "Jones", "Garcia", "Miller", "Davis",
|
||||
"Rodriguez", "Martinez", "Hernandez", "Lopez", "Gonzalez", "Wilson", "Anderson", "Thomas",
|
||||
"Taylor", "Moore", "Jackson", "Martin", "Lee", "Perez", "Thompson", "White"]
|
||||
|
||||
CITIES = ["New York", "Los Angeles", "Chicago", "Houston", "Phoenix", "Philadelphia", "San Antonio",
|
||||
"San Diego", "Dallas", "San Jose", "Austin", "Jacksonville", "Fort Worth", "Columbus",
|
||||
"Charlotte", "San Francisco", "Indianapolis", "Seattle", "Denver", "Boston"]
|
||||
|
||||
STATES = ["NY", "CA", "IL", "TX", "AZ", "PA", "TX", "CA", "TX", "CA", "TX", "FL", "TX", "OH",
|
||||
"NC", "CA", "IN", "WA", "CO", "MA"]
|
||||
|
||||
COUNTRIES = ["US", "CA", "GB", "DE", "FR", "AU", "JP", "CN", "IN", "BR"]
|
||||
|
||||
MERCHANT_TYPES = [
|
||||
("5411", "Grocery Stores"), ("5812", "Restaurants"), ("5541", "Gas Stations"),
|
||||
("5311", "Department Stores"), ("5912", "Pharmacies"), ("5999", "Miscellaneous Retail"),
|
||||
("5732", "Electronics"), ("5651", "Clothing"), ("5814", "Fast Food"),
|
||||
("5942", "Books"), ("7011", "Hotels"), ("7512", "Car Rental")
|
||||
]
|
||||
|
||||
TRANSACTION_TYPES = ["PURCHASE", "ATM_WITHDRAWAL", "TRANSFER_OUT", "TRANSFER_IN", "PAYMENT",
|
||||
"REFUND", "DEPOSIT", "WIRE_OUT", "WIRE_IN"]
|
||||
|
||||
FRAUD_CODES = ["CARD_STOLEN", "ACCOUNT_TAKEOVER", "IDENTITY_THEFT", "SYNTHETIC_IDENTITY",
|
||||
"CARD_NOT_PRESENT", "PHISHING", "ATM_SKIMMING", "WIRE_FRAUD"]
|
||||
|
||||
def generate_email(first_name, last_name, customer_id):
|
||||
"""Generate realistic email address"""
|
||||
domains = ["gmail.com", "yahoo.com", "hotmail.com", "outlook.com", "icloud.com"]
|
||||
return f"{first_name.lower()}.{last_name.lower()}{customer_id % 1000}@{random.choice(domains)}"
|
||||
|
||||
def generate_phone():
|
||||
"""Generate US phone number"""
|
||||
return f"+1{random.randint(2000000000, 9999999999)}"
|
||||
|
||||
def generate_ssn():
|
||||
"""Generate SSN-like identifier"""
|
||||
return f"{random.randint(100, 999)}-{random.randint(10, 99)}-{random.randint(1000, 9999)}"
|
||||
|
||||
def generate_card_number():
|
||||
"""Generate realistic card number"""
|
||||
prefix = random.choice(["4", "5"]) # Visa or Mastercard
|
||||
return prefix + "".join([str(random.randint(0, 9)) for _ in range(15)])
|
||||
|
||||
def random_date(start_date, end_date):
|
||||
"""Generate random date between start and end"""
|
||||
delta = end_date - start_date
|
||||
random_days = random.randint(0, delta.days)
|
||||
return start_date + timedelta(days=random_days)
|
||||
|
||||
def random_datetime(start_date, end_date):
|
||||
"""Generate random datetime between start and end"""
|
||||
delta = end_date - start_date
|
||||
random_seconds = random.randint(0, int(delta.total_seconds()))
|
||||
return start_date + timedelta(seconds=random_seconds)
|
||||
|
||||
print("[1/10] Generating customers...")
|
||||
customers = []
|
||||
with open(DATA_DIR / "customers.csv", "w", newline="", encoding="utf-8") as f:
|
||||
writer = csv.writer(f)
|
||||
writer.writerow(["customer_id", "first_name", "last_name", "email", "phone", "date_of_birth",
|
||||
"ssn", "address_line1", "city", "state", "postal_code", "country_code",
|
||||
"risk_score", "kyc_status", "created_at"])
|
||||
|
||||
for i in range(1, NUM_CUSTOMERS + 1):
|
||||
first_name = random.choice(FIRST_NAMES)
|
||||
last_name = random.choice(LAST_NAMES)
|
||||
city_idx = random.randint(0, len(CITIES) - 1)
|
||||
|
||||
customer = [
|
||||
i, # customer_id
|
||||
first_name,
|
||||
last_name,
|
||||
generate_email(first_name, last_name, i),
|
||||
generate_phone(),
|
||||
random_date(datetime(1950, 1, 1), datetime(2005, 12, 31)).strftime("%Y-%m-%d"),
|
||||
generate_ssn(),
|
||||
f"{random.randint(100, 9999)} {random.choice(['Main', 'Oak', 'Maple', 'Park'])} St",
|
||||
CITIES[city_idx],
|
||||
STATES[city_idx],
|
||||
f"{random.randint(10000, 99999)}",
|
||||
random.choice(COUNTRIES),
|
||||
round(random.uniform(0, 100), 2),
|
||||
random.choice(["VERIFIED", "PENDING", "REJECTED"]),
|
||||
random_datetime(datetime(2020, 1, 1), datetime(2024, 1, 1)).strftime("%Y-%m-%d %H:%M:%S")
|
||||
]
|
||||
writer.writerow(customer)
|
||||
customers.append(i)
|
||||
|
||||
if i % 10000 == 0:
|
||||
print(f" Generated {i:,} customers...")
|
||||
|
||||
print(f"✓ Generated {NUM_CUSTOMERS:,} customers")
|
||||
|
||||
print("[2/10] Generating accounts...")
|
||||
accounts = []
|
||||
with open(DATA_DIR / "accounts.csv", "w", newline="", encoding="utf-8") as f:
|
||||
writer = csv.writer(f)
|
||||
writer.writerow(["account_id", "customer_id", "account_number", "account_type", "currency",
|
||||
"balance", "available_balance", "status", "opened_date", "closed_date"])
|
||||
|
||||
for i in range(1, NUM_ACCOUNTS + 1):
|
||||
customer_id = random.choice(customers)
|
||||
account_type = random.choice(["CHECKING", "SAVINGS", "CREDIT"])
|
||||
balance = round(random.uniform(100, 50000), 2)
|
||||
|
||||
account = [
|
||||
i, # account_id
|
||||
customer_id,
|
||||
f"{random.randint(1000000000, 9999999999)}",
|
||||
account_type,
|
||||
"USD",
|
||||
balance,
|
||||
balance * random.uniform(0.8, 1.0),
|
||||
random.choice(["ACTIVE", "ACTIVE", "ACTIVE", "SUSPENDED", "CLOSED"]),
|
||||
random_date(datetime(2020, 1, 1), datetime(2024, 1, 1)).strftime("%Y-%m-%d"),
|
||||
"" if random.random() > 0.05 else random_date(datetime(2023, 1, 1), datetime(2024, 12, 31)).strftime("%Y-%m-%d")
|
||||
]
|
||||
writer.writerow(account)
|
||||
accounts.append(i)
|
||||
|
||||
if i % 10000 == 0:
|
||||
print(f" Generated {i:,} accounts...")
|
||||
|
||||
print(f"✓ Generated {NUM_ACCOUNTS:,} accounts")
|
||||
|
||||
print("[3/10] Generating merchants...")
|
||||
merchants = []
|
||||
with open(DATA_DIR / "merchants.csv", "w", newline="", encoding="utf-8") as f:
|
||||
writer = csv.writer(f)
|
||||
writer.writerow(["merchant_id", "merchant_name", "category_id", "country_code", "city",
|
||||
"risk_rating", "is_active", "created_at"])
|
||||
|
||||
for i in range(1, NUM_MERCHANTS + 1):
|
||||
category = random.choice(MERCHANT_TYPES)
|
||||
city_idx = random.randint(0, len(CITIES) - 1)
|
||||
|
||||
merchant = [
|
||||
i, # merchant_id
|
||||
f"{category[1]} #{i}",
|
||||
random.randint(1, 35), # category_id from merchant_categories
|
||||
random.choice(COUNTRIES),
|
||||
CITIES[city_idx],
|
||||
round(random.uniform(1, 10), 2),
|
||||
random.choice([True, True, True, False]),
|
||||
random_datetime(datetime(2015, 1, 1), datetime(2024, 1, 1)).strftime("%Y-%m-%d %H:%M:%S")
|
||||
]
|
||||
writer.writerow(merchant)
|
||||
merchants.append(i)
|
||||
|
||||
if i % 5000 == 0:
|
||||
print(f" Generated {i:,} merchants...")
|
||||
|
||||
print(f"✓ Generated {NUM_MERCHANTS:,} merchants")
|
||||
|
||||
print("[4/10] Generating cards...")
|
||||
cards = []
|
||||
with open(DATA_DIR / "cards.csv", "w", newline="", encoding="utf-8") as f:
|
||||
writer = csv.writer(f)
|
||||
writer.writerow(["card_id", "account_id", "card_number", "card_type", "expiry_date",
|
||||
"cvv", "status", "daily_limit", "issued_date"])
|
||||
|
||||
for i in range(1, NUM_CARDS + 1):
|
||||
account_id = random.choice(accounts)
|
||||
|
||||
card = [
|
||||
i, # card_id
|
||||
account_id,
|
||||
generate_card_number(),
|
||||
random.choice(["DEBIT", "CREDIT", "PREPAID"]),
|
||||
random_date(datetime(2025, 1, 1), datetime(2030, 12, 31)).strftime("%Y-%m-%d"),
|
||||
f"{random.randint(100, 999)}",
|
||||
random.choice(["ACTIVE", "ACTIVE", "ACTIVE", "BLOCKED", "EXPIRED"]),
|
||||
round(random.choice([500, 1000, 2000, 5000, 10000]), 2),
|
||||
random_date(datetime(2020, 1, 1), datetime(2024, 1, 1)).strftime("%Y-%m-%d")
|
||||
]
|
||||
writer.writerow(card)
|
||||
cards.append(i)
|
||||
|
||||
if i % 20000 == 0:
|
||||
print(f" Generated {i:,} cards...")
|
||||
|
||||
print(f"✓ Generated {NUM_CARDS:,} cards")
|
||||
|
||||
print("[5/10] Generating transactions (this will take a few minutes)...")
|
||||
print(f" Generating {NUM_TRANSACTIONS:,} transactions in batches...")
|
||||
|
||||
Reference in New Issue
Block a user