Advertisement
Guest User

Untitled

a guest
Jan 16th, 2014
117
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.62 KB | None | 0 0
  1. class Middleware(object):
  2.     refcode = None
  3.  
  4.     def process_request(self, request):
  5.         if 'ref' not in request.COOKIES and 'ref' in request.GET:
  6.             refcode = request.GET.get('ref', None)
  7.             try:
  8.                 object = Model.objects.get(code=refcode)
  9.                 self.refcode = refcode
  10.             except Model.DoesNotExist:
  11.                 pass
  12.         return None
  13.  
  14.     def process_response(self, request, response):
  15.         if self.refcode is not None:
  16.             # expires in 30 days
  17.             response.set_cookie('ref', self.refcode, max_age=60 * 60 * 24 * 30)
  18.         return response
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement