from extensions import db
from models.user import User
from models.contact import Contact

def seed_contacts():
    # Supondo que já exista um usuário no banco
    user = User.query.filter_by(email="user1@exemplo.com").first()
    
    if not user:
        print("Usuário não encontrado. Certifique-se de que o usuário foi criado.")
        return

    # Dados dos contatos que queremos criar
    contacts_data = [
        {
            "phone": "987654321",
            "contact_type": "pessoal",
            "description": "Contato de emergência",
        },
        {
            "phone": "912345678",
            "contact_type": "comercial",
            "description": "Contato de trabalho",
        },
    ]

    for contact_data in contacts_data:
        # Verifica se o contato já existe no banco para o mesmo usuário
        existing_contact = Contact.query.filter_by(
            phone=contact_data["phone"], user_id=user.id
        ).first()
        
        if existing_contact:
            print(f"Contato com o telefone {contact_data['phone']} já existe.")
            continue

        # Cria e adiciona o contato
        new_contact = Contact(
            phone=contact_data["phone"],
            contact_type=contact_data["contact_type"],
            description=contact_data["description"],
            user_id=user.id
        )
        db.session.add(new_contact)

    # Confirma as alterações no banco
    db.session.commit()
    print(f"Contatos para {user.email} criados com sucesso!")