70 lines
1.8 KiB
Python
70 lines
1.8 KiB
Python
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django_core.models import AbstractBaseModel
|
|
from model_bakery import baker
|
|
|
|
from core.apps.payment.enums.payment import PaymentMethod, PaymentStatus
|
|
|
|
|
|
class PaymentModel(AbstractBaseModel):
|
|
valuation = models.ForeignKey(
|
|
"evaluation.ValuationModel",
|
|
on_delete=models.CASCADE,
|
|
related_name="payments",
|
|
verbose_name=_("valuation"),
|
|
)
|
|
payer = models.ForeignKey(
|
|
"accounts.User",
|
|
on_delete=models.SET_NULL,
|
|
null=True,
|
|
blank=True,
|
|
related_name="payments",
|
|
verbose_name=_("payer"),
|
|
)
|
|
amount = models.DecimalField(
|
|
verbose_name=_("amount"),
|
|
max_digits=15,
|
|
decimal_places=2,
|
|
)
|
|
payment_method = models.CharField(
|
|
verbose_name=_("payment method"),
|
|
max_length=50,
|
|
choices=PaymentMethod.choices,
|
|
default=PaymentMethod.CASH,
|
|
)
|
|
status = models.CharField(
|
|
verbose_name=_("status"),
|
|
max_length=50,
|
|
choices=PaymentStatus.choices,
|
|
default=PaymentStatus.PENDING,
|
|
)
|
|
transaction_id = models.CharField(
|
|
verbose_name=_("transaction ID"),
|
|
max_length=255,
|
|
blank=True,
|
|
null=True,
|
|
unique=True,
|
|
)
|
|
paid_at = models.DateTimeField(
|
|
verbose_name=_("paid at"),
|
|
blank=True,
|
|
null=True,
|
|
)
|
|
note = models.TextField(
|
|
verbose_name=_("note"),
|
|
blank=True,
|
|
)
|
|
|
|
def __str__(self):
|
|
return f"Payment #{self.pk} — {self.amount} ({self.get_status_display()})"
|
|
|
|
@classmethod
|
|
def _baker(cls):
|
|
return baker.make(cls)
|
|
|
|
class Meta:
|
|
db_table = "Payment"
|
|
verbose_name = _("Payment")
|
|
verbose_name_plural = _("Payments")
|
|
ordering = ["-created_at"]
|