github
/
haproxy-wi
mirror of https://github.com/Aidaho12/haproxy-wi
1
0
Fork 0
Code Issues Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/modules/db/sql.py
View File

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