Just commit

This commit is contained in:
Donald Zou
2025-06-02 12:04:01 +08:00
parent 29600cb54c
commit bdf557fde3
7 changed files with 111 additions and 27 deletions

View File

@@ -1,6 +1,10 @@
from flask import Blueprint, render_template, abort, request, Flask, current_app
from functools import wraps
from flask import Blueprint, render_template, abort, request, Flask, current_app, session
import os
client = Blueprint('client', __name__, template_folder=os.path.abspath("./static/client/dist"))
from modules.WireguardConfiguration import WireguardConfiguration
from modules.DashboardConfig import DashboardConfig
def ResponseObject(status=True, message=None, data=None, status_code = 200) -> Flask.response_class:
response = Flask.make_response(current_app, {
@@ -12,16 +16,39 @@ def ResponseObject(status=True, message=None, data=None, status_code = 200) -> F
response.content_type = "application/json"
return response
def login_required(f):
@wraps(f)
def func(*args, **kwargs):
if session.get("username") is None or session.get("role") != "client":
return ResponseObject(False, "Unauthorized access.", data=None, status_code=401)
return f(*args, **kwargs)
return func
prefix = '/client'
@client.before_request
def clientBeforeRequest():
if request.method.lower() == 'options':
return ResponseObject(True)
@client.get(prefix)
def clientIndex():
return render_template('client.html')
def createClientBlueprint(wireguardConfigurations: dict[WireguardConfiguration], dashboardConfig: DashboardConfig):
from modules.DashboardClients import DashboardClients
DashboardClients = DashboardClients()
client = Blueprint('client', __name__, template_folder=os.path.abspath("./static/client/dist"))
prefix = f'{dashboardConfig.GetConfig("Server", "app_prefix")[1]}/client'
@client.before_request
def clientBeforeRequest():
if request.method.lower() == 'options':
return ResponseObject(True)
@client.get(prefix)
def ClientIndex():
print(wireguardConfigurations.keys())
return render_template('client.html')
@client.post(f'{prefix}/api/signup')
def ClientAPI_SignUp():
data = request.json
status, msg = DashboardClients.SignUp(**data)
return ResponseObject(status, msg)
return client