UserCookieMiddleWare

Complexity

Total Complexity

8

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

3 Methods

Rating	Name	Duplication	Size	Complexity
A	process_response()	0	11	4
A	get_client_ip()	0	7	2
A	process_request()	0	7	2

import random

from django.conf import settings

api_cookie_name = settings.WS_ADDRESS_COOKIE_NAME
api_port = settings.API_PORT
from chat import local


class UserCookieMiddleWare(object):
	"""
	Middleware to set user cookie
	If user is authenticated and there is no cookie, set the cookie,
	If the user is not authenticated and the cookie remains, delete it
	"""

	def process_response(self, request, response):

		if not request.COOKIES.get(api_cookie_name):
			domain_address = request.get_host().split(':')[0]
			api_address = "ws://%s:%s/" % (domain_address, api_port)
			response.set_cookie(api_cookie_name, api_address)
		# force create Session for annon
		if hasattr(request, 'session') and not request.session.session_key:
			request.session.save()
			request.session.modified = True
		return response

	def process_request(self, request):
		try:
			local.random
		except AttributeError:
			local.random = str(random.randint(0, 10000)).rjust(4, '0')
			local.user = str(getattr(request.user, 'username', '')).rjust(8, ' ')
			local.client_ip = self.get_client_ip(request)

	def get_client_ip(self, request):
		x_forwarded_for = request.META.get('HTTP_X_FORWARDED_FOR')
		if x_forwarded_for:
			ip = x_forwarded_for.split(',')[-1].strip()
		else:
			ip = request.META.get('REMOTE_ADDR')
		return ip