cablegates/pandora/user/models.py

# -*- 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.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


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 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):
    result = {}
    for key in ('username', ):
        result[key] = getattr(user, key)
    result['group'] = 'user'
    if user.is_staff:
        result['group'] = 'admin'
    elif user.has_perm('0x.vip'): #FIXME: permissions
        result['group'] = 'vip'
    result['preferences'] = get_preferences(user)
    result['ui'] = get_ui(user)
    return result


def get_ui(user):
    return site_config['user']['ui']
    '''
    return {
        "columns": ["id", "title", "director", "country", "year", "language", "genre"],
        "findQuery": {"conditions": [], "operator": ""},
        "groupsQuery": {"conditions": [], "operator": "|"},
        "groupsSize": 128,
        "itemView": "timeline",
        "listQuery": {"conditions": [], "operator": ""},
        "listsSize": 192,
        "listView": "icons",
        "sections": ["history", "lists", "public", "featured"],
        "showGroups": True,
        "showInfo": True,
        "showLists": True,
        "showMovies": True,
        "sort": settings.DEFAULT_SORT,
        "theme": settings.DEFAULT_THEME
    }
    '''


def get_preferences(user):
    prefs = user.get_profile().preferences
    prefs['email'] = user.email
    return prefs


def get_preference(user, key, value=None):
    if key in ('email', ):
        value = getattr(user, key)
    else:
        q = Preference.objects.filter(user=user, key=key)
        if q.count() > 0:
            value = json.loads(q[0].value)
    return value


def set_preference(user, key, value):
    if key in ('email', ):
        setattr(user, key, value)
        user.save()
    else:
        value = json.dumps(value)
        q = Preference.objects.filter(user=user, key=key)
        if q.count() > 0:
            p = q[0]
            p.value = value
            p.save()
        else:
            p = Preference(user=user, key=key, value=value)
            p.save()