src/db/verify_migration.py

import sys
from pathlib import Path
sys.path.append(str(Path(__file__).parent.parent.parent))
from src.db.local_database import LocalDatabase

def verify():
    print("🔍 Verifying data access via LocalDatabase (TiDB)...")
    db = LocalDatabase()
    
    stats = db.get_stats()
    print("\nDatabase Statistics:")
    print(f"Total Entries: {stats.get('total_entries')}")
    print(f"By Type: {stats.get('by_type')}")
    print(f"Top Tickers: {stats.get('top_tickers')}")
    
    if stats.get('total_entries', 0) > 0:
        print("\n✅ Verification Successful: Data is accessible!")
    else:
        print("\n❌ Verification Failed: No data found.")

if __name__ == "__main__":
    verify()