root
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
import os
|
|
import sys
|
|
from sqlmodel import Session, select, create_engine, inspect
|
|
|
|
# Add root directory to sys.path
|
|
sys.path.append(os.getcwd())
|
|
|
|
from app.database import engine
|
|
from app.models.auth import UserTable
|
|
from app.models.watchlist import WatchlistItemTable, WatchlistSettingsTable
|
|
from app.models.favorites import FavoriteTable
|
|
from app.models.sonarr import SonarrMappingTable, SonarrConfigTable
|
|
|
|
def audit_db():
|
|
print("--- Ohm Stream Downloader: SQL Database Audit ---")
|
|
|
|
if not os.path.exists("ohm_streaming.db"):
|
|
print("ERROR: ohm_streaming.db not found!")
|
|
return
|
|
|
|
inspector = inspect(engine)
|
|
tables = inspector.get_table_names()
|
|
print(f"Tables found: {', '.join(tables)}")
|
|
|
|
expected_tables = ["users", "watchlist_items", "watchlist_settings", "favorites", "sonarr_mappings", "sonarr_config", "alembic_version"]
|
|
missing = [t for t in expected_tables if t not in tables]
|
|
if missing:
|
|
print(f"WARNING: Missing tables: {', '.join(missing)}")
|
|
else:
|
|
print("SUCCESS: All core tables are present.")
|
|
|
|
with Session(engine) as session:
|
|
# Check users
|
|
users_count = len(session.exec(select(UserTable)).all())
|
|
print(f"Users: {users_count}")
|
|
|
|
# Check watchlist
|
|
watchlist_count = len(session.exec(select(WatchlistItemTable)).all())
|
|
print(f"Watchlist Items: {watchlist_count}")
|
|
|
|
# Check settings
|
|
settings_count = len(session.exec(select(WatchlistSettingsTable)).all())
|
|
print(f"Watchlist Settings entries: {settings_count}")
|
|
|
|
# Check favorites
|
|
fav_count = len(session.exec(select(FavoriteTable)).all())
|
|
print(f"Favorites: {fav_count}")
|
|
|
|
# Check Sonarr
|
|
sonarr_map_count = len(session.exec(select(SonarrMappingTable)).all())
|
|
print(f"Sonarr Mappings: {sonarr_map_count}")
|
|
|
|
# Sample data check
|
|
if fav_count > 0:
|
|
sample_fav = session.exec(select(FavoriteTable).limit(1)).first()
|
|
print(f"Sample Favorite: {sample_fav.title} ({sample_fav.provider})")
|
|
|
|
if __name__ == "__main__":
|
|
audit_db()
|