web_server.py

from hoshino import Service, priv
from quart import request, Blueprint, jsonify, render_template

import string
import random
import nonebot
import hoshino

from datetime import *
from . import util
from .constant import config


auth = Blueprint('auth', __name__, url_prefix='/auth', template_folder="./vue", static_folder='./vue',
                 static_url_path='')
bot = nonebot.get_bot()
app = bot.server_app
manage_password = config.PASSWORD  # 管理密码请在authMS.py中修改


@auth.route('/')
async def index():
    return await render_template("index.html")


@auth.route('/api/login', methods=['POST'])
async def login_auth():
    password = request.args.get('password')
    if password == manage_password:
        return 'success'
    return 'failed'


@auth.route('/api/get/key', methods=['GET'])
async def get_key():
    password = request.args.get('password')
    if password != manage_password:
        return 'failed'
    return jsonify(util.get_key_list())


@auth.route('/api/add/key', methods=['POST'])
async def add_key():
    password = request.args.get('password')
    if password != manage_password:
        return 'failed'
    if request.method == 'POST':
        duration = int(request.args.get('duration'))
        num = int(request.args.get('num'))
        for _ in range(num):
            util.add_key(duration)
        return 'success'
    return 'failed'


@auth.route('/api/del/key', methods=['DELETE'])
async def del_key():
    password = request.args.get('password')
    if password != manage_password:
        return 'failed'
    if request.method == 'DELETE':
        key = request.args.get('key')
        return jsonify(util.del_key(key))


@auth.route('/api/update/key', methods=['POST'])
async def update_key():
    password = request.args.get('password')
    if password != manage_password:
        return 'failed'
    key = request.args.get('key')
    duration = int(request.args.get('duration'))
    return jsonify(util.update_key(key, duration))


@auth.route('/api/get/group', methods=['GET'])
async def get_group():
    password = request.args.get('password')
    if password != manage_password:
        return 'failed'
    return jsonify(await util.get_authed_group_list())


@auth.route('/api/add/group', methods=['POST'])
@auth.route('/api/update/group', methods=['POST'])
async def update_group():
    password = request.args.get('password')
    if password != manage_password:
        return 'failed'
    gid = int(request.args.get('gid'))
    time_change = int(request.args.get('duration'))
    await util.change_authed_time(gid, time_change)
    return 'success'


@auth.route('/api/activate', methods=['POST'])
async def activate_group():
    key = request.args.get('key')
    gid = request.args.get('gid')
    if await util.reg_group(gid, key):
        return 'success'
    return 'failed'


@auth.route('/api/notify/group', methods=['POST'])
async def notify_group():
    password = request.args.get('password')
    if password != manage_password:
        return 'failed'
    gid = int(request.args.get('gid'))
    msg = request.args.get('msg')
    if await util.notify_group(gid, msg):
        return 'success'
    return 'failed'


@auth.route('/api/gun/group', methods=['POST'])
async def gun_group():
    password = request.args.get('password')
    if password != manage_password:
        return 'failed'
    gid = int(request.args.get('gid'))
    if await util.gun_group(gid):
        return 'success'
    return 'failed'