You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
1013 B
Python
29 lines
1013 B
Python
5 years ago
|
from django.shortcuts import HttpResponseRedirect, reverse
|
||
|
|
from core.models import UserShippingAddress
|
||
|
|
|
||
|
|
|
||
|
|
class EnforceShippingAddressCreationMiddleware(object):
|
||
|
|
def __init__(self, get_response):
|
||
|
|
self.get_response = get_response
|
||
|
|
|
||
|
|
def __call__(self, request):
|
||
|
|
|
||
|
|
# If current user is authenticated, get their shipping information and current page
|
||
|
|
# If current page is not them editing their address or logging out, redirect them
|
||
|
|
if request.user.is_authenticated:
|
||
|
|
profile = UserShippingAddress.objects.filter(user=request.user).first()
|
||
|
|
is_profile_absent = profile is None
|
||
|
|
|
||
|
|
allowed_paths = [
|
||
|
|
reverse('edit_shipping'),
|
||
|
|
reverse('logout')
|
||
|
|
]
|
||
|
|
on_allowed_path = request.path not in allowed_paths
|
||
|
|
|
||
|
|
if is_profile_absent and on_allowed_path:
|
||
|
|
return HttpResponseRedirect(reverse('edit_shipping'))
|
||
|
|
|
||
|
|
response = self.get_response(request)
|
||
|
|
|
||
|
|
return response
|