web2.0-backend/webapp/models.py

"""Webapp app models."""

from django.db import models
from django.utils import timezone
from datetime import timedelta
from django.contrib.auth.models import User
from webapp.utils import month_from_now
from django.utils.translation import ugettext_lazy as _
from django.contrib.auth.models import User
from auditlog.registry import auditlog
from phonenumber_field.modelfields import PhoneNumberField
from django.contrib.postgres.fields import JSONField

import logging


VERBOSE_NAME = _('Webapp')


class Tag(models.Model):
    """Model for tag."""

    slug = models.SlugField(primary_key=True)
    name = models.CharField(max_length=127)
    icon = models.ImageField()

    class Meta:
        verbose_name = _('Tag')
        verbose_name_plural = _('Tags')

    def __str__(self):
        return _('Tag: {}').format(self.slug)


class BaseFeed(models.Model):
    """Model containing something showing on some info feed."""

    tags = models.ManyToManyField(Tag, related_name="feeds", blank=True)
    visible = models.BooleanField(default=True)
    title = models.CharField(max_length=255)
    description = models.CharField(max_length=255)
    content = models.TextField()


class Feed(BaseFeed):
    """Model representing feed."""

    publish_time = models.DateTimeField(default=timezone.now)
    autohide = models.DateTimeField(default=month_from_now)

    def __str__(self):
        return _('Feed: {}').format(self.title)

    class Meta:
        verbose_name = _('Feed')
        verbose_name_plural = _('Feeds')


class Event(BaseFeed):
    """Model for event."""

    start_time = models.DateTimeField(default=timezone.now)
    end_time = models.DateTimeField(default=timezone.now)
    registration = models.ForeignKey(
        'Registration', on_delete=models.CASCADE, null=True)

    def __str__(self):
        return _('Event: {}').format(self.title)

    class Meta:
        verbose_name = _('Event')
        verbose_name_plural = _('Events')


class Registration(models.Model):
    """Model for event registration."""

    name = models.CharField(max_length=255)
    email = models.EmailField()
    options = JSONField()

    def __str__(self):
        return _('Registration: {}').format(self.name)

    class Meta:
        verbose_name = _('Registration')
        verbose_name_plural = _('Registrations')


class BaseRole(models.Model):
    """Base model for occupations/roles."""

    CATEGORIES = (
        ('corporate', _('Corporate affairs')),
        ('freshman', _('Freshmen')),
        ('international', _('International')),
        ('external', _('External affairs')),
        ('media', _('Media')),
        ('tech', _('Technology')),
        ('wellbeing', _('Wellbeing')),
        ('elepaja', _('Elepaja')),
        ('ceremonies', _('Ceremonies')),
        ('culture', _('Culture')),
        ('studies', _('Studies')),
        ('sosso', _('Sössö magazine')),
        ('alumni', _('Alumni relations')),
        ('others', _('Others')),
    )

    name = models.CharField(_('Name'), max_length=255)
    is_board = models.BooleanField(_('Board member'))
    category = models.CharField(_('Category'), choices=CATEGORIES, default='others', max_length=255)

    def __str__(self):
        n = self.name.capitalize()
        return '{} ({})'.format(n, _('board member')) if self.is_board else n


class PresetRole(BaseRole):
    """Model representing a preset occupation in the guild."""

    description = models.TextField(_('Description'))


class Role(PresetRole):
    """
    Model for Role.

    Model representing an active or historical occupation
    in an official's history.
    """

    class Meta:
        """Meta class for Role model."""

        verbose_name = _('Role')
        verbose_name_plural = _('Roles')

    start_date = models.DateField(_('Start date'))
    end_date = models.DateField(_('End date'))
    official = models.ForeignKey('Official', related_name='roles')


class Official(User):
    """Model representing a guild official."""

    class Meta:
        """Meta class for Official class."""

        verbose_name = _('Official')
        verbose_name_plural = _('Officials')

    phone_number = PhoneNumberField(_('Phone number'))


# Ohlhafv
class OhlhafvChallenge(models.Model):
    """Model containing all info about ohlhafv challenge."""

    class Meta:
        verbose_name = _('Ohlhafv challenge')
        verbose_name_plural = _('Ohlhafv challenges')

    SERIES_CHOICES = (
        ('0.33 L', '0.33 L'),
        ('0.5 L', '0.5 L'),
        ('1.0 L', '1.0 L'),
        ('Team', _('Team Challenge (1 x 0.33 L, 2 x 0.5 L, 1 x 1.0 L)'))
    )

    challenger = models.CharField(_('Challenger'), max_length=255)
    victim = models.CharField(_('Victim'), max_length=255)
    challenger_email = models.EmailField(_('Challenger email'))
    victim_email = models.EmailField(_('Victim email'))
    series = models.CharField(_('Series'), choices=SERIES_CHOICES, max_length=10)
    message = models.TextField(_('Message'), blank=True, null=False)

    def __str__(self):
        """Return model info."""
        return _('Ohlhafv challenge: {} vs. {}').format(self.challenger, self.victim)


auditlog.register(Tag)
auditlog.register(Feed)
auditlog.register(Event)
auditlog.register(PresetRole)
auditlog.register(Role)
auditlog.register(Official)
auditlog.register(OhlhafvChallenge)