"""
Delete test sites from database
"""

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent))

from src.database.session import db_manager
from src.database.repositories import SiteDeploymentRepository

def delete_test_sites():
    """Delete test sites with IDs 398, 399, 400, 401"""
    db_manager.initialize()
    session = db_manager.get_session()
    site_repo = SiteDeploymentRepository(session)
    
    test_site_ids = [398, 399, 400, 401]
    
    try:
        for site_id in test_site_ids:
            site = site_repo.get_by_id(site_id)
            if site:
                hostname = site.custom_hostname or site.pull_zone_bcdn_hostname
                print(f"Deleting site {site_id}: {site.site_name} ({hostname})")
                if site_repo.delete(site_id):
                    print(f"  Deleted site {site_id}")
                else:
                    print(f"  Failed to delete site {site_id}")
            else:
                print(f"  Site {site_id} not found, skipping")
        
        print(f"\nDeleted {len(test_site_ids)} test sites")
        
    except Exception as e:
        print(f"Error deleting sites: {e}")
        import traceback
        traceback.print_exc()
        session.rollback()
    finally:
        session.close()

if __name__ == "__main__":
    delete_test_sites()