cosmo/direct_update_config.py

54 lines
1.6 KiB
Python

#!/usr/bin/env python3
"""Direct update celestial_type_configs using SQL"""
import asyncio
import sys
import os
import json
sys.path.append(os.path.join(os.getcwd(), "backend"))
from app.database import AsyncSessionLocal
from sqlalchemy import select, update
from app.models.db.system_settings import SystemSettings
# Unified ratio for all types
UNIFIED_RATIO = 0.00008
async def direct_update():
"""Direct SQL update of celestial_type_configs"""
async with AsyncSessionLocal() as session:
# Get the setting record
stmt = select(SystemSettings).where(SystemSettings.key == 'celestial_type_configs')
result = await session.execute(stmt)
setting = result.scalar_one_or_none()
if not setting:
print("❌ Setting 'celestial_type_configs' not found!")
return
# Parse the JSON value
current_value = json.loads(setting.value) if isinstance(setting.value, str) else setting.value
print("BEFORE UPDATE:")
print(json.dumps(current_value, indent=2))
# Update all ratios
updated_config = {}
for body_type, config in current_value.items():
updated_config[body_type] = {
'ratio': UNIFIED_RATIO,
'default': config.get('default', 0.5)
}
# Update the raw_value (stored as JSON string)
setting.raw_value = json.dumps(updated_config)
await session.commit()
print("\nAFTER UPDATE:")
print(json.dumps(updated_config, indent=2))
print(f"\n✅ Successfully updated all ratios to {UNIFIED_RATIO}")
if __name__ == "__main__":
asyncio.run(direct_update())