#from django.db import models
from djongo import models
from django_enumfield import enum

class PatentType(enum.Enum):
    PATENT = 1
    TECHNOLOGY_TRANSFER = 2

    __labels__ = {
        PATENT: "Patent",
        TECHNOLOGY_TRANSFER: "Technology Transfer",
    }

    __default__ = PATENT

class Patent(models.Model):
    _id = models.ObjectIdField()
    patent_type = enum.EnumField(PatentType, default=PatentType.PATENT)
    author = models.CharField(max_length=255)
    title = models.CharField(max_length=255)
    abstract = models.TextField()
    year = models.IntegerField(blank=True)
    application_num = models.CharField(blank=True,max_length=255)
    grant_num = models.CharField(blank=True,max_length=255)
    grant_date = models.DateField(blank=True)
    date_created = models.DateTimeField(auto_now_add=True, blank=True)
    date_updated = models.DateTimeField(auto_now=True, blank=True)

    def __str__(self):
        return self.title