diff --git a/advlabdb/models.py b/advlabdb/models.py index c410e21..46d88c1 100644 --- a/advlabdb/models.py +++ b/advlabdb/models.py @@ -10,8 +10,7 @@ from decimal import ROUND_HALF_UP, Decimal from flask import flash from flask_login import current_user -from flask_security.models import fsqla_v2 as fsqla -from flask_security.models.fsqla_v2 import FsRoleMixin, FsUserMixin +from flask_security.models.fsqla_v3 import FsModels, FsRoleMixin, FsUserMixin from flask_sqlalchemy import SQLAlchemy from sqlalchemy import func, select @@ -30,7 +29,7 @@ MIN_PART_NUMBER = 1 db = SQLAlchemy() # For Flask-Security-Too -fsqla.FsModels.set_db_info(db) +FsModels.set_db_info(db) def get_count(table):