mirror of
https://github.com/FunnyWolf/agentic-soc-platform.git
synced 2026-08-22 13:12:56 +02:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import datetime
|
|
|
|
from rest_framework import exceptions
|
|
from rest_framework.authentication import TokenAuthentication
|
|
|
|
from Lib.configs import EXPIRE_MINUTES
|
|
from Lib.xcache import Xcache
|
|
|
|
|
|
class BaseAuth(TokenAuthentication):
|
|
def authenticate_credentials(self, key=None):
|
|
# search cached user token
|
|
cache_user = Xcache.alive_token(key)
|
|
if cache_user:
|
|
return cache_user, key
|
|
|
|
# search user token in database
|
|
model = self.get_model()
|
|
try:
|
|
token = model.objects.select_related('user').get(key=key)
|
|
except model.DoesNotExist:
|
|
raise exceptions.AuthenticationFailed()
|
|
|
|
if not token.user.is_active:
|
|
raise exceptions.AuthenticationFailed()
|
|
|
|
# token timeout clean
|
|
time_now = datetime.datetime.now()
|
|
if token.created < time_now - datetime.timedelta(minutes=EXPIRE_MINUTES):
|
|
token.delete()
|
|
raise exceptions.AuthenticationFailed()
|
|
|
|
# cache token
|
|
if token:
|
|
Xcache.set_token_user(key, token.user)
|
|
return token.user, token
|