from src.services.postgresql_manager import PostgreSQLManager
from src.models.base import Base
from src.models.scheduled_job import ScheduledJob
from src.utils.config import config
import logging
from sqlalchemy import text

if __name__ == "__main__":
    logging.basicConfig(level=logging.INFO)
    pm = PostgreSQLManager(config.DATABASE_URL)
    with pm.get_db_session() as session:
        print("Dropping scheduled_jobs table...")
        session.execute(text("DROP TABLE IF EXISTS scheduled_jobs"))
        session.commit()
        
    print("Re-creating tables...")
    pm.init_db()
    print("Done.")