from django.conf import settings
from django.db import models


class Expense(models.Model):
    class ExpenseType(models.TextChoices):
        RENT = 'RENT', 'Rent'
        TRANSPORT = 'TRANSPORT', 'Transport'
        SALARY = 'SALARY', 'Salary'
        UTILITIES = 'UTILITIES', 'Utilities'
        DAMAGE = 'DAMAGE', 'Damaged Stock'
        OTHER = 'OTHER', 'Other'

    expense_type = models.CharField(max_length=30, choices=ExpenseType.choices)
    description = models.CharField(max_length=255)
    amount = models.DecimalField(max_digits=15, decimal_places=2)
    expense_date = models.DateField()
    created_by = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.PROTECT,
        related_name='expenses',
    )
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        ordering = ['-expense_date']

    def __str__(self):
        return f'{self.get_expense_type_display()} - {self.amount}'