49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
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!")
|