from django.contrib.auth import get_user_model
from django.db import models
from utils.base_model import BaseModel
from apps.project.models import Project as ProjectModel
# Create your models here.


CUser = get_user_model()


class ArzeshEmtiyaz(BaseModel):
    price = models.IntegerField(null=False, blank=False)

    def __str__(self):
        return f'{self.price}'

class Emtiyazat(BaseModel):
    cuser = models.ForeignKey(CUser, on_delete=models.CASCADE,  related_name='emtiyazat')
    credit = models.IntegerField(null=True, blank=True, default=None)
    debit = models.IntegerField(null=True, blank=True, default=None)
    desc = models.TextField(null=True, blank=True, default=None)
    project = models.ForeignKey(ProjectModel, on_delete=models.CASCADE, related_name='emtiyaz_project', null=True, blank=True)
    operator = models.ForeignKey(CUser, on_delete=models.CASCADE, related_name='emtiyazat_operator')

    class Meta:
        ordering = ['-created_at']