v8.0.1: Set user_group_id from Flask global object

Added logic to set user_group_id based on Flask global object `g`, with a fallback to default value if an exception occurs. This enhances the flexibility of `user_group_id` assignment based on runtime context.
pull/399/head v8.0.1
Aidaho 2024-09-14 11:58:47 +03:00
parent 202513fcfd
commit 71ed2cc756
1 changed files with 6 additions and 1 deletions

View File

@ -1,3 +1,5 @@
from flask import g
from app.modules.db.db_model import GeoipCodes, Setting, Role
from app.modules.db.common import out_error
@ -6,7 +8,10 @@ def get_setting(param, **kwargs):
if kwargs.get('group_id'):
user_group_id = kwargs.get('group_id')
else:
user_group_id = 1
try:
user_group_id = g.user_params['group_id']
except Exception:
user_group_id = 1
if kwargs.get('all') and not kwargs.get('section'):
query = Setting.select().where(Setting.group == user_group_id).order_by(Setting.section.desc())