mirror of
https://github.com/DMOJ/online-judge.git
synced 2024-11-25 16:32:37 +08:00
22 lines
783 B
Python
22 lines
783 B
Python
from django.utils.timezone import now
|
|
|
|
from judge.models import Profile
|
|
|
|
|
|
class LogUserAccessMiddleware(object):
|
|
def __init__(self, get_response=None):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
response = self.get_response(request)
|
|
|
|
if (hasattr(request, 'user') and request.user.is_authenticated and
|
|
not getattr(request, 'no_profile_update', False)):
|
|
updates = {'last_access': now()}
|
|
# Decided on using REMOTE_ADDR as nginx will translate it to the external IP that hits it.
|
|
if request.META.get('REMOTE_ADDR'):
|
|
updates['ip'] = request.META.get('REMOTE_ADDR')
|
|
Profile.objects.filter(user_id=request.user.pk).update(**updates)
|
|
|
|
return response
|