models.py 1.25 KB
Newer Older
Skia's avatar
Skia committed
1
2
3
from django.db import models
from django.utils.translation import ugettext_lazy as _
from django.conf import settings
Skia's avatar
Skia committed
4
from django.core.urlresolvers import reverse
Skia's avatar
Skia committed
5
6
7
8
9
10
11
12
13

from club.models import Club
from accounting.models import Product
from core.models import Group

class Counter(models.Model):
    name = models.CharField(_('name'), max_length=30)
    club = models.ForeignKey(Club, related_name="counters")
    products = models.ManyToManyField(Product, related_name="counters", blank=True)
Skia's avatar
Skia committed
14
15
    type = models.CharField(_('subscription type'),
            max_length=255,
Skia's avatar
Skia committed
16
            choices=[('BAR',_('Bar')), ('OFFICE',_('Office'))])
Skia's avatar
Skia committed
17
18
19
20
21
    edit_groups = models.ManyToManyField(Group, related_name="editable_counters", blank=True)
    view_groups = models.ManyToManyField(Group, related_name="viewable_counters", blank=True)

    def __getattribute__(self, name):
        if name == "owner_group":
Skia's avatar
Skia committed
22
            return Group(name=self.club.unix_name+settings.SITH_BOARD_SUFFIX)
Skia's avatar
Skia committed
23
24
25
26
        return object.__getattribute__(self, name)

    def __str__(self):
        return self.name
Skia's avatar
Skia committed
27
28
29
30
31

    def get_absolute_url(self):
        return reverse('counter:details', kwargs={'counter_id': self.id})

    def can_be_viewed_by(self, user):
Skia's avatar
Skia committed
32
        return user.is_in_group(settings.SITH_MAIN_BOARD_GROUP)