from models.license import License
from extensions import db
from datetime import datetime, timedelta

def seed_licenses():
    licenses = [
        License(
            license_key="ABC123",
            service_id=1,
            user_id=1,
            max_service_instances=3,
            start_date=datetime.now(),
            end_date=datetime.now() + timedelta(days=365)
        ),
        License(
            license_key="XYZ789",
            service_id=2,
            user_id=2,
            max_service_instances=3,
            start_date=datetime.now(),
            end_date=None,  # Licença vitalícia
        ),
    ]
    
    for license in licenses:
        # Verifica se a licença com a mesma chave já existe
        existing_license = License.query.filter_by(license_key=license.license_key).first()
        if existing_license:
            print(f"Licença com chave '{license.license_key}' já existe. Pulando a criação.")
            continue  # Se já existir, pula a inserção desta licença
        
        # Se não existir, adiciona a licença
        db.session.add(license)
    
    db.session.commit()
    print("Licenças inseridas com sucesso!")