77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
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()
|