47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
import uuid
|
||
from gvsdsdk.auth.cookie import ELT_COOKIE_NAME
|
||
|
||
|
||
class ELTAuthentication:
|
||
def authenticate(self, request):
|
||
elt_token = request.COOKIES.get(ELT_COOKIE_NAME)
|
||
if not elt_token:
|
||
return None
|
||
|
||
try:
|
||
token_uuid = uuid.UUID(elt_token)
|
||
except (ValueError, AttributeError):
|
||
return None
|
||
|
||
from gvsdsdk.models import User, EphemeralToken
|
||
import datetime
|
||
|
||
try:
|
||
elt = EphemeralToken.objects.get(
|
||
ELTUUID=token_uuid,
|
||
ELTStatus=1,
|
||
)
|
||
except EphemeralToken.DoesNotExist:
|
||
return None
|
||
|
||
now_utc = datetime.datetime.now(datetime.timezone.utc)
|
||
if elt.ELTEndTime.replace(tzinfo=None) < now_utc.replace(tzinfo=None):
|
||
return None
|
||
|
||
# 滑动续期：剩余时间不足7天时，延长30天
|
||
remaining = elt.ELTEndTime.replace(tzinfo=None) - now_utc.replace(tzinfo=None)
|
||
if remaining < datetime.timedelta(days=7):
|
||
elt.ELTEndTime = now_utc + datetime.timedelta(days=30)
|
||
elt.save(update_fields=['ELTEndTime'])
|
||
|
||
try:
|
||
user = User.objects.get(UserUUID=elt.ELTUserUUID)
|
||
except User.DoesNotExist:
|
||
return None
|
||
|
||
request.elt = elt
|
||
return (user, elt_token)
|
||
|
||
def authenticate_header(self, request):
|
||
return 'Bearer'
|