"""
models for metacategories and protocol
"""
from colorful.fields import RGBColorField
from django.db import models
from django.utils.translation import ugettext_lazy as _

from .base import Model
from .auth import GroupProfile


class Protocol(Model):
    title = models.CharField(max_length=255, verbose_name=_('Protocol|title'))
    group_profile = models.OneToOneField(GroupProfile, on_delete=models.CASCADE)
    class Meta:
        verbose_name = _('Protocol')
        verbose_name_plural = _('Protocols')



class Category(models.Model):
    title = models.CharField(max_length=255, verbose_name=_('Category|title'))
    color = RGBColorField(verbose_name=_('Category|color'))
    need_comment = models.BooleanField(default=False, verbose_name=_('Category|need_comment'))
    description = models.TextField(null=True, blank=True, verbose_name=_('Category|description'))
    protocol = models.ForeignKey(
        Protocol,
        verbose_name=_('Category|protocol'),
        related_name='categories',
        on_delete=models.CASCADE
    )
    class Meta:
        verbose_name = _('Category')
        verbose_name_plural = _('Categories')