bornhack-website/villages/models.py

69 lines
1.9 KiB
Python
Raw Normal View History

2016-07-05 21:43:18 +00:00
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
2016-07-05 21:43:18 +00:00
class Village(CreatedUpdatedModel, UUIDModel):
class Meta:
ordering = ['name']
contact = models.ForeignKey('auth.User')
2016-12-28 23:15:13 +00:00
camp = models.ForeignKey('camps.Camp')
2016-07-05 21:43:18 +00:00
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."
)
2016-07-05 21:43:18 +00:00
2016-07-05 21:55:05 +00:00
private = models.BooleanField(
2016-07-05 21:58:46 +00:00
default=False,
help_text='Check if your village is invite only. Leave unchecked to welcome strangers.'
2016-07-05 21:43:18 +00:00
)
deleted = models.BooleanField(
default=False,
)
objects = VillageQuerySet.as_manager()
2016-07-05 21:43:18 +00:00
def __str__(self):
return self.name
def get_absolute_url(self):
2016-12-28 23:15:13 +00:00
return reverse_lazy('village_detail', kwargs={'camp_slug': self.camp.slug, 'slug': self.slug})
2016-07-05 21:43:18 +00:00
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()