feat: add api of body record 2
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
24
app/util/auth.py
Normal file
24
app/util/auth.py
Normal 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
|
@@ -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
|
||||
|
Reference in New Issue
Block a user