52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
from django.db import models
|
|
from django.conf import settings
|
|
from django.db.models import Sum
|
|
|
|
AUTH_USER_MODEL = getattr(settings, 'AUTH_USER_MODEL', 'auth.User')
|
|
|
|
|
|
class Refund(models.Model):
|
|
title = models.CharField(max_length=100)
|
|
date = models.DateField()
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
user = models.ForeignKey(
|
|
AUTH_USER_MODEL,
|
|
on_delete=models.PROTECT,
|
|
null=True
|
|
)
|
|
|
|
def eur_value(self) -> float:
|
|
value_sum = self.payment_set.all().aggregate(Sum('value')).get('value__sum')
|
|
if value_sum:
|
|
return value_sum / 100
|
|
else:
|
|
return 0
|
|
|
|
def __str__(self) -> str:
|
|
return "{0} on {1} for {2}".format(self.title, self.date, self.eur_value())
|
|
|
|
|
|
class Payment(models.Model):
|
|
title = models.CharField(max_length=100)
|
|
date = models.DateField()
|
|
value = models.IntegerField()
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
user = models.ForeignKey(
|
|
AUTH_USER_MODEL,
|
|
on_delete=models.PROTECT
|
|
)
|
|
refund = models.ForeignKey(
|
|
Refund,
|
|
on_delete=models.SET_NULL,
|
|
null=True,
|
|
blank=True
|
|
)
|
|
|
|
def __str__(self) -> str:
|
|
return '{0} on {1} for {2}'.format(self.title, self.date, self.value / 100)
|
|
|
|
def eur_value(self) -> float:
|
|
return self.value / 100
|