Files
WebTech/seed.py

41 lines
933 B
Python

from application import db, app
from application.dash.models import Service
from application.auth.models import User
from werkzeug.security import generate_password_hash
"""
new_strikers = [
se(name="Erik", strike="y", age=44),
Striker(name="Henk", strike="n", age=88),
]
"""
new_users = [
User(
username="admin",
password=generate_password_hash("admin"),
is_admin=True,
),
User(
username="test",
password=generate_password_hash("test123"),
is_admin=False,
),
]
new_services = [
Service(name="test123", url="http://google.com", user_id=1),
Service(name="Netflix", url="https://www.netflix.com", user_id=2),
]
with app.app_context():
# Remove all existing
Service.query.delete()
User.query.delete()
db.session.commit()
# Then add new
db.session.add_all(new_services)
db.session.add_all(new_users)
db.session.commit()