cosmo/backend/check_db_data.py

33 lines
1.0 KiB
Python

import asyncio
import logging
import sys
import os
# Add backend directory to path
sys.path.append(os.path.join(os.getcwd(), "backend"))
from app.database import AsyncSessionLocal
from sqlalchemy import select
from app.models.db.celestial_body import CelestialBody
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
async def check_data():
async with AsyncSessionLocal() as session:
# Check Earth and Jupiter
stmt = select(CelestialBody).where(CelestialBody.name.in_(['Earth', 'Jupiter']))
result = await session.execute(stmt)
bodies = result.scalars().all()
print(f"Found {len(bodies)} bodies.")
for body in bodies:
print(f"Body: {body.name} (ID: {body.id})")
print(f" Extra Data (Raw): {body.extra_data}")
if body.extra_data:
print(f" Real Radius: {body.extra_data.get('real_radius')}")
else:
print(" Extra Data is None/Empty")
if __name__ == "__main__":
asyncio.run(check_data())