You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
4.7 KiB
113 lines
4.7 KiB
from django.db import models
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.postgres.fields import ArrayField, JSONField
|
|
|
|
from constance import config
|
|
from paymentwall import Pingback
|
|
from polymorphic.models import PolymorphicModel
|
|
|
|
from apps.course.models import Course
|
|
from apps.school.models import SchoolSchedule
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class AuthorBalance(models.Model):
|
|
IN = 0
|
|
OUT = 1
|
|
TYPE_CHOICES = (
|
|
(IN, 'In'),
|
|
(OUT, 'Out'),
|
|
)
|
|
PENDING = 0
|
|
ACCEPTED = 1
|
|
DECLINED = 2
|
|
STATUS_CHOICES = (
|
|
(PENDING, 'Pending'),
|
|
(ACCEPTED, 'Accepted'),
|
|
(DECLINED, 'Declined'),
|
|
)
|
|
author = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name='Автор',
|
|
null=True, blank=True, related_name='balances')
|
|
type = models.PositiveSmallIntegerField('Тип', choices=TYPE_CHOICES, default=0)
|
|
amount = models.DecimalField('Итого', max_digits=8, decimal_places=2, default=0)
|
|
commission = models.DecimalField('Комиссия', max_digits=8, decimal_places=2, default=0)
|
|
status = models.PositiveSmallIntegerField('Статус', choices=STATUS_CHOICES, default=0)
|
|
payment = models.OneToOneField('Payment', on_delete=models.CASCADE, null=True, blank=True, verbose_name='Платёж')
|
|
cause = models.TextField('Причина отказа', null=True, blank=True)
|
|
|
|
class Meta:
|
|
verbose_name = 'Баланс'
|
|
verbose_name_plural = 'Балансы'
|
|
|
|
|
|
class Payment(PolymorphicModel):
|
|
PW_STATUS_CHOICES = (
|
|
(Pingback.PINGBACK_TYPE_REGULAR, 'regular',),
|
|
(Pingback.PINGBACK_TYPE_GOODWILL, 'goodwill',),
|
|
(Pingback.PINGBACK_TYPE_NEGATIVE, 'negative',),
|
|
(Pingback.PINGBACK_TYPE_RISK_UNDER_REVIEW, 'risk under review',),
|
|
(Pingback.PINGBACK_TYPE_RISK_REVIEWED_ACCEPTED, 'risk reviewed accepted',),
|
|
(Pingback.PINGBACK_TYPE_RISK_REVIEWED_DECLINED, 'risk reviewed declined',),
|
|
(Pingback.PINGBACK_TYPE_RISK_AUTHORIZATION_VOIDED, 'risk authorization voided',),
|
|
(Pingback.PINGBACK_TYPE_SUBSCRIPTION_CANCELLATION, 'subscription cancelation',),
|
|
(Pingback.PINGBACK_TYPE_SUBSCRIPTION_EXPIRED, 'subscription expired',),
|
|
(Pingback.PINGBACK_TYPE_SUBSCRIPTION_PAYMENT_FAILED, 'subscription payment failed',),
|
|
)
|
|
user = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name='Пользователь', related_name='payments')
|
|
amount = models.DecimalField('Итого', max_digits=8, decimal_places=2, default=0, editable=False)
|
|
status = models.PositiveSmallIntegerField('Статус платежа', choices=PW_STATUS_CHOICES, null=True, editable=False)
|
|
data = JSONField('Данные платежа от провайдера', default={}, editable=False)
|
|
|
|
class Meta:
|
|
verbose_name = 'Платеж'
|
|
verbose_name_plural = 'Платежи'
|
|
|
|
def calc_commission(self):
|
|
return self.amount * config.SERVICE_COMMISSION / 100
|
|
|
|
|
|
class CoursePayment(Payment):
|
|
course = models.ForeignKey(Course, on_delete=models.CASCADE, verbose_name='Курс', related_name='payments')
|
|
|
|
class Meta:
|
|
verbose_name = 'Платеж за курс'
|
|
verbose_name_plural = 'Платежи за курсы'
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.amount = self.course.price
|
|
super().save(*args, **kwargs)
|
|
author_balance = getattr(self, 'authorbalance', None)
|
|
if not author_balance:
|
|
AuthorBalance.objects.create(
|
|
author=self.course.author,
|
|
amount=self.amount,
|
|
payment=self,
|
|
commission=self.calc_commission(),
|
|
)
|
|
else:
|
|
author_balance.amount = self.amount
|
|
author_balance.commission = self.calc_commission()
|
|
author_balance.save()
|
|
|
|
|
|
class SchoolPayment(Payment):
|
|
weekdays = ArrayField(models.IntegerField(), size=7, verbose_name='Дни недели')
|
|
date_start = models.DateField('Дата начала подписки', null=True, blank=True)
|
|
date_end = models.DateField('Дата окончания подписки', null=True, blank=True)
|
|
|
|
class Meta:
|
|
verbose_name = 'Платеж за школу'
|
|
verbose_name_plural = 'Платежи за школу'
|
|
|
|
def save(self, *args, **kwargs):
|
|
aggregate = SchoolSchedule.objects.filter(
|
|
weekday__in=self.weekdays,
|
|
).aggregate(
|
|
models.Sum('month_price'),
|
|
models.Sum('day_discount'),
|
|
)
|
|
month_price_sum = aggregate.get('month_price__sum', 0)
|
|
day_discount_sum = aggregate.get('day_discount__sum', 0) if len(self.weekdays) == 7 else 0
|
|
self.amount = month_price_sum - day_discount_sum
|
|
super().save(*args, **kwargs)
|
|
|