Big-Link-Man/scripts/delete_test_sites.py

47 lines
1.3 KiB
Python

"""
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()