Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Middleware(object):
- refcode = None
- def process_request(self, request):
- if 'ref' not in request.COOKIES and 'ref' in request.GET:
- refcode = request.GET.get('ref', None)
- try:
- object = Model.objects.get(code=refcode)
- self.refcode = refcode
- except Model.DoesNotExist:
- pass
- return None
- def process_response(self, request, response):
- if self.refcode is not None:
- # expires in 30 days
- response.set_cookie('ref', self.refcode, max_age=60 * 60 * 24 * 30)
- return response
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement