mirror of
https://github.com/more-tech4-magnum-opus/backend.git
synced 2024-12-04 00:23:44 +03:00
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
|
from django.core.validators import MinValueValidator
|
||
|
|
from django.db import models
|
||
|
|
|
||
|
|
# Create your models here.
|
||
|
|
from users.models import User
|
||
|
|
|
||
|
|
|
||
|
|
class Transaction(models.Model):
|
||
|
|
user_from = models.ForeignKey(
|
||
|
|
User, related_name="transactions_from", on_delete=models.CASCADE
|
||
|
|
)
|
||
|
|
user_to = models.ForeignKey(
|
||
|
|
User, related_name="transactions_to", on_delete=models.CASCADE
|
||
|
|
)
|
||
|
|
|
||
|
|
amount = models.IntegerField(validators=[MinValueValidator(1)])
|
||
|
|
|
||
|
|
created = models.DateTimeField(auto_now_add=True)
|
||
|
|
|
||
|
|
hash = models.CharField(max_length=256, unique=True)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return f"transaction from {self.user_from} to {self.user_to}"
|
||
|
|
|
||
|
|
|
||
|
|
class AdminTransaction(models.Model):
|
||
|
|
class TransactionType(models.TextChoices):
|
||
|
|
FROM = "PAYMENT", "payment"
|
||
|
|
TO = "SALARY", "salary"
|
||
|
|
|
||
|
|
type = models.CharField(max_length=7, choices=TransactionType.choices)
|
||
|
|
amount = models.IntegerField(validators=[MinValueValidator(1)])
|
||
|
|
user = models.ForeignKey(
|
||
|
|
User, related_name="admin_transactions", on_delete=models.CASCADE
|
||
|
|
)
|
||
|
|
|
||
|
|
created = models.DateTimeField(auto_now_add=True)
|
||
|
|
|
||
|
|
hash = models.CharField(max_length=256, unique=True)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def user_from(self):
|
||
|
|
return "system" if self.type == self.TransactionType.TO else self.user.username
|
||
|
|
|
||
|
|
@property
|
||
|
|
def user_to(self):
|
||
|
|
return (
|
||
|
|
"system" if self.type == self.TransactionType.FROM else self.user.username
|
||
|
|
)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return str(self.amount)
|