from extensions import db
from models.company import Company

def seed_company():
    companys = [
        # Empresa 1
        {
            'name': "Tech Solutions LTDA",
            "fantasy_name": "TechSol",
            "cnpj": "12.345.678/0001-90",
            "type": "TI",
            'email': "contato@techsol.com.br",
            "email_sec": "financeiro@techsol.com.br",
            "phone": "(11) 91234-5678",
            "phone_sec": "(11) 4002-8922",
            "cep": "01001-000",
            "rua": "Av. Paulista",
            "numero": "1000",
            "bairro": "Bela Vista",
            "complemento": "Bloco A",
            "estado": "SP",
            "cidade": "São Paulo",
            'service_instance_id': 1,
            "parent_id":1
        },

        # Empresa 2
        {
            'name': "Inova Sistemas SA",
            "fantasy_name": "InovaSys",
            "cnpj": "98.765.432/0001-10",
            "type": "Consultoria",
            'email': "suporte@inovasys.com",
            "email_sec": "vendas@inovasys.com",
            "phone": "(21) 99876-5432",
            "phone_sec": "(21) 3003-1234",
            "cep": "20040-010",
            "rua": "Rua do Ouvidor",
            "numero": "150",
            "bairro": "Centro",
            "complemento": "Sala 305",
            "estado": "RJ",
            "cidade": "Rio de Janeiro",
            'service_instance_id': 1,
            "parent_id":2
        },
    ]

    for company_data in companys:
        existing_company = db.session.query(Company).filter_by(
            name=company_data['name']
        ).first()

        if not existing_company:
            company = Company(
                name=company_data['name'],
                fantasy_name=company_data['fantasy_name'],
                cnpj=company_data['cnpj'],
                type=company_data['type'],
                email=company_data['email'],
                email_sec=company_data['email_sec'],
                phone=company_data['phone'],
                phone_sec=company_data['phone_sec'],
                cep=company_data['cep'],
                rua=company_data['rua'],
                numero=company_data['numero'],
                bairro=company_data['bairro'],
                complemento=company_data['complemento'],
                estado=company_data['estado'],
                cidade=company_data['cidade'],
                service_instance_id=company_data['service_instance_id'],
                parent_id=company_data['parent_id']
            )
            db.session.add(company)

    db.session.commit()