feat: add api of body record 2
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
2023-08-12 20:15:48 +08:00
parent c30c644f7c
commit 95621cf3fe
4 changed files with 31 additions and 23 deletions

24
app/util/auth.py Normal file
View File

@@ -0,0 +1,24 @@
from functools import wraps
from flask import request, session
from app.api.v1.exception.auth import AuthError
def login_required(f):
"""
登陆保护,验证用户是否登陆
"""
from app import rpc
@wraps(f)
def wrapper(*args, **kwargs):
token = request.headers.get("Authorization", default=None)
if not token:
raise AuthError()
user_id = rpc.admin.identify(token)
if not user_id:
raise AuthError()
session['user_id'] = user_id
return f(*args, **kwargs)
return wrapper

View File

@@ -1,11 +1,6 @@
import os
from itertools import groupby
from operator import itemgetter
from functools import wraps
from flask import request, session
from app.api.v1.exception.auth import AuthError
def split_group(dict_list, key):
@@ -20,20 +15,4 @@ def split_group(dict_list, key):
basedir = os.getcwd()
def login_required(f):
"""
登陆保护,验证用户是否登陆
"""
from app import rpc
@wraps(f)
def wrapper(*args, **kwargs):
token = request.headers.get("Authorization", default=None)
if not token:
raise AuthError()
user_id = rpc.admin.identify(token)
if not user_id:
raise AuthError()
session['user_id'] = user_id
return f(*args, **kwargs)
return wrapper