This commit is contained in:
yumoqing 2025-07-31 16:01:10 +08:00
parent 3d0253d297
commit 5aa3a06966
2 changed files with 15 additions and 24 deletions

View File

@ -50,8 +50,8 @@ where a.userid = b.id
return None
return recs[0]
async def bearer_auth(request):
auth = request.headers.get('Authorization')
async def bearer_auth(sor, request):
auth = request.headers.get('Authentication')
if auth is None:
return None
if not auth.startswith('Bearer '):
@ -60,16 +60,12 @@ async def bearer_auth(request):
client_ip = request['client_ip']
if apikey is None:
return None
db = DBPools()
dbname = get_dbname()
async with db.sqlorContext(dbname) as sor:
user = await get_apikey_user(sor, apikey, client_ip)
await user_login(user.id, username=user.username, userorgid=user.orgid)
return user.id
return None
async def deerer_auth(request):
auth = request.headers.get('Authorization')
async def deerer_auth(sor, request):
auth = request.headers.get('Authentication')
if auth is None:
return None
if not auth.startswith('Deerer '):
@ -77,9 +73,6 @@ async def deerer_auth(request):
client_ip = request['client_ip']
deer_data = auth[7:]
appid, cyber = bear_data.split('-:-')
db = DBPools()
dbname = get_dbname()
async with db.sqlorContext(dbname) as sor:
secretkey = await get_secretkey(sor, appid)
txt = aes_decrypt_ecb(secretkey, cyber)
t, apikey = txt.split(':')
@ -87,8 +80,6 @@ async def deerer_auth(request):
await user_login(user.id, username=user.username, userorgid=user.orgid)
return user.id
return None
def return_error(code):
return {
'status':'error',

View File

@ -5,6 +5,6 @@ from rbac.check_perm import register_auth_method
def load_dapi():
env = ServerEnv()
env.sync_user = sync_user
register_auth_method('Bearer', bearer_auth)
register_auth_method('Deerer', deerer_auth)
register_auth_method('Bearer ', bearer_auth)
register_auth_method('Deerer ', deerer_auth)