129 lines
3.1 KiB
Python
129 lines
3.1 KiB
Python
import datetime
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from utils.models import UUIDModel, CreatedUpdatedModel
|
|
|
|
from .managers import CampQuerySet
|
|
|
|
|
|
class Camp(CreatedUpdatedModel, UUIDModel):
|
|
class Meta:
|
|
verbose_name = _('Camp')
|
|
verbose_name_plural = _('Camps')
|
|
|
|
name = models.CharField(
|
|
verbose_name=_('Name'),
|
|
help_text=_('Name of the camp, ie. Bornhack.'),
|
|
max_length=255,
|
|
)
|
|
|
|
start = models.DateTimeField(
|
|
verbose_name=_('Start date'),
|
|
help_text=_('When the camp starts.'),
|
|
unique=True,
|
|
)
|
|
|
|
end = models.DateTimeField(
|
|
verbose_name=_('End date'),
|
|
help_text=_('When the camp ends.'),
|
|
unique=True,
|
|
)
|
|
|
|
shop_open = models.BooleanField(
|
|
verbose_name=_('Shop open?'),
|
|
help_text=_('Whether the shop is open or not.'),
|
|
default=False,
|
|
)
|
|
|
|
objects = CampQuerySet.as_manager()
|
|
|
|
def __str__(self):
|
|
return _('{} {}').format(
|
|
self.name,
|
|
self.start.year,
|
|
)
|
|
|
|
def create_days(self):
|
|
delta = self.end - self.start
|
|
for day_offset in range(0, delta.days + 1):
|
|
day, created = self.days.get_or_create(
|
|
date=self.start + datetime.timedelta(days=day_offset)
|
|
)
|
|
|
|
def save(self, **kwargs):
|
|
super(Camp, self).save(**kwargs)
|
|
self.create_days()
|
|
|
|
|
|
class Day(CreatedUpdatedModel, UUIDModel):
|
|
class Meta:
|
|
verbose_name = _('Day')
|
|
verbose_name_plural = _('Days')
|
|
|
|
camp = models.ForeignKey(
|
|
'camps.Camp',
|
|
verbose_name=_('Camp'),
|
|
help_text=_('Which camp does this day belong to.'),
|
|
related_name='days',
|
|
)
|
|
|
|
date = models.DateField(
|
|
verbose_name=_('Date'),
|
|
help_text=_('What date?')
|
|
)
|
|
|
|
|
|
class Expense(CreatedUpdatedModel, UUIDModel):
|
|
class Meta:
|
|
verbose_name = _('Expense')
|
|
verbose_name_plural = _('Expenses')
|
|
|
|
camp = models.ForeignKey(
|
|
'camps.Camp',
|
|
verbose_name=_('Camp'),
|
|
help_text=_('The camp to which this expense relates to.'),
|
|
)
|
|
|
|
description = models.CharField(
|
|
verbose_name=_('Description'),
|
|
help_text=_('What this expense covers.'),
|
|
max_length=255,
|
|
)
|
|
|
|
amount = models.DecimalField(
|
|
verbose_name=_('Amount'),
|
|
help_text=_('The amount of the expense.'),
|
|
max_digits=7,
|
|
decimal_places=2,
|
|
)
|
|
|
|
CURRENCIES = [
|
|
('btc', 'BTC'),
|
|
('dkk', 'DKK'),
|
|
('eur', 'EUR'),
|
|
('sek', 'SEK'),
|
|
]
|
|
|
|
currency = models.CharField(
|
|
verbose_name=_('Currency'),
|
|
help_text=_('What currency the amount is in.'),
|
|
choices=CURRENCIES,
|
|
max_length=3,
|
|
)
|
|
|
|
covered_by = models.ForeignKey(
|
|
'auth.User',
|
|
verbose_name=_('Covered by'),
|
|
help_text=_('Which user, if any, covered this expense.'),
|
|
null=True,
|
|
blank=True,
|
|
)
|
|
|
|
def __str__(self):
|
|
return _('{} {} for {}').format(
|
|
self.amount,
|
|
self.get_currency_display(),
|
|
self.camp,
|
|
)
|