from application.search import views
from application.developers import views
from application.auth.models import User, UserRole
# kirjautumistoiminnallisuus 2
@login_manager.user_loader
def load_user(user_id):
return User.query.get(user_id)
# tietokannan luonti
# materiaalin try - except johti ajoittain herokussa virheeseen kun dyno käynnistyy uudellen.
from sqlalchemy import MetaData
metadata = MetaData(db.engine, reflect=True)
if not metadata.tables:
db.create_all()
# Alustetaan tietokanta jos sitä ei ole vielä luotu
admin = User("Admin Hallitsija", "admin", "admin_salasana", None)
admin_role = Role("ADMIN")
normal = User("Esimerkki Käyttäjä", "testi", "salasana", None)
default_role = Role("DEFAULT")
db.session.add_all((admin, normal, admin_role, default_role))
db.session.commit()
# vasta nyt pääavaimet on luotu
admin.add_roles(admin_role, default_role)
normal.add_role(default_role)
