Files
aitbc/scripts/cleanup_fake_gpus_db.py
AITBC System b033923756 chore: normalize file permissions across repository
- Remove executable permissions from configuration files (.editorconfig, .env.example, .gitignore)
- Remove executable permissions from documentation files (README.md, LICENSE, SECURITY.md)
- Remove executable permissions from web assets (HTML, CSS, JS files)
- Remove executable permissions from data files (JSON, SQL, YAML, requirements.txt)
- Remove executable permissions from source code files across all apps
- Add executable permissions to Python
2026-03-08 11:26:18 +01:00

79 lines
2.3 KiB
Python
Executable File

#!/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)