Kod źródłowy modułu watchdog_kj_kultura.organizations.models

from autoslug.fields import AutoSlugField
from django.conf import settings
from django.contrib.postgres.fields import JSONField
from django.core.urlresolvers import reverse
from django.db import models
from django.db.models import Case, When
from django.utils.encoding import python_2_unicode_compatible
from django.utils.translation import ugettext_lazy as _
from djgeojson.fields import PointField
from model_utils.models import TimeStampedModel
from teryt_tree.models import JednostkaAdministracyjna

from .validators import is_allnum


class MetaCategoryQuerySet(models.QuerySet):
    pass


@python_2_unicode_compatible
class MetaCategory(TimeStampedModel):
    name = models.CharField(verbose_name=_("Name"), max_length=50)
    key = models.CharField(verbose_name=_("Key"),
                           max_length=50,
                           validators=[is_allnum],
                           help_text=_("They are permitted only Latin characters and numbers."))
    user = models.ForeignKey(settings.AUTH_USER_MODEL)
    objects = MetaCategoryQuerySet.as_manager()

    class Meta:
        verbose_name = _("MetaCategory")
        verbose_name_plural = _("MetaCategorys")
        ordering = ['pk', ]

    def __str__(self):
        return self.name


class CategoryQuerySet(models.QuerySet):

    def with_organization_count(self):
        return self.annotate(organization_count=models.Count('organization'))


@python_2_unicode_compatible
class Category(TimeStampedModel):
    name = models.CharField(verbose_name=_("Name"), max_length=50)
    slug = AutoSlugField(populate_from='name', verbose_name=_("Slug"), unique=True)
    objects = CategoryQuerySet.as_manager()

    class Meta:
        verbose_name = _("Category")
        verbose_name_plural = _("Categories")
        ordering = ['created', ]

    def __str__(self):
        return self.name

    def get_absolute_url(self):
        return reverse('organizations:list', kwargs={'category': self.slug})


class OrganizationQuerySet(models.QuerySet):

    def area(self, jst):
        return self.filter(jst__tree_id=jst.tree_id,
                           jst__lft__range=(jst.lft, jst.rght))

    def switch_visibility(self):
        return self.update(visible=Case(When(visible=True, then=False),
                                        When(visible=False, then=True),
                                        default=False))

    def with_jst_tree_selected(self):
        return self.select_related('jst__parent',
                                   'jst__parent__parent',
                                   'jst__parent__parent__parent')

    def visible(self):
        return self.filter(visible=True)


@python_2_unicode_compatible
class Organization(TimeStampedModel):
    name = models.CharField(verbose_name=_("Name"), max_length=50)
    slug = AutoSlugField(populate_from='name', verbose_name=_("Slug"), unique=True)
    email = models.EmailField(verbose_name=_("E-mail"))
    jst = models.ForeignKey(JednostkaAdministracyjna,
                            verbose_name=_("Unit of administrative division"))
    user = models.ForeignKey(settings.AUTH_USER_MODEL)
    pos = PointField(verbose_name=_("Position"), null=True, blank=True)
    category = models.ForeignKey(to=Category,
                                 verbose_name=_("Category"),
                                 null=True,
                                 blank=True)
    visible = models.BooleanField(verbose_name=_("Public visible"),
                                  help_text=_("Check to mark organization as public visible"),
                                  default=False)
    meta = JSONField(verbose_name=_("Metadata"), default={})
    objects = OrganizationQuerySet.as_manager()

    def geocode_input(self):
        if self.pos:
            return "{0}, {1}".format(self.pos.coords[1], self.pos.coords[0])
        return self.name or "Warszawa, mazowieckie, Polska"

    def set_geopy_point(self, point):
        self.pos = {'type': 'Point',
                    'coordinates': [point.point.longitude, point.point.latitude]}

    def jst_list(self):
        r = [self.jst.pk]
        jst = self.jst
        while jst.parent:
            r = r + [jst.parent_id, ]
            jst = jst.parent
        return r

    class Meta:
        verbose_name = _("Organization")
        verbose_name_plural = _("Organizations")
        ordering = ['created', ]

    def __str__(self):
        return self.name

    def get_absolute_url(self):
        return reverse('organizations:details', kwargs={'slug': self.slug})

    @property
    def absolute_url(self):
        return self.get_absolute_url()