Changelog: https://roxy-wi.org/changelog#7.0.0
pull/364/head
Aidaho 2023-09-30 14:54:44 +03:00
parent bc8bdc36b2
commit 9ec2cd3c8f
4 changed files with 19 additions and 12 deletions

View File

@ -12,6 +12,7 @@ app.jinja_env.add_extension('jinja2.ext.loopcontrols')
cache = Cache(config={'CACHE_TYPE': 'SimpleCache', "CACHE_DEFAULT_TIMEOUT": 3000}) cache = Cache(config={'CACHE_TYPE': 'SimpleCache', "CACHE_DEFAULT_TIMEOUT": 3000})
cache.init_app(app) cache.init_app(app)
login_manager = LoginManager(app) login_manager = LoginManager(app)
login_manager.login_view = 'app.login.login_page'
app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True
app.config['PERMANENT_SESSION_LIFETIME'] = timedelta(days=3) app.config['PERMANENT_SESSION_LIFETIME'] = timedelta(days=3)
@ -63,3 +64,5 @@ app.register_blueprint(server_bp, url_prefix='/server')
from app.routes.admin import bp as admin_bp from app.routes.admin import bp as admin_bp
app.register_blueprint(admin_bp, url_prefix='/admin') app.register_blueprint(admin_bp, url_prefix='/admin')
from app import login

View File

@ -1,5 +1,3 @@
import os
import sys
import uuid import uuid
import distro import distro
@ -8,14 +6,11 @@ from flask import render_template, request, redirect, url_for, flash, make_respo
from flask_login import login_user, login_required, logout_user, current_user from flask_login import login_user, login_required, logout_user, current_user
from app import app, login_manager, cache from app import app, login_manager, cache
import app.modules.db.sql as sql
sys.path.append(os.path.join(sys.path[0], '/var/www/haproxy-wi/app')) import app.modules.server.server as server_mod
import app.modules.roxywi.common as roxywi_common
import modules.db.sql as sql import app.modules.roxywi.auth as roxywi_auth
from modules.db.db_model import * import app.modules.roxy_wi_tools as roxy_wi_tools
import modules.server.server as server_mod
import modules.roxywi.common as roxywi_common
import modules.roxywi.auth as roxywi_auth
@app.before_request @app.before_request
@ -41,7 +36,7 @@ def load_user(user_id):
user_obj = cache.get(user) user_obj = cache.get(user)
if user_obj is None: if user_obj is None:
query = User.get(User.user_id == user_id) query = sql.get_user_id(user_id)
cache.set(user, query, timeout=360) cache.set(user, query, timeout=360)
return query return query

View File

@ -731,6 +731,15 @@ def get_user_name_by_uuid(uuid):
return user.username return user.username
def get_user_id(user_id: int) -> int:
try:
query = User.get(User.user_id == user_id)
except Exception as e:
out_error(e)
else:
return query
def get_user_id_by_uuid(uuid): def get_user_id_by_uuid(uuid):
try: try:
query = User.select(User.user_id).join(UUID, on=(User.user_id == UUID.user_id)).where(UUID.uuid == uuid) query = User.select(User.user_id).join(UUID, on=(User.user_id == UUID.user_id)).where(UUID.uuid == uuid)

View File

@ -2,7 +2,7 @@ import os
import sys import sys
import pytz import pytz
from flask import render_template, request, redirect, url_for, make_response from flask import render_template, request, redirect, url_for
from flask_login import login_required from flask_login import login_required
sys.path.append(os.path.join(sys.path[0], '/var/www/haproxy-wi/app')) sys.path.append(os.path.join(sys.path[0], '/var/www/haproxy-wi/app'))