mirror of https://github.com/Aidaho12/haproxy-wi
parent
bc8bdc36b2
commit
9ec2cd3c8f
|
@ -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
|
||||||
|
|
17
app/login.py
17
app/login.py
|
@ -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
|
||||||
|
|
||||||
|
|
|
@ -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)
|
||||||
|
|
|
@ -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'))
|
||||||
|
|
Loading…
Reference in New Issue