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 DgCategory(BaseModel):
    name = models.CharField(max_length=30, null=True, blank=True)
    desc = models.CharField(max_length=30, null=True, blank=True)
    is_active = models.BooleanField(default=True)

    def __str__(self):
        return str(self.name)


class DgCode(BaseModel):
    category = models.ForeignKey(DgCategory, on_delete=models.CASCADE, null=False, blank=False)
    code = models.CharField(max_length=30, null=False, blank=False)
    is_used = models.BooleanField(default=False, null=False, blank=False)
    buy_price = models.PositiveIntegerField(null=True, blank=True)
    desc = models.CharField(max_length=30, null=True, blank=True)
    for_user = models.ForeignKey(CUserModel, on_delete=models.SET_NULL, null=True, blank=True)

    def __str__(self):
        return f'{self.category.name} - {self.code}'