6bcbe76c7d
- Change database from in-memory to file-based SQLite at aitbc_coordinator.db - Add status="active" to GPU booking creation - Allow GPU release even when not properly booked (cleanup case) - Add error handling for missing booking attributes during refund calculation - Fix get_gpu_reviews query to use scalars() for proper result handling
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Script to clean up fake GPU entries from the marketplace
|
|
"""
|
|
|
|
import requests
|
|
import sys
|
|
|
|
def delete_fake_gpu(gpu_id):
|
|
"""Delete a fake GPU from the marketplace"""
|
|
try:
|
|
response = requests.delete(f"http://localhost:8000/v1/marketplace/gpu/{gpu_id}")
|
|
if response.status_code == 200:
|
|
print(f"✅ Successfully deleted fake GPU: {gpu_id}")
|
|
return True
|
|
else:
|
|
print(f"❌ Failed to delete {gpu_id}: {response.status_code}")
|
|
return False
|
|
except Exception as e:
|
|
print(f"❌ Error deleting {gpu_id}: {e}")
|
|
return False
|
|
|
|
def main():
|
|
"""Main cleanup function"""
|
|
print("=== CLEANING UP FAKE GPU OFFERS ===")
|
|
|
|
# List of fake GPU IDs to delete
|
|
fake_gpus = [
|
|
"gpu_1bdf8e86",
|
|
"gpu_1b7da9e0",
|
|
"gpu_9cff5bc2",
|
|
"gpu_ebef80a5",
|
|
"gpu_979b24b8",
|
|
"gpu_e5ab817d"
|
|
]
|
|
|
|
print(f"Found {len(fake_gpus)} fake GPUs to delete")
|
|
|
|
deleted_count = 0
|
|
for gpu_id in fake_gpus:
|
|
if delete_fake_gpu(gpu_id):
|
|
deleted_count += 1
|
|
|
|
print(f"\n🎉 Cleanup complete! Deleted {deleted_count}/{len(fake_gpus)} fake GPUs")
|
|
|
|
# Show remaining GPUs
|
|
print("\n📋 Remaining GPUs in marketplace:")
|
|
try:
|
|
response = requests.get("http://localhost:8000/v1/marketplace/gpu/list")
|
|
if response.status_code == 200:
|
|
data = response.json()
|
|
if 'items' in data:
|
|
for gpu in data['items']:
|
|
print(f" 🎮 {gpu['id']}: {gpu['model']} - {gpu['status']}")
|
|
else:
|
|
print(" No GPUs found")
|
|
else:
|
|
print(f" Error fetching GPU list: {response.status_code}")
|
|
except Exception as e:
|
|
print(f" Error: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|