mirror of
https://github.com/hyper-neptune/yuuki.git
synced 2024-11-16 03:14:36 +08:00
91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
import hashlib
|
|
import json
|
|
import random
|
|
import time
|
|
|
|
from flask import Flask, render_template, Response, request, redirect
|
|
from flask_bootstrap import Bootstrap
|
|
|
|
from gevent.pywsgi import WSGIServer
|
|
|
|
wa_app = Flask(__name__)
|
|
Yuuki_Handle = None
|
|
|
|
passports = []
|
|
password = str(hash(random.random()))
|
|
|
|
class Yuuki_WebAdmin:
|
|
def __init__(self, Yuuki):
|
|
global Yuuki_Handle
|
|
Yuuki_Handle = Yuuki
|
|
|
|
self.app = wa_app
|
|
Bootstrap(self.app)
|
|
|
|
@staticmethod
|
|
@wa_app.route("/")
|
|
def index():
|
|
if "yuuki_admin" in request.cookies:
|
|
if request.cookies["yuuki_admin"] in passports:
|
|
return render_template(
|
|
'manage.html',
|
|
name=Yuuki_Handle.YuukiConfigs["name"],
|
|
version=Yuuki_Handle.YuukiConfigs["version"],
|
|
LINE_Media_server=Yuuki_Handle.LINE_Media_server,
|
|
profileName=Yuuki_Handle.profile.displayName,
|
|
pictureStatus=Yuuki_Handle.profile.pictureStatus
|
|
)
|
|
else:
|
|
response = redirect("/")
|
|
response.set_cookie(
|
|
key='yuuki_admin',
|
|
value='',
|
|
expires=0
|
|
)
|
|
return response
|
|
else:
|
|
return render_template(
|
|
'index.html',
|
|
name=Yuuki_Handle.YuukiConfigs["name"],
|
|
version=Yuuki_Handle.YuukiConfigs["version"]
|
|
)
|
|
|
|
@staticmethod
|
|
@wa_app.route("/verify", methods=['GET', 'POST'])
|
|
def verify():
|
|
result = {"status": 403}
|
|
if request.method == "POST" and "code" in request.values:
|
|
if request.values["code"] == password:
|
|
seed = hash(random.random() + time.time())
|
|
seed = str(seed).encode('utf-8')
|
|
session_key = hashlib.sha256(seed).hexdigest()
|
|
passports.append(session_key)
|
|
result = {"status": 200, "session": session_key}
|
|
else:
|
|
result = {"status": 401}
|
|
return Response(json.dumps(result), mimetype='application/json')
|
|
|
|
|
|
@staticmethod
|
|
@wa_app.route("/logout")
|
|
def logout():
|
|
response = redirect("/")
|
|
if "yuuki_admin" in request.cookies:
|
|
if request.cookies.get("yuuki_admin") in passports:
|
|
passports.remove(request.cookies.get("yuuki_admin"))
|
|
response.set_cookie(
|
|
key='yuuki_admin',
|
|
value='',
|
|
expires=0
|
|
)
|
|
return response
|
|
|
|
@staticmethod
|
|
def set_password(code):
|
|
global password
|
|
password = code
|
|
|
|
def start(self, admin_password):
|
|
self.set_password(admin_password)
|
|
http_server = WSGIServer(('', 2020), self.app)
|
|
http_server.serve_forever()
|