52244c3ca5
- Change from in-memory database to file-based SQLite at coordinator.db - Remove create_db_and_tables() call as tables already exist - Use same database path as coordinator-api for consistency - Apply database path fix to both cleanup_fake_gpus() and show_remaining_gpus()
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Direct database cleanup for fake GPU entries
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
sys.path.insert(0, '/home/oib/windsurf/aitbc/apps/coordinator-api/src')
|
|
|
|
from sqlmodel import Session, select
|
|
from sqlalchemy import create_engine
|
|
from app.domain.gpu_marketplace import GPURegistry
|
|
|
|
def cleanup_fake_gpus():
|
|
"""Clean up fake GPU entries from database"""
|
|
print("=== DIRECT DATABASE CLEANUP ===")
|
|
|
|
# Use the same database as coordinator
|
|
db_path = "/home/oib/windsurf/aitbc/apps/coordinator-api/data/coordinator.db"
|
|
engine = create_engine(f"sqlite:///{db_path}")
|
|
|
|
fake_gpus = [
|
|
"gpu_1bdf8e86",
|
|
"gpu_1b7da9e0",
|
|
"gpu_9cff5bc2",
|
|
"gpu_ebef80a5",
|
|
"gpu_979b24b8",
|
|
"gpu_e5ab817d"
|
|
]
|
|
|
|
with Session(engine) as session:
|
|
deleted_count = 0
|
|
|
|
for gpu_id in fake_gpus:
|
|
gpu = session.exec(select(GPURegistry).where(GPURegistry.id == gpu_id)).first()
|
|
if gpu:
|
|
print(f"🗑️ Deleting fake GPU: {gpu_id} - {gpu.model}")
|
|
session.delete(gpu)
|
|
deleted_count += 1
|
|
else:
|
|
print(f"❓ GPU not found: {gpu_id}")
|
|
|
|
try:
|
|
session.commit()
|
|
print(f"✅ Successfully deleted {deleted_count} fake GPUs")
|
|
except Exception as e:
|
|
print(f"❌ Error committing changes: {e}")
|
|
session.rollback()
|
|
return False
|
|
|
|
return True
|
|
|
|
def show_remaining_gpus():
|
|
"""Show remaining GPUs after cleanup"""
|
|
print("\n📋 Remaining GPUs in marketplace:")
|
|
|
|
# Use the same database as coordinator
|
|
db_path = "/home/oib/windsurf/aitbc/apps/coordinator-api/data/coordinator.db"
|
|
engine = create_engine(f"sqlite:///{db_path}")
|
|
|
|
with Session(engine) as session:
|
|
gpus = session.exec(select(GPURegistry)).all()
|
|
|
|
if gpus:
|
|
for gpu in gpus:
|
|
print(f" 🎮 {gpu.id}: {gpu.model} - {gpu.status} - {gpu.price_per_hour} AITBC/hr")
|
|
else:
|
|
print(" No GPUs found")
|
|
|
|
return len(gpus)
|
|
|
|
if __name__ == "__main__":
|
|
if cleanup_fake_gpus():
|
|
remaining = show_remaining_gpus()
|
|
print(f"\n🎉 Cleanup complete! {remaining} GPUs remaining in marketplace")
|
|
else:
|
|
print("\n❌ Cleanup failed!")
|
|
sys.exit(1)
|