37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
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!")
|