Refactor to have all models in the models package.
This commit is contained in:
		@@ -1,7 +1,7 @@
 | 
			
		||||
from flask import Blueprint, render_template, request, session, redirect
 | 
			
		||||
from flask.helpers import flash, url_for
 | 
			
		||||
from flask.json import jsonify
 | 
			
		||||
from .user.models import User
 | 
			
		||||
from .models.users import User
 | 
			
		||||
from uuid import uuid4
 | 
			
		||||
from security.database import decrypt_find_one, encrypted_update
 | 
			
		||||
from werkzeug.security import check_password_hash
 | 
			
		||||
@@ -21,7 +21,7 @@ auth = Blueprint(
 | 
			
		||||
@admin_account_required
 | 
			
		||||
@login_required
 | 
			
		||||
def account():
 | 
			
		||||
    from .forms import UpdateAccountForm
 | 
			
		||||
    from .models.forms import UpdateAccountForm
 | 
			
		||||
    form = UpdateAccountForm()
 | 
			
		||||
    _id = get_id_from_cookie()
 | 
			
		||||
    user = decrypt_find_one(db.users, {'_id': _id})
 | 
			
		||||
@@ -46,7 +46,7 @@ def account():
 | 
			
		||||
@admin_account_required
 | 
			
		||||
@disable_if_logged_in
 | 
			
		||||
def login():
 | 
			
		||||
    from .forms import LoginForm
 | 
			
		||||
    from .models.forms import LoginForm
 | 
			
		||||
    form = LoginForm()
 | 
			
		||||
    if request.method == 'GET':
 | 
			
		||||
        return render_template('/admin/auth/login.html', form=form)
 | 
			
		||||
@@ -72,7 +72,7 @@ def logout():
 | 
			
		||||
@auth.route('/register/', methods=['GET','POST'])
 | 
			
		||||
@disable_on_registration
 | 
			
		||||
def register():
 | 
			
		||||
    from .forms import RegistrationForm
 | 
			
		||||
    from .models.forms import RegistrationForm
 | 
			
		||||
    form = RegistrationForm()
 | 
			
		||||
    if request.method == 'GET':
 | 
			
		||||
        return render_template('/admin/auth/register.html', form=form)
 | 
			
		||||
@@ -93,7 +93,7 @@ def register():
 | 
			
		||||
@admin_account_required
 | 
			
		||||
@disable_if_logged_in
 | 
			
		||||
def reset():
 | 
			
		||||
    from .forms import ResetPasswordForm
 | 
			
		||||
    from .models.forms import ResetPasswordForm
 | 
			
		||||
    form = ResetPasswordForm()
 | 
			
		||||
    if request.method == 'GET':
 | 
			
		||||
        return render_template('/admin/auth/reset.html', form=form)
 | 
			
		||||
@@ -128,7 +128,7 @@ def reset_gateway(token1,token2):
 | 
			
		||||
@admin_account_required
 | 
			
		||||
@disable_if_logged_in
 | 
			
		||||
def update_password_():
 | 
			
		||||
    from .forms import UpdatePasswordForm
 | 
			
		||||
    from .models.forms import UpdatePasswordForm
 | 
			
		||||
    form = UpdatePasswordForm()
 | 
			
		||||
    if request.method == 'GET':
 | 
			
		||||
        if 'reset_validated' not in session:
 | 
			
		||||
 
 | 
			
		||||
@@ -4,14 +4,14 @@ from functools import wraps
 | 
			
		||||
 | 
			
		||||
from werkzeug.security import check_password_hash
 | 
			
		||||
from security.database import decrypt_find, decrypt_find_one
 | 
			
		||||
from .user.models import User
 | 
			
		||||
from .models.users import User
 | 
			
		||||
from flask_mail import Message
 | 
			
		||||
from main import db
 | 
			
		||||
from uuid import uuid4
 | 
			
		||||
import secrets
 | 
			
		||||
from main import mail
 | 
			
		||||
from datetime import datetime, date, timedelta
 | 
			
		||||
from .models import Test
 | 
			
		||||
from .models.tests import Test
 | 
			
		||||
 | 
			
		||||
views = Blueprint(
 | 
			
		||||
    'admin_views',
 | 
			
		||||
@@ -82,7 +82,7 @@ def settings():
 | 
			
		||||
@admin_account_required
 | 
			
		||||
@login_required
 | 
			
		||||
def users():
 | 
			
		||||
    from .forms import CreateUserForm
 | 
			
		||||
    from .models.forms import CreateUserForm
 | 
			
		||||
    form = CreateUserForm()
 | 
			
		||||
    if request.method == 'GET':
 | 
			
		||||
        users_list = decrypt_find(db.users, {})
 | 
			
		||||
@@ -132,7 +132,7 @@ def delete_user(_id:str):
 | 
			
		||||
    if _id == get_id_from_cookie():
 | 
			
		||||
        flash('Cannot delete your own user account.', 'error')
 | 
			
		||||
        return redirect(url_for('admin_views.users'))
 | 
			
		||||
    from .forms import DeleteUserForm
 | 
			
		||||
    from .models.forms import DeleteUserForm
 | 
			
		||||
    form = DeleteUserForm()
 | 
			
		||||
    user = decrypt_find_one(db.users, {'_id': _id})
 | 
			
		||||
    if request.method == 'GET':
 | 
			
		||||
@@ -181,7 +181,7 @@ def update_user(_id:str):
 | 
			
		||||
    if _id == get_id_from_cookie():
 | 
			
		||||
        flash('Cannot delete your own user account.', 'error')
 | 
			
		||||
        return redirect(url_for('admin_views.users'))
 | 
			
		||||
    from .forms import UpdateUserForm
 | 
			
		||||
    from .models.forms import UpdateUserForm
 | 
			
		||||
    form = UpdateUserForm()
 | 
			
		||||
    user = decrypt_find_one( db.users, {'_id': _id})
 | 
			
		||||
    if request.method == 'GET':
 | 
			
		||||
@@ -251,7 +251,7 @@ def tests(filter=''):
 | 
			
		||||
    if filter not in ['', 'create', 'active', 'scheduled', 'expired', 'all']:
 | 
			
		||||
        return abort(404)
 | 
			
		||||
    if filter == 'create':
 | 
			
		||||
        from .forms import CreateTest
 | 
			
		||||
        from .models.forms import CreateTest
 | 
			
		||||
        form = CreateTest()
 | 
			
		||||
        form.time_limit.default='none'
 | 
			
		||||
        form.process()
 | 
			
		||||
@@ -281,7 +281,7 @@ def tests(filter=''):
 | 
			
		||||
@admin_account_required
 | 
			
		||||
@login_required
 | 
			
		||||
def _tests():
 | 
			
		||||
    from .forms import CreateTest
 | 
			
		||||
    from .models.forms import CreateTest
 | 
			
		||||
    form = CreateTest()
 | 
			
		||||
    form.time_limit.default='none'
 | 
			
		||||
    form.process()
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user