Commit
This commit is contained in:
Joan
410
old/migrate_account_player_separation.py
Normal file
410
old/migrate_account_player_separation.py
Normal file
@@ -0,0 +1,410 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Major Database Migration: Account/Player Separation
|
||||
====================================================
|
||||
|
||||
This migration separates authentication (accounts) from gameplay (characters):
|
||||
- Creates new 'accounts' table for login credentials
|
||||
- Creates new 'characters' table for game data
|
||||
- Migrates existing 'players' data to both tables
|
||||
- Updates foreign keys in related tables
|
||||
- Drops old 'players' table
|
||||
|
||||
IMPORTANT: This is a breaking change. Backup your database first!
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import asyncpg
|
||||
import os
|
||||
from datetime import datetime
|
||||
|
||||
# Database connection
|
||||
DB_USER = os.getenv("POSTGRES_USER")
|
||||
DB_PASS = os.getenv("POSTGRES_PASSWORD")
|
||||
DB_NAME = os.getenv("POSTGRES_DB")
|
||||
DB_HOST = os.getenv("POSTGRES_HOST", "echoes_of_the_ashes_db")
|
||||
DB_PORT = os.getenv("POSTGRES_PORT", "5432")
|
||||
|
||||
DATABASE_URL = f"postgresql://{DB_USER}:{DB_PASS}@{DB_HOST}:{DB_PORT}/{DB_NAME}"
|
||||
|
||||
|
||||
async def main():
|
||||
print("=" * 70)
|
||||
print("ACCOUNT/PLAYER SEPARATION MIGRATION")
|
||||
print("=" * 70)
|
||||
print()
|
||||
|
||||
conn = await asyncpg.connect(DATABASE_URL)
|
||||
|
||||
try:
|
||||
# Step 0: Check if migration already ran
|
||||
print("Step 0: Checking migration status...")
|
||||
tables_exist = await conn.fetchval("""
|
||||
SELECT EXISTS (
|
||||
SELECT FROM information_schema.tables
|
||||
WHERE table_name = 'accounts'
|
||||
)
|
||||
""")
|
||||
|
||||
if tables_exist:
|
||||
print("⚠️ Accounts table already exists. Migration may have already run.")
|
||||
print(" Cleaning up previous migration attempt...")
|
||||
await conn.execute("DROP TABLE IF EXISTS characters CASCADE;")
|
||||
await conn.execute("DROP TABLE IF EXISTS accounts CASCADE;")
|
||||
await conn.execute("DROP TABLE IF EXISTS players_backup_20251109 CASCADE;")
|
||||
print("✅ Cleaned up existing tables")
|
||||
print()
|
||||
|
||||
# Step 1: Backup existing players table
|
||||
print("Step 1: Creating backup of players table...")
|
||||
await conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS players_backup_20251109 AS
|
||||
SELECT * FROM players;
|
||||
""")
|
||||
backup_count = await conn.fetchval(
|
||||
"SELECT COUNT(*) FROM players_backup_20251109"
|
||||
)
|
||||
print(f"✅ Backed up {backup_count} players to players_backup_20251109")
|
||||
print()
|
||||
|
||||
# Step 2: Create temporary mapping table
|
||||
print("Step 2: Creating temporary mapping table...")
|
||||
await conn.execute("""
|
||||
CREATE TEMP TABLE IF NOT EXISTS player_character_mapping (
|
||||
old_player_id INTEGER,
|
||||
new_character_id INTEGER
|
||||
);
|
||||
""")
|
||||
print("✅ Created temporary mapping table")
|
||||
print()
|
||||
|
||||
# Step 3: Create accounts table
|
||||
print("Step 3: Creating accounts table...")
|
||||
await conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS accounts (
|
||||
id SERIAL PRIMARY KEY,
|
||||
email VARCHAR(255) UNIQUE NOT NULL,
|
||||
password_hash VARCHAR(255),
|
||||
steam_id VARCHAR(255) UNIQUE,
|
||||
account_type VARCHAR(20) DEFAULT 'web',
|
||||
premium_expires_at REAL,
|
||||
email_verified BOOLEAN DEFAULT FALSE,
|
||||
email_verification_token VARCHAR(255),
|
||||
password_reset_token VARCHAR(255),
|
||||
password_reset_expires REAL,
|
||||
created_at REAL DEFAULT EXTRACT(EPOCH FROM NOW()),
|
||||
last_login_at REAL,
|
||||
CONSTRAINT check_account_type CHECK (account_type IN ('web', 'steam'))
|
||||
);
|
||||
""")
|
||||
print("✅ Created accounts table")
|
||||
|
||||
# Create indexes for accounts
|
||||
await conn.execute("""
|
||||
CREATE INDEX IF NOT EXISTS idx_accounts_email ON accounts(email);
|
||||
""")
|
||||
await conn.execute("""
|
||||
CREATE INDEX IF NOT EXISTS idx_accounts_steam_id ON accounts(steam_id);
|
||||
""")
|
||||
print("✅ Created indexes on accounts table")
|
||||
print()
|
||||
|
||||
# Step 4: Create characters table
|
||||
print("Step 4: Creating characters table...")
|
||||
await conn.execute("""
|
||||
CREATE TABLE IF NOT EXISTS characters (
|
||||
id SERIAL PRIMARY KEY,
|
||||
account_id INTEGER NOT NULL REFERENCES accounts(id) ON DELETE CASCADE,
|
||||
name VARCHAR(100) UNIQUE NOT NULL,
|
||||
avatar_data TEXT,
|
||||
|
||||
-- RPG Stats
|
||||
level INTEGER DEFAULT 1,
|
||||
xp INTEGER DEFAULT 0,
|
||||
hp INTEGER DEFAULT 100,
|
||||
max_hp INTEGER DEFAULT 100,
|
||||
stamina INTEGER DEFAULT 100,
|
||||
max_stamina INTEGER DEFAULT 100,
|
||||
|
||||
-- Base Attributes
|
||||
strength INTEGER DEFAULT 0,
|
||||
agility INTEGER DEFAULT 0,
|
||||
endurance INTEGER DEFAULT 0,
|
||||
intellect INTEGER DEFAULT 0,
|
||||
unspent_points INTEGER DEFAULT 0,
|
||||
|
||||
-- Game State
|
||||
location_id VARCHAR(255) DEFAULT 'cabin',
|
||||
is_dead BOOLEAN DEFAULT FALSE,
|
||||
last_movement_time REAL DEFAULT 0,
|
||||
|
||||
-- Timestamps
|
||||
created_at REAL DEFAULT EXTRACT(EPOCH FROM NOW()),
|
||||
last_played_at REAL DEFAULT EXTRACT(EPOCH FROM NOW()),
|
||||
|
||||
CONSTRAINT check_unspent_points CHECK (unspent_points >= 0)
|
||||
);
|
||||
""")
|
||||
print("✅ Created characters table")
|
||||
|
||||
# Create indexes for characters
|
||||
await conn.execute("""
|
||||
CREATE INDEX IF NOT EXISTS idx_characters_account_id ON characters(account_id);
|
||||
""")
|
||||
await conn.execute("""
|
||||
CREATE INDEX IF NOT EXISTS idx_characters_name ON characters(name);
|
||||
""")
|
||||
await conn.execute("""
|
||||
CREATE INDEX IF NOT EXISTS idx_characters_location_id ON characters(location_id);
|
||||
""")
|
||||
print("✅ Created indexes on characters table")
|
||||
print()
|
||||
|
||||
# Step 5: Migrate existing players to accounts and characters
|
||||
print("Step 5: Migrating existing players...")
|
||||
|
||||
# Get all existing players
|
||||
players = await conn.fetch("SELECT * FROM players ORDER BY id")
|
||||
print(f"Found {len(players)} players to migrate")
|
||||
|
||||
migrated = 0
|
||||
character_names_used = set()
|
||||
|
||||
for player in players:
|
||||
# Generate email if not present
|
||||
email = player['email']
|
||||
if not email:
|
||||
username = player['username'] or f"player_{player['id']}"
|
||||
email = f"{username}@echoes-migrated.local"
|
||||
|
||||
# Ensure unique character name
|
||||
char_name = player['name']
|
||||
if char_name in character_names_used or char_name == "Survivor":
|
||||
# Make it unique
|
||||
char_name = f"{player['username'] or 'Survivor'}_{player['id']}"
|
||||
character_names_used.add(char_name)
|
||||
|
||||
# Convert to timestamp (float)
|
||||
now_timestamp = datetime.utcnow().timestamp()
|
||||
|
||||
# Create account
|
||||
account_id = await conn.fetchval("""
|
||||
INSERT INTO accounts (
|
||||
email, password_hash, steam_id, account_type,
|
||||
premium_expires_at, created_at, last_login_at
|
||||
)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
RETURNING id
|
||||
""",
|
||||
email,
|
||||
player['password_hash'],
|
||||
player.get('steam_id'),
|
||||
player.get('account_type', 'web'),
|
||||
player.get('premium_expires_at'), # Keep as is (NULL or timestamp)
|
||||
now_timestamp,
|
||||
now_timestamp
|
||||
)
|
||||
|
||||
# Create character from player data
|
||||
character_id = await conn.fetchval("""
|
||||
INSERT INTO characters (
|
||||
account_id, name, avatar_data,
|
||||
level, xp, hp, max_hp, stamina, max_stamina,
|
||||
strength, agility, endurance, intellect, unspent_points,
|
||||
location_id, is_dead, last_movement_time,
|
||||
created_at, last_played_at
|
||||
)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19)
|
||||
RETURNING id
|
||||
""",
|
||||
account_id,
|
||||
char_name, # Use unique character name
|
||||
None, # avatar_data
|
||||
player['level'],
|
||||
player['xp'],
|
||||
player['hp'],
|
||||
player['max_hp'],
|
||||
player['stamina'],
|
||||
player['max_stamina'],
|
||||
player['strength'],
|
||||
player['agility'],
|
||||
player['endurance'],
|
||||
player['intellect'],
|
||||
player['unspent_points'],
|
||||
player['location_id'],
|
||||
player['is_dead'],
|
||||
player['last_movement_time'],
|
||||
now_timestamp,
|
||||
now_timestamp
|
||||
)
|
||||
|
||||
# Store mapping for foreign key updates
|
||||
await conn.execute("""
|
||||
INSERT INTO player_character_mapping (old_player_id, new_character_id)
|
||||
VALUES ($1, $2)
|
||||
""", player['id'], character_id)
|
||||
|
||||
migrated += 1
|
||||
if migrated % 10 == 0:
|
||||
print(f" Migrated {migrated}/{len(players)} players...")
|
||||
|
||||
print("✅ Migrated {migrated} players to accounts and characters")
|
||||
print()
|
||||
|
||||
# Step 6: Update foreign keys in related tables
|
||||
print("Step 6: Updating foreign keys in related tables...")
|
||||
|
||||
# Update inventory table
|
||||
if await conn.fetchval("""
|
||||
SELECT EXISTS (
|
||||
SELECT FROM information_schema.tables
|
||||
WHERE table_name = 'inventory'
|
||||
)
|
||||
"""):
|
||||
print(" Updating inventory.player_id -> character_id...")
|
||||
|
||||
# Add new character_id column
|
||||
await conn.execute("""
|
||||
ALTER TABLE inventory
|
||||
ADD COLUMN IF NOT EXISTS character_id INTEGER;
|
||||
""")
|
||||
|
||||
# Copy player_id to character_id using mapping
|
||||
await conn.execute("""
|
||||
UPDATE inventory i
|
||||
SET character_id = m.new_character_id
|
||||
FROM player_character_mapping m
|
||||
WHERE i.player_id = m.old_player_id;
|
||||
""")
|
||||
|
||||
# Drop old player_id column and rename
|
||||
await conn.execute("""
|
||||
ALTER TABLE inventory DROP COLUMN IF EXISTS player_id;
|
||||
""")
|
||||
|
||||
# Add foreign key constraint
|
||||
await conn.execute("""
|
||||
ALTER TABLE inventory
|
||||
ADD CONSTRAINT fk_inventory_character
|
||||
FOREIGN KEY (character_id) REFERENCES characters(id) ON DELETE CASCADE;
|
||||
""")
|
||||
|
||||
print(" ✅ Updated inventory table")
|
||||
|
||||
# Update equipment table
|
||||
if await conn.fetchval("""
|
||||
SELECT EXISTS (
|
||||
SELECT FROM information_schema.tables
|
||||
WHERE table_name = 'equipment'
|
||||
)
|
||||
"""):
|
||||
print(" Updating equipment.player_id -> character_id...")
|
||||
|
||||
await conn.execute("""
|
||||
ALTER TABLE equipment
|
||||
ADD COLUMN IF NOT EXISTS character_id INTEGER;
|
||||
""")
|
||||
|
||||
await conn.execute("""
|
||||
UPDATE equipment e
|
||||
SET character_id = m.new_character_id
|
||||
FROM player_character_mapping m
|
||||
WHERE e.player_id = m.old_player_id;
|
||||
""")
|
||||
|
||||
await conn.execute("""
|
||||
ALTER TABLE equipment DROP COLUMN IF EXISTS player_id;
|
||||
""")
|
||||
|
||||
await conn.execute("""
|
||||
ALTER TABLE equipment
|
||||
ADD CONSTRAINT fk_equipment_character
|
||||
FOREIGN KEY (character_id) REFERENCES characters(id) ON DELETE CASCADE;
|
||||
""")
|
||||
|
||||
print(" ✅ Updated equipment table")
|
||||
|
||||
# Update dropped_items table
|
||||
if await conn.fetchval("""
|
||||
SELECT EXISTS (
|
||||
SELECT FROM information_schema.tables
|
||||
WHERE table_name = 'dropped_items'
|
||||
)
|
||||
"""):
|
||||
# Check if column exists
|
||||
has_player_col = await conn.fetchval("""
|
||||
SELECT EXISTS (
|
||||
SELECT FROM information_schema.columns
|
||||
WHERE table_name = 'dropped_items'
|
||||
AND column_name = 'dropped_by_player_id'
|
||||
)
|
||||
""")
|
||||
|
||||
if has_player_col:
|
||||
print(" Updating dropped_items.dropped_by_player_id -> dropped_by_character_id...")
|
||||
|
||||
await conn.execute("""
|
||||
ALTER TABLE dropped_items
|
||||
ADD COLUMN IF NOT EXISTS dropped_by_character_id INTEGER;
|
||||
""")
|
||||
|
||||
await conn.execute("""
|
||||
UPDATE dropped_items d
|
||||
SET dropped_by_character_id = m.new_character_id
|
||||
FROM player_character_mapping m
|
||||
WHERE d.dropped_by_player_id = m.old_player_id;
|
||||
""")
|
||||
|
||||
await conn.execute("""
|
||||
ALTER TABLE dropped_items DROP COLUMN IF EXISTS dropped_by_player_id;
|
||||
""")
|
||||
|
||||
print(" ✅ Updated dropped_items table")
|
||||
else:
|
||||
print(" ⏭️ Skipping dropped_items (no dropped_by_player_id column)")
|
||||
|
||||
print("✅ Updated all foreign key references")
|
||||
print()
|
||||
|
||||
# Step 7: Drop old players table
|
||||
print("Step 7: Dropping old players table...")
|
||||
print("⚠️ WARNING: About to drop players table (backup exists as players_backup_20251109)")
|
||||
print(" Press Ctrl+C within 5 seconds to cancel...")
|
||||
await asyncio.sleep(5)
|
||||
|
||||
await conn.execute("DROP TABLE IF EXISTS players;")
|
||||
print("✅ Dropped players table")
|
||||
print()
|
||||
|
||||
# Step 8: Summary
|
||||
print("=" * 70)
|
||||
print("MIGRATION COMPLETED SUCCESSFULLY!")
|
||||
print("=" * 70)
|
||||
|
||||
account_count = await conn.fetchval("SELECT COUNT(*) FROM accounts")
|
||||
character_count = await conn.fetchval("SELECT COUNT(*) FROM characters")
|
||||
|
||||
print(f"✅ Created {account_count} accounts")
|
||||
print(f"✅ Created {character_count} characters")
|
||||
print(f"✅ Backup preserved in: players_backup_20251109")
|
||||
print()
|
||||
print("Next steps:")
|
||||
print("1. Update application code to use new schema")
|
||||
print("2. Rebuild and restart API container")
|
||||
print("3. Test authentication and character selection")
|
||||
print("4. Update frontend to show character selection screen")
|
||||
print()
|
||||
|
||||
except Exception as e:
|
||||
print(f"\n❌ ERROR: {e}")
|
||||
print("\nRolling back changes...")
|
||||
await conn.execute("ROLLBACK;")
|
||||
print("Migration failed. Database unchanged.")
|
||||
raise
|
||||
|
||||
finally:
|
||||
await conn.close()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user