from django.db import models
from django.db.models import Sum
from django.utils import timezone

from users.models import User

TYPES = (
    ('minus', 'Снятие'),
    ('plus', 'Приход'),

)


class InvoiceHistory(models.Model):
    comment = models.TextField(blank=True)
    created = models.DateTimeField(default=timezone.now)
    sum = models.DecimalField(max_digits=10, decimal_places=0, default=0, null=True, blank=True)
    balance = models.DecimalField(max_digits=10, decimal_places=0, default=0)
    type = models.CharField(max_length=20, blank=True, null=True)
    user = models.ForeignKey(User, related_name='invoice_history')

    def __str__(self):
        return self.comment

    def save(self, *args, **kwargs):
        if self.pk is None:
            current_sum_info = InvoiceHistory.objects.filter(user=self.user).aggregate(Sum('sum'))
            current_sum = current_sum_info['sum__sum'] or 0
            self.balance = current_sum + self.sum
        super().save(*args, **kwargs)

    class Meta:
        verbose_name = 'Счет(История)'
        verbose_name_plural = 'Счет(История)'
        ordering = ('-created',)


class WithDraw(models.Model):
    sum = models.DecimalField(max_digits=10, decimal_places=0)
    created = models.DateTimeField(default=timezone.now)
    yandex_card = models.CharField(max_length=30)
    user = models.ForeignKey(User, related_name='with_draw')

    def __str__(self):
        return self.yandex_card

    class Meta:
        verbose_name = 'Заявка на вывод средств'
        verbose_name_plural = 'Заявки на вывод средств'
        ordering = ('-created',)