mirror of
https://github.com/DMOJ/online-judge.git
synced 2024-11-25 16:32:37 +08:00
28 lines
759 B
Python
28 lines
759 B
Python
import pytz
|
|
from django.conf import settings
|
|
from django.db import connection
|
|
from django.utils import timezone
|
|
from django.utils.timezone import make_aware
|
|
|
|
|
|
class TimezoneMiddleware(object):
|
|
def __init__(self, get_response=None):
|
|
self.get_response = get_response
|
|
|
|
def get_timezone(self, request):
|
|
tzname = settings.DEFAULT_USER_TIME_ZONE
|
|
if request.profile:
|
|
tzname = request.profile.timezone
|
|
return pytz.timezone(tzname)
|
|
|
|
def __call__(self, request):
|
|
with timezone.override(self.get_timezone(request)):
|
|
return self.get_response(request)
|
|
|
|
|
|
def from_database_time(datetime):
|
|
tz = connection.timezone
|
|
if tz is None:
|
|
return datetime
|
|
return make_aware(datetime, tz)
|