def create(self, request, user=None):
self._user = user
while True:
try:
self._session_key = self._get_new_session_key()
self._session_rk = Session(
id=self._session_key,
data=self.encode({}),
user=self._user,
ip=self._ip,
agent=request.META.get('HTTP_USER_AGENT', ''),
start=timezone.now(),
last=timezone.now(),
admin=request.firewall.admin,
)
self._session_rk.save(force_insert=True)
if settings.USER_EXTENSIONS_PRELOAD:
self._session_rk = self._session_rk.select_related(*settings.USER_EXTENSIONS_PRELOAD)
if user:
# Update user data
user.set_last_visit(
self.get_ip(request),
request.META.get('HTTP_USER_AGENT', '')
)
user.save(force_update=True)
break
except CreateError:
# Key wasn't unique. Try again.
continue
class CrawlerSession(MisagoSession):
"""
Crawler Session controller
"""
def __init__(self, request):
self.serializer = import_by_path(settings.SESSION_SERIALIZER)
self.matched = True
self.started = False
self.team = False
self._ip = self.get_ip(request)
self._session_key = md5('%s-%s' %
(request.user.username, self._ip)).hexdigest()
try:
self._session_rk = Session.objects.get(id=self._session_key)
self._session_key = self._session_rk.id
except Session.DoesNotExist:
self.create(request)
def create(self, request):
self._session_rk = Session(id=self._session_key,
data=self.encode({}),
crawler=request.user.username,
ip=self._ip,
agent=request.META.get(
'HTTP_USER_AGENT', ''),
start=timezone.now(),
last=timezone.now(),
matched=True)
while True:
try:
self._session_rk.save(force_insert=True)
break
except CreateError:
continue
except IntegrityError:
try:
self._session_rk = Session.objects.get(
id=self._session_key)
except Session.DoesNotExist:
continue
def human_session(self):
return False
class CrawlerSession(MisagoSession):
"""
Crawler Session controller
"""
def __init__(self, request):
self.serializer = import_by_path(settings.SESSION_SERIALIZER)
self.matched = True
self.started = False
self.team = False
self._ip = self.get_ip(request)
self._session_key = md5('%s-%s' % (request.user.username, self._ip)).hexdigest()
try:
self._session_rk = Session.objects.get(id=self._session_key)
self._session_key = self._session_rk.id
except Session.DoesNotExist:
self.create(request)
def create(self, request):
self._session_rk = Session(
id=self._session_key,
data=self.encode({}),
crawler=request.user.username,
ip=self._ip,
agent=request.META.get('HTTP_USER_AGENT', ''),
start=timezone.now(),
last=timezone.now(),
matched=True
)
while True:
try:
self._session_rk.save(force_insert=True)
break
except CreateError:
continue
except IntegrityError:
try:
self._session_rk = Session.objects.get(id=self._session_key)
except Session.DoesNotExist:
continue
def human_session(self):
return False
def create(self, request):
self._session_rk = Session(
id=self._session_key,
data=self.encode({}),
crawler=request.user.username,
ip=self._ip,
agent=request.META.get('HTTP_USER_AGENT', ''),
start=timezone.now(),
last=timezone.now(),
matched=True
)
while True:
try:
self._session_rk.save(force_insert=True)
break
except CreateError:
continue
except IntegrityError:
try:
self._session_rk = Session.objects.get(id=self._session_key)
except Session.DoesNotExist:
continue
def create(self, request):
self._session_rk = Session(
id=self._session_key,
data=self.encode({}),
crawler=request.user.username,
ip=self._ip,
agent=request.META.get('HTTP_USER_AGENT', ''),
start=timezone.now(),
last=timezone.now(),
matched=True
)
while True:
try:
self._session_rk.save(force_insert=True)
break
except CreateError:
try:
self._session_rk = Session.objects.get(id=self._session_key)
except Session.DoesNotExist:
continue
class HumanSession(MisagoSession):
"""
Human Session controller
"""
def __init__(self, request):
self.started = False
self.matched = False
self.expired = False
self.team = False
self.rank = None
self.remember_me = None
self._user = None
self._ip = self.get_ip(request)
self._session_token = None
if request.firewall.admin:
self._cookie_sid = settings.COOKIES_PREFIX + 'ASID'
else:
self._cookie_sid = settings.COOKIES_PREFIX + 'SID'
try:
# Do we have correct session ID?
if self._cookie_sid not in request.COOKIES or len(request.COOKIES[self._cookie_sid]) != 42:
raise IncorrectSessionException()
self._session_key = request.COOKIES[self._cookie_sid]
self._session_rk = Session.objects.select_related('user', 'rank')
if settings.USER_EXTENSIONS_PRELOAD:
self._session_rk = self._session_rk.select_related(*settings.USER_EXTENSIONS_PRELOAD)
self._session_rk = self._session_rk.get(
pk=self._session_key,
admin=request.firewall.admin
)
# IP invalid
if request.settings.sessions_validate_ip and self._session_rk.ip != self._ip:
raise IncorrectSessionException()
# Session expired
if timezone.now() - self._session_rk.last > timedelta(seconds=settings.SESSION_LIFETIME):
self.expired = True
raise IncorrectSessionException()
# Change session to matched and extract session user
if self._session_rk.matched:
self.matched = True
else:
self.started = True
self._user = self._session_rk.user
self.team = self._session_rk.team
except (Session.DoesNotExist, IncorrectSessionException):
# Attempt autolog
try:
self.remember_me = auth_remember(request, self.get_ip(request))
self.create(request, user=self.remember_me.user)
self.started = True
self._session_rk.matched = True
except AuthException as e:
# Autolog failed
self.create(request)
self.id = self._session_rk.id
# Make cookie live longer
if request.firewall.admin:
request.cookiejar.set('ASID', self._session_rk.id)
else:
request.cookiejar.set('SID', self._session_rk.id)
def create(self, request, user=None):
self._user = user
while True:
try:
self._session_key = self._get_new_session_key()
self._session_rk = Session(
id=self._session_key,
data=self.encode({}),
user=self._user,
ip=self._ip,
agent=request.META.get('HTTP_USER_AGENT', ''),
start=timezone.now(),
last=timezone.now(),
admin=request.firewall.admin,
)
self._session_rk.save(force_insert=True)
if settings.USER_EXTENSIONS_PRELOAD:
self._session_rk = self._session_rk.select_related(*settings.USER_EXTENSIONS_PRELOAD)
if user:
# Update user data
user.set_last_visit(
self.get_ip(request),
request.META.get('HTTP_USER_AGENT', '')
)
user.save(force_update=True)
break
except CreateError:
# Key wasn't unique. Try again.
continue
def save(self):
self._session_rk.user = self._user
self._session_rk.team = self.team
self._session_rk.rank_id = self.rank
super(HumanSession, self).save()
def human_session(self):
return True
def session_expired(self):
return self.expired
def get_user(self):
if self._user == None:
return Guest()
return self._user
def set_user(self, user=None):
self._user = user
def sign_out(self, request):
try:
if self._user.is_authenticated():
if not request.firewall.admin:
cookie_token = settings.COOKIES_PREFIX + 'TOKEN'
if cookie_token in request.COOKIES:
if len(request.COOKIES[cookie_token]) > 0:
Token.objects.filter(id=request.COOKIES[cookie_token]).delete()
request.cookiejar.delete('TOKEN')
self._user = None
request.user = Guest()
except AttributeError:
pass
