middleware.py 662 B

12345678910111213141516171819
  1. from sessions import SessionCrawler, SessionHuman
  2. class SessionMiddleware(object):
  3. def process_request(self, request):
  4. try:
  5. if request.user.is_crawler():
  6. # Crawler Session
  7. request.session = SessionCrawler(request)
  8. except AttributeError:
  9. # Human Session
  10. request.session = SessionHuman(request)
  11. request.user = request.session.get_user()
  12. def process_response(self, request, response):
  13. try:
  14. request.session.save(request, response)
  15. except AttributeError:
  16. pass
  17. return response