from django.contrib.auth import get_user_model
from django.db import models
from utils.base_model import BaseModel
# Create your models here.
CUserModel = get_user_model()


class ProjectCodeHead(BaseModel):
    title = models.CharField(max_length=300, null=False, blank=False)
    count = models.IntegerField(null=False, blank=False)

    def __str__(self):
        return self.title

    class Meta:
        ordering = ['-created_at']


class ProjectCode(BaseModel):
    project_code_head = models.ForeignKey(ProjectCodeHead, on_delete=models.CASCADE)
    code = models.CharField(max_length=20, null=False, blank=False)
    is_active = models.BooleanField(default=True)
    is_used = models.BooleanField(default=False)
    mojri_user = models.ForeignKey(CUserModel, on_delete=models.CASCADE, null=True, blank=True)

    def __str__(self):
        return self.code

    class Meta:
        ordering = ['-created_at']