# login functionality part 2
from application.auth.models import User
@login_manager.user_loader
def load_user(user_id):
return User.query.get(user_id)
# database creation
try:
db.create_all()
except:
pass
# If there are no accounts create admin account
users_amount = User.amount_of_users()
if users_amount == 0:
User.create_admin_account()
# If there are no categories, create default category
# (default category is needed for adding items)
from application.categories.models import Category
categories_amount = Category.amount_of_categories()
if categories_amount == 0:
Category.create_default_category()
