80 lines
2.9 KiB
Python
Executable File
80 lines
2.9 KiB
Python
Executable File
from django.db import models
|
|
from django.db.models.signals import pre_delete, pre_save
|
|
from django.dispatch.dispatcher import receiver
|
|
from django.urls import reverse
|
|
|
|
from app.base.forms.fields import CurrencyField, DateTimeField
|
|
from app.base.forms.utils import datetime_now
|
|
|
|
from typing import TYPE_CHECKING
|
|
if TYPE_CHECKING:
|
|
from datetime import datetime
|
|
from decimal import Decimal
|
|
from app.base.models import Account, Booking
|
|
|
|
|
|
class Transaction(models.Model):
|
|
account: 'models.ForeignKey[Account]' = models.ForeignKey(
|
|
'Account', on_delete=models.CASCADE, verbose_name='Konto')
|
|
amount: 'models.DecimalField[Decimal]' = CurrencyField(
|
|
'Betrag')
|
|
booking: 'models.OneToOneField[Booking]|models.OneToOneField[None]' = \
|
|
models.OneToOneField('Booking', on_delete=models.CASCADE,
|
|
verbose_name='Zugehörige Zeitbuchung',
|
|
blank=True, null=True, default=None)
|
|
description = models.CharField(
|
|
'Beschreibung', max_length=500)
|
|
time_stamp: 'models.DateTimeField[datetime]' = DateTimeField(
|
|
'Datum / Uhrzeit', editable=False)
|
|
|
|
class Meta:
|
|
verbose_name = 'Transaktion'
|
|
verbose_name_plural = 'Transaktionen'
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('transaction:detail', kwargs={'pk': self.pk})
|
|
|
|
def __str__(self):
|
|
return f'Transaktion über {self.amount}€ von {self.account}'
|
|
|
|
@staticmethod
|
|
def upsertFromBooking(booking: 'Booking'):
|
|
amount = booking.calculated_price
|
|
should_exist = booking.end_time and amount
|
|
description = f'{booking.type.label} ({booking.duration or 0} Min)'
|
|
|
|
# Create or update existing Transaction
|
|
transaction = Transaction.objects.filter(booking=booking).first()
|
|
if transaction:
|
|
if should_exist:
|
|
transaction.amount = -amount
|
|
transaction.description = description
|
|
transaction.save()
|
|
else:
|
|
transaction.delete()
|
|
|
|
elif should_exist:
|
|
Transaction.objects.create(
|
|
account=booking.user.account,
|
|
amount=-amount,
|
|
description=description,
|
|
booking=booking,
|
|
)
|
|
|
|
|
|
# Signals
|
|
|
|
@receiver(pre_delete, sender=Transaction, dispatch_uid='transaction_pre_del')
|
|
def transaction_pre_delete(sender, instance: Transaction, **kwargs):
|
|
instance.account.update_balance(-instance.amount)
|
|
|
|
|
|
@receiver(pre_save, sender=Transaction, dispatch_uid='transaction_pre_save')
|
|
def transaction_pre_save(sender, instance: Transaction, **kwargs):
|
|
if instance._state.adding:
|
|
instance.time_stamp = datetime_now()
|
|
instance.account.update_balance(instance.amount)
|
|
elif instance.pk: # basically an "else", but just to be sure
|
|
prev = Transaction.objects.get(pk=instance.pk)
|
|
instance.account.update_balance(instance.amount - prev.amount)
|