OpenSlides/openslides/system/models.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
    openslides.system.models
    ~~~~~~~~~~~~~~~~~~~~~~~~

    Models for the system app.

    :copyright: 2011 by the OpenSlides team, see AUTHORS.
    :license: GNU GPL, see LICENSE for more details.
"""
from pickle import dumps, loads

from django.db import models
from django.utils.translation import ugettext as _

DEFAULT_DATA  = {
    'event_name': 'OpenSlides',
    'event_description': 'Presentation and voting system',
    'agenda_countdown_time': 60,
    'application_min_supporters': 4,
    'application_preamble': 'Die Versammlung möge beschließen,',
    'application_pdf_ballot_papers_selection': '1',
    'application_pdf_title': _('Applications'),
    'assignment_pdf_ballot_papers_selection': '1',
    'assignment_pdf_title': _('Elections'),
    'system_url': 'http://127.0.0.1:8000',
    'system_welcometext': 'Welcome to OpenSlides!',
}

class ConfigStore(models.Model):
    key = models.CharField(max_length=100, primary_key=True)
    value = models.CharField(max_length=100)

    def __unicode__(self):
        return self.key

    class Meta:
        verbose_name = 'config'
        permissions = (
            ('can_manage_system', "Can manage system configuration"),
        )


class Config(object):
    def load_config(self):
        self.config = {}
        for key, value in ConfigStore.objects.all().values_list():
            self.config[key] = loads(str(value))

    def __getitem__(self, key):
        try:
            self.config
        except AttributeError:
            self.load_config()
        try:
            return self.config[key]
        except KeyError:
            pass
        try:
            return DEFAULT_DATA[key]
        except KeyError:
            return None

    def __setitem__(self, key, value):
        try:
            c = ConfigStore.objects.get(pk=key)
        except ConfigStore.DoesNotExist:
            c = ConfigStore(pk=key)
        c.value = dumps(value)
        c.save()
        self.config[key] = value