def process_response(self, request, response):
""" Read in the users IP address from the request object. If we
find a matching GeoIPRecord for it and that country code has an
active redirect then inject the HTML into the template to show
a lightbox to the user """
# If the user has the ignore cookie set and we're meant to respect
# it then just return the response already
if ('redirect_ignore' in request.COOKIES and self.ignore_cookie):
return response
# If we should only process on the homepage then return if we're on
# anything but that before continuing
if (self.homepage_only and request.path != '/'):
return response
# The following is for inplace debugging forcing a particular IP
if settings.DEBUG and self.DEBUG_IP:
inbound_ip = self.DEBUG_IP
else:
inbound_ip = request.META['REMOTE_ADDR']
self.user_code = GeoIPRecord.get_code(inbound_ip)
if (geo_setting('REDIRECT_ALL')
and self.user_code in geo_setting('REDIRECT_CODES')):
context = dict(incoming_country_code=self.user_code,
target_domain=self.REDIRECT_DOMAIN)
inject_data = render_to_string('geoip/base_redirect.html', context)
else:
inject_data = self.redirect_from_admin()
if '/admin' not in request.path and inject_data:
# Slight hack for now to avoid spoiling admin
if not self.ignore_paths or \
(self.ignore_paths and request.path not in self.ignore_paths):
# We'll only inject the redirect code if we are not set to
# process the IgnoreURL's or if its not in the ignore list
response.content = smart_str(response.content) + \
smart_str(inject_data)
return response
def process_response(self, request, response):
""" Read in the users IP address from the request object. If we
find a matching GeoIPRecord for it and that country code has an
active redirect then inject the HTML into the template to show
a lightbox to the user """
# If the user has the ignore cookie set and we're meant to respect
# it then just return the response already
if ('redirect_ignore' in request.COOKIES and self.ignore_cookie):
return response
# If we should only process on the homepage then return if we're on
# anything but that before continuing
if (self.homepage_only and request.path != '/'):
return response
# The following is for inplace debugging forcing a particular IP
if settings.DEBUG and self.DEBUG_IP:
inbound_ip = self.DEBUG_IP
else:
inbound_ip = request.META['REMOTE_ADDR']
self.user_code = GeoIPRecord.get_code(inbound_ip)
if(geo_setting('REDIRECT_ALL') and self.user_code in geo_setting('REDIRECT_CODES')):
context = dict(incoming_country_code=self.user_code, target_domain=self.REDIRECT_DOMAIN)
inject_data = render_to_string('geoip/base_redirect.html', context)
else:
inject_data = self.redirect_from_admin()
if '/admin' not in request.path and inject_data:
# Slight hack for now to avoid spoiling admin
if not self.ignore_paths or \
(self.ignore_paths and request.path not in self.ignore_paths):
# We'll only inject the redirect code if we are not set to
# process the IgnoreURL's or if its not in the ignore list
response.content = smart_str(response.content) + \
smart_str(inject_data)
return response
def redirect_from_admin(self):
redirect_list = IPRedirectEntry.objects.all()
ccodes = redirect_list.values_list('incoming_country_code', flat=True)
if self.user_code in ccodes:
code_index = list(ccodes).index(self.user_code) # use this to grab the template data
redirect_data = redirect_list[code_index] # in theory this is the right one
context = dict(redirect_data.__dict__)
inject_data = render_to_string('geoip/custom_redirect.html', context)
return inject_data
else:
if geo_setting('FAIL_ON_MISSING'):
raise NoGeoRedirectFound('Could not find a geo-redirect for this county in admin')
else:
return None
def redirect_from_admin(self):
redirect_list = IPRedirectEntry.objects.all()
ccodes = redirect_list.values_list('incoming_country_code', flat=True)
if self.user_code in ccodes:
code_index = list(ccodes).index(
self.user_code) # use this to grab the template data
redirect_data = redirect_list[
code_index] # in theory this is the right one
context = dict(redirect_data.__dict__)
inject_data = render_to_string('geoip/custom_redirect.html',
context)
return inject_data
else:
if geo_setting('FAIL_ON_MISSING'):
raise NoGeoRedirectFound(
'Could not find a geo-redirect for this county in admin')
else:
return None
def get_code(input_ip):
""" Takes a string IP address and returns the country code if it could
find one for it """
try:
input_decimal = GeoIPRecord.to_decimal(input_ip)
except InvalidDottedIP:
raise InvalidDottedIP
bounds = GeoIPRecord.objects.filter(start_decimal__lte=input_decimal,
end_decimal__gte=input_decimal)
# In theory we should only have 1 record here
try:
return bounds[0].country_code
except IndexError:
if settings.DEBUG and geo_setting('FAIL_ON_MISSING'):
raise NoGeoIPException
else:
pass
def get_code(input_ip):
""" Takes a string IP address and returns the country code if it could
find one for it """
try:
input_decimal = GeoIPRecord.to_decimal(input_ip)
except InvalidDottedIP:
raise InvalidDottedIP
bounds = GeoIPRecord.objects.filter(start_decimal__lte=input_decimal,
end_decimal__gte=input_decimal)
# In theory we should only have 1 record here
try:
return bounds[0].country_code
except IndexError:
if settings.DEBUG and geo_setting('FAIL_ON_MISSING'):
raise NoGeoIPException
else:
pass
def __init__(self):
# Pull in and grab our conf for this from various settings files
# (could probably drop the if/elses here)
if geo_setting('DEBUG_IP'):
self.DEBUG_IP = geo_setting('DEBUG_IP')
else:
self.DEBUG_IP = False
if geo_setting('REDIRECT_DOMAIN'):
self.REDIRECT_DOMAIN = geo_setting('REDIRECT_DOMAIN')
else:
self.REDIRECT_DOMAIN = None
if geo_setting('PROCESS_IGNORES'):
self.ignore_paths = IgnoreURL.objects.all()\
.values_list('url', flat=True)
else:
self.ignore_paths = None
self.homepage_only = geo_setting('HOMEPAGE_ONLY')
self.ignore_cookie = geo_setting('SET_IGNORE_COOKIE')
self.user_code = None
def __init__(self):
# Pull in and grab our conf for this from various settings files
# (could probably drop the if/elses here)
if geo_setting('DEBUG_IP'):
self.DEBUG_IP = geo_setting('DEBUG_IP')
else:
self.DEBUG_IP = False
if geo_setting('REDIRECT_DOMAIN'):
self.REDIRECT_DOMAIN = geo_setting('REDIRECT_DOMAIN')
else:
self.REDIRECT_DOMAIN = None
if geo_setting('PROCESS_IGNORES'):
self.ignore_paths = IgnoreURL.objects.all()\
.values_list('url', flat=True)
else:
self.ignore_paths = None
self.homepage_only = geo_setting('HOMEPAGE_ONLY')
self.ignore_cookie = geo_setting('SET_IGNORE_COOKIE')
self.user_code = None