bornhack-website/villages/models.py

from __future__ import unicode_literals

from django.core.urlresolvers import reverse_lazy
from django.db import models
from django.utils.text import slugify

from utils.models import CreatedUpdatedModel, UUIDModel

from .managers import VillageQuerySet


class Village(CreatedUpdatedModel, UUIDModel):

    class Meta:
        ordering = ['name']

    contact = models.ForeignKey('auth.User')
    camp = models.ForeignKey('camps.Camp')
    name = models.CharField(max_length=255)
    slug = models.SlugField(max_length=255, blank=True)
    description = models.TextField(
        help_text="A descriptive text about your village. Markdown is supported."
    )

    private = models.BooleanField(
        default=False,
        help_text='Check if your village is invite only. Leave unchecked to welcome strangers.'
    )

    deleted = models.BooleanField(
        default=False,
    )

    objects = VillageQuerySet.as_manager()

    def __str__(self):
        return self.name

    def get_absolute_url(self):
        return reverse_lazy('village_detail', kwargs={'camp_slug': self.camp.slug, 'slug': self.slug})

    def save(self, **kwargs):
        if (
            not self.pk or
            not self.slug or
            Village.objects.filter(slug=self.slug).count() > 1
        ):
            slug = slugify(self.name)
            incrementer = 1

            # We have to make sure that the slug won't clash with current slugs
            while Village.objects.filter(slug=slug).exists():
                if incrementer == 1:
                    slug = '{}-1'.format(slug)
                else:
                    slug = '{}-{}'.format(
                        '-'.join(slug.split('-')[:-1]),
                        incrementer
                    )
                incrementer += 1
            self.slug = slug

        super(Village, self).save(**kwargs)

    def delete(self, using=None, keep_parents=False):
        self.deleted = True
        self.save()