mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2026-02-05 06:07:58 +08:00
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from base64 import b64decode
|
||
|
||
from flask import Blueprint, request
|
||
|
||
from core.api_user import APIUser
|
||
from core.error import PostError
|
||
from core.sql import Connect
|
||
|
||
from .api_auth import api_try, request_json_handle, role_required
|
||
from .api_code import success_return
|
||
|
||
bp = Blueprint('token', __name__, url_prefix='/token')
|
||
|
||
|
||
@bp.route('', methods=['POST'])
|
||
@request_json_handle(request, required_keys=['auth'])
|
||
@api_try
|
||
def token_post(data):
|
||
'''
|
||
登录，获取token\
|
||
{'auth': base64('<user_id>:<password>')}
|
||
'''
|
||
try:
|
||
auth_decode = bytes.decode(b64decode(data['auth']))
|
||
except:
|
||
raise PostError(api_error_code=-100)
|
||
if not ':' in auth_decode:
|
||
raise PostError(api_error_code=-100)
|
||
name, password = auth_decode.split(':', 1)
|
||
|
||
with Connect() as c:
|
||
user = APIUser(c)
|
||
user.login(name, password, request.remote_addr)
|
||
return success_return({'token': user.token, 'user_id': user.user_id})
|
||
|
||
|
||
@bp.route('', methods=['GET'])
|
||
@role_required(request, ['select_me', 'select'])
|
||
@api_try
|
||
def token_get(user):
|
||
'''判断登录有效性'''
|
||
return success_return()
|
||
|
||
|
||
@bp.route('', methods=['DELETE'])
|
||
@role_required(request, ['change_me', 'select_me', 'select'])
|
||
@api_try
|
||
def token_delete(user):
|
||
'''登出'''
|
||
with Connect() as c:
|
||
user.c = c
|
||
user.logout()
|
||
return success_return()
|