230 lines
8 KiB
Python
230 lines
8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
import copy
|
|
from datetime import datetime
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.db import models
|
|
from django.db.models import Max
|
|
from django.conf import settings
|
|
|
|
import ox
|
|
from ox.django.fields import DictField
|
|
from ox.utils import json
|
|
|
|
from itemlist.models import List, Position
|
|
|
|
import managers
|
|
|
|
class SessionData(models.Model):
|
|
session_key = models.CharField(max_length=40, primary_key=True)
|
|
user = models.ForeignKey(User, unique=True, null=True, blank=True, related_name='data')
|
|
firstseen = models.DateTimeField(auto_now_add=True, db_index=True)
|
|
lastseen = models.DateTimeField(default=datetime.now, db_index=True)
|
|
username = models.CharField(max_length=255, null=True, db_index=True)
|
|
level = models.IntegerField(default=0)
|
|
|
|
timesseen = models.IntegerField(default=0)
|
|
ip = models.CharField(default='', max_length=255)
|
|
useragent = models.CharField(default='', max_length=255)
|
|
windowsize = models.CharField(default='', max_length=255)
|
|
screensize = models.CharField(default='', max_length=255)
|
|
info = DictField(default={})
|
|
|
|
|
|
objects = managers.SessionDataManager()
|
|
|
|
def __unicode__(self):
|
|
return u"%s" % self.session_key
|
|
|
|
def save(self, *args, **kwargs):
|
|
if self.user:
|
|
self.username = self.user.username
|
|
self.level = self.user.get_profile().level
|
|
self.firstseen = self.user.date_joined
|
|
else:
|
|
self.level = 0
|
|
super(SessionData, self).save(*args, **kwargs)
|
|
|
|
@classmethod
|
|
def get_or_create(cls, request):
|
|
session_key = request.session.session_key
|
|
if request.user.is_authenticated():
|
|
cls.objects.filter(user=request.user).update(session_key=session_key)
|
|
data, created = cls.objects.get_or_create(session_key=session_key)
|
|
if request.user.is_authenticated():
|
|
data.user = request.user
|
|
data.info = json.loads(request.POST.get('data', '{}'))
|
|
screen = data.info.get('screen', {})
|
|
if 'height' in screen and 'width' in screen:
|
|
data.screensize = '%sx%s' % (screen['width'], screen['height'])
|
|
window = data.info.get('window', {})
|
|
if 'outerHeight' in window and 'outerWidth' in window:
|
|
data.windowsize = '%sx%s' % (window['outerWidth'], window['outerHeight'])
|
|
data.ip = request.META['REMOTE_ADDR']
|
|
data.useragent = request.META['HTTP_USER_AGENT']
|
|
if not data.timesseen:
|
|
data.timesseen = 0
|
|
data.timesseen += 1
|
|
data.lastseen = datetime.now()
|
|
data.save()
|
|
return data
|
|
|
|
def get_id(self):
|
|
return self.user and ox.to26(self.user.id) or self.session_key
|
|
|
|
def json(self, keys=None, user=None):
|
|
j = {
|
|
'disabled': False,
|
|
'email': '',
|
|
'firstseen': self.firstseen,
|
|
'ip': self.ip,
|
|
'id': self.get_id(),
|
|
'lastseen': self.lastseen,
|
|
'level': 'guest',
|
|
'notes': '',
|
|
'numberoflists': 0,
|
|
'screensize': self.screensize,
|
|
'timesseen': self.timesseen,
|
|
'username': self.username or '',
|
|
'useragent': self.useragent,
|
|
'windowsize': self.windowsize,
|
|
}
|
|
if self.user:
|
|
p = self.user.get_profile()
|
|
j['disabled'] = not self.user.is_active
|
|
j['email'] = self.user.email
|
|
j['level'] = p.get_level()
|
|
j['notes'] = p.notes
|
|
j['numberoflists'] = self.user.lists.count()
|
|
if keys:
|
|
for key in j.keys():
|
|
if key not in keys:
|
|
del j[key]
|
|
return j
|
|
|
|
class UserProfile(models.Model):
|
|
reset_code = models.CharField(max_length=255, blank=True, null=True, unique=True)
|
|
user = models.ForeignKey(User, unique=True, related_name='profile')
|
|
|
|
level = models.IntegerField(default=1)
|
|
files_updated = models.DateTimeField(default=datetime.now)
|
|
newsletter = models.BooleanField(default=True)
|
|
ui = DictField(default={})
|
|
preferences = DictField(default={})
|
|
|
|
notes = models.TextField(default='')
|
|
|
|
def get_ui(self):
|
|
return get_ui(self.ui, self.user)
|
|
|
|
def set_level(self, level):
|
|
self.level = settings.CONFIG['userLevels'].index(level)
|
|
|
|
def get_level(self):
|
|
return settings.CONFIG['userLevels'][self.level]
|
|
|
|
def capability(self, capability):
|
|
return settings.CONFIG['capabilities'][capability].get(self.get_level()) == True
|
|
|
|
def user_post_save(sender, instance, **kwargs):
|
|
profile, new = UserProfile.objects.get_or_create(user=instance)
|
|
SessionData.objects.filter(user=instance).update(level=profile.level,
|
|
username=instance.username)
|
|
models.signals.post_save.connect(user_post_save, sender=User)
|
|
|
|
def profile_post_save(sender, instance, **kwargs):
|
|
SessionData.objects.filter(user=instance.user).update(level=instance.level,
|
|
username=instance.user.username)
|
|
models.signals.post_save.connect(profile_post_save, sender=UserProfile)
|
|
|
|
def get_ui(user_ui, user=None):
|
|
ui = {}
|
|
config = copy.deepcopy(settings.CONFIG)
|
|
ui.update(config['user']['ui'])
|
|
def updateUI(ui, new):
|
|
'''
|
|
only update set keys in dicts
|
|
'''
|
|
for key in new:
|
|
if isinstance(new[key], dict) and key in ui:
|
|
ui[key] = updateUI(ui[key], new[key])
|
|
elif isinstance(ui, dict):
|
|
ui[key] = new[key]
|
|
return ui
|
|
ui = updateUI(ui, user_ui)
|
|
if not 'lists' in ui:
|
|
ui['lists'] = {}
|
|
|
|
def add(lists, section):
|
|
ids = []
|
|
for l in lists:
|
|
qs = Position.objects.filter(section=section)
|
|
if section == 'featured':
|
|
try:
|
|
pos = Position.objects.get(list=l, section=section)
|
|
created = False
|
|
except Position.DoesNotExist:
|
|
pos = Position(list=l, section=section, user=user)
|
|
pos.save()
|
|
created = True
|
|
else:
|
|
pos, created = Position.objects.get_or_create(list=l, user=user, section=section)
|
|
qs = qs.filter(user=user)
|
|
if created:
|
|
pos.position = qs.aggregate(Max('position'))['position__max'] + 1
|
|
pos.save()
|
|
id = l.get_id()
|
|
'''
|
|
if id not in ui['lists']:
|
|
ui['lists'][id] = {}
|
|
ui['lists'][id].update(ui['lists'][''])
|
|
'''
|
|
ids.append(id)
|
|
return ids
|
|
|
|
ids = ['']
|
|
if user:
|
|
ids += add(user.lists.exclude(status="featured"), 'personal')
|
|
ids += add(user.subscribed_lists.filter(status='public'), 'public')
|
|
ids += add(List.objects.filter(status='featured'), 'featured')
|
|
for i in ui['lists'].keys():
|
|
if i not in ids:
|
|
del ui['lists'][i]
|
|
return ui
|
|
|
|
def init_user(user, request=None):
|
|
data = SessionData.get_or_create(request)
|
|
if user.is_anonymous():
|
|
result = settings.CONFIG['user'].copy()
|
|
result['ui'] = get_ui(json.loads(request.session.get('ui', '{}')))
|
|
else:
|
|
profile = user.get_profile()
|
|
result = {}
|
|
for key in ('username', ):
|
|
result[key] = getattr(user, key)
|
|
result['level'] = profile.get_level()
|
|
result['groups'] = [g.name for g in user.groups.all()]
|
|
result['email'] = user.email
|
|
result['ui'] = profile.get_ui()
|
|
result['volumes'] = [v.json() for v in user.volumes.all()]
|
|
return result
|
|
|
|
def user_json(user, keys=None):
|
|
p = user.get_profile()
|
|
j = {
|
|
'disabled': not user.is_active,
|
|
'email': user.email,
|
|
'firstseen': user.date_joined,
|
|
'id': ox.to26(user.id),
|
|
'lastseen': user.last_login,
|
|
'level': p.get_level(),
|
|
'notes': p.notes,
|
|
'numberoflists': user.lists.count(),
|
|
'username': user.username,
|
|
}
|
|
if keys:
|
|
for key in j.keys():
|
|
if key not in keys:
|
|
del j[key]
|
|
return j
|