30 lines
809 B
Python
30 lines
809 B
Python
from sqlalchemy.orm import Session
|
|
|
|
from api.database import SessionLocal
|
|
from api.hasher import argon2_hasher
|
|
import api.models as models
|
|
|
|
|
|
def seed():
|
|
db: Session = SessionLocal()
|
|
|
|
if db.query(models.User).count() == 0:
|
|
admin = models.User(name="admin", password_hash=argon2_hasher.hash("admin"))
|
|
db.add(admin)
|
|
db.commit()
|
|
|
|
if db.query(models.Service).count() == 0:
|
|
lb = models.Loadbalancer()
|
|
lb.servers.append(
|
|
models.LoadbalancerServers(address="http://192.168.2.134:8000/")
|
|
)
|
|
lb.servers.append(
|
|
models.LoadbalancerServers(address="http://192.168.2.132:80/")
|
|
)
|
|
|
|
test_service = models.Service(name="service")
|
|
test_service.loadbalancer = lb
|
|
|
|
db.add(test_service)
|
|
db.commit()
|