..
Viewing
tz.py
32 lines (25 loc) • 1.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32 | import requests
# make sure you add `TimezoneMiddleware` appropriately in settings.py
class TimezoneMiddleware(object):
"""
Middleware to properly handle the users timezone
"""
def __init__(self, get_response):
self.get_response = get_response
def __call__(self, request):
# get the user's timezone from the cookie
user_timezone = request.COOKIES.get('user_timezone')
remote_ip = request.META.get('REMOTE_ADDR')
if not user_timezone:
x_forwarded_for = request.META.get('HTTP_X_FORWARDED_FOR')
if x_forwarded_for:
remote_ip = x_forwarded_for.split(',')[0]
else:
remote_ip = request.META.get('REMOTE_ADDR')
geo_data = requests.get(f'http://ip-api.com/json/{remote_ip}').json()
user_timezone = geo_data['timezone'] if 'timezone' in geo_data else 'UTC'
if user_timezone:
response = self.get_response(request)
response.set_cookie('user_timezone', user_timezone)
return response
return self.get_response(request)
|
|