#!/usr/bin/env python3
"""
Manual migration to add rating fields to contracts table
Run this if automatic migration fails
"""

from app import create_app
from app.extensions import db

def add_rating_columns():
    app = create_app()
    with app.app_context():
        try:
            # Add columns with proper defaults
            db.engine.execute("""
                ALTER TABLE contracts 
                ADD COLUMN contractor_rated BOOLEAN DEFAULT FALSE NOT NULL;
            """)
            
            db.engine.execute("""
                ALTER TABLE contracts 
                ADD COLUMN worker_rated BOOLEAN DEFAULT FALSE NOT NULL;
            """)
            
            db.engine.execute("""
                ALTER TABLE contracts 
                ADD COLUMN mutual_rating_completed_date TIMESTAMP NULL;
            """)
            
            print("✅ Successfully added rating columns to contracts table")
            
        except Exception as e:
            print(f"❌ Error adding columns: {e}")
            db.session.rollback()

if __name__ == "__main__":
    add_rating_columns()