cvat-ai--cvat
wehub-resource-sync
55 行
1.5 KiB
Python
55 行
1.5 KiB
Python
# Copyright (C) CVAT.ai Corporation
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from typing import Protocol
|
|
from uuid import uuid4
|
|
|
|
from django.conf import settings
|
|
from django.utils.timezone import now
|
|
|
|
|
|
class WithUUID(Protocol):
|
|
uuid: str
|
|
|
|
|
|
class RequestTrackingMiddleware:
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
@staticmethod
|
|
def _generate_id():
|
|
return str(uuid4())
|
|
|
|
def __call__(self, request):
|
|
request.uuid = self._generate_id()
|
|
response = self.get_response(request)
|
|
response.headers["X-Request-Id"] = request.uuid
|
|
|
|
return response
|
|
|
|
|
|
class LastActivityMiddleware:
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
response = self.get_response(request)
|
|
|
|
if request.user.is_authenticated:
|
|
profile = getattr(request.user, "profile", None)
|
|
if not profile:
|
|
return response
|
|
|
|
last_activity_date = profile.last_activity_date
|
|
if (
|
|
not last_activity_date
|
|
or (now() - last_activity_date) > settings.USER_LAST_ACTIVITY_UPDATE_MIN_INTERVAL
|
|
):
|
|
# such way we avoid failing and any db updates if the Profile was removed during the request
|
|
from cvat.apps.engine.models import Profile
|
|
|
|
Profile.objects.filter(user_id=request.user.id).update(last_activity_date=now())
|
|
|
|
return response
|