# -*- coding: utf-8 -*- # vi:si:et:sw=4:sts=4:ts=4 import os 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 from ox.utils import json from ox.django.fields import DictField from app.models import site_config from itemlist.models import List, Position class UserProfile(models.Model): reset_token = models.TextField(blank=True, null=True, unique=True) user = models.ForeignKey(User, unique=True) files_updated = models.DateTimeField(default=datetime.now) newsletter = models.BooleanField(default=True) ui = DictField(default={}) preferences = DictField(default={}) def get_preferences(self): prefs = self.preferences prefs['email'] = self.user.email return prefs def get_ui(self): ui = {} config = site_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]) else: ui[key] = new[key] return ui ui = updateUI(ui, self.ui) if not 'lists' in ui: ui['lists'] = {} ui['lists'][''] = config['uiDefaults']['list'] 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=self.user) pos.save() created = True else: pos, created = Position.objects.get_or_create(list=l, user=self.user, section=section) qs = qs.filter(user=self.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 = [''] ids += add(self.user.lists.exclude(status="featured"), 'personal') ids += add(self.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 user_post_save(sender, instance, **kwargs): profile, new = UserProfile.objects.get_or_create(user=instance) models.signals.post_save.connect(user_post_save, sender=User) def get_user_json(user): profile = user.get_profile() result = {} for key in ('username', ): result[key] = getattr(user, key) result['admin'] = user.is_staff result['groups'] = [g.name for g in user.groups.all()] result['preferences'] = profile.get_preferences() result['ui'] = profile.get_ui() return result