@login_manager.user_loader def load_user(user_id): """ load a User by a given user id :param user_id: the User's id :return: the User or None """ return User.query.get(user_id) def create_initial_user(): """ create an initial User if there is none yet """ # check if there is any user in the db if User.query.count() == 0: # create a user user = User(username="******", password=hashlib.sha256( "admin".encode("utf8")).hexdigest(), email="*****@*****.**") user.is_admin = True db.session.add(user) db.session.commit() on_application_ready.append(create_initial_user)
import atexit from app import on_application_ready, app from app.modules.mod_process.process_repository import ProcessRepository def register_jinja2_functions(): """ register count_process_* and encoding_active() functions for jinja2 """ from app.modules.mod_process.process_repository import ProcessRepository app.jinja_env.globals.update(count_processes_active=ProcessRepository.count_processes_active) app.jinja_env.globals.update(count_processes_queued=ProcessRepository.count_processes_queued) app.jinja_env.globals.update(count_processes_total=ProcessRepository.count_processes_total) app.jinja_env.globals.update(encoding_active=lambda: ProcessRepository.encoding_active) def fail_on_exit(): """ on exit cancel all processes """ ProcessRepository.cancel_all_processes() # run fail method when this Thread is still running and the program quits unexpectedly atexit.register(fail_on_exit) # register common jinja2 functions on_application_ready.append(register_jinja2_functions)
def register_jinja2_functions(): """ register count_process_* and encoding_active() functions for jinja2 """ from app.modules.mod_process.process_repository import ProcessRepository app.jinja_env.globals.update( count_processes_active=ProcessRepository.count_processes_active) app.jinja_env.globals.update( count_processes_queued=ProcessRepository.count_processes_queued) app.jinja_env.globals.update( count_processes_total=ProcessRepository.count_processes_total) app.jinja_env.globals.update( encoding_active=lambda: ProcessRepository.encoding_active) def fail_on_exit(): """ on exit cancel all processes """ ProcessRepository.cancel_all_processes() # run fail method when this Thread is still running and the program quits unexpectedly atexit.register(fail_on_exit) # register common jinja2 functions on_application_ready.append(register_jinja2_functions)
login_manager.localize_callback = _ @login_manager.user_loader def load_user(user_id): """ load a User by a given user id :param user_id: the User's id :return: the User or None """ return User.query.get(user_id) def create_initial_user(): """ create an initial User if there is none yet """ # check if there is any user in the db if User.query.count() == 0: # create a user user = User(username="******", password=hashlib.sha256("admin".encode("utf8")).hexdigest(), email="*****@*****.**") user.is_admin = True db.session.add(user) db.session.commit() on_application_ready.append(create_initial_user)