"""Models for app comment."""
from django.contrib.contenttypes import fields as generic
from django.db import models
from django.utils.translation import gettext_lazy as _

from account.models import User
from utils.models import ProjectBaseMixin
from utils.querysets import ContentTypeQuerySetMixin
from translation.models import Language
from location.models import Country


class CommentQuerySet(ContentTypeQuerySetMixin):
    """QuerySets for Comment model."""

    def by_user(self, user: User):
        """Return comments by author"""
        return self.filter(user=user)

    def annotate_is_mine_status(self, user):
        """Annotate belonging status"""
        return self.annotate(is_mine=models.Case(
            models.When(
                models.Q(user=user if user.is_authenticated else None),
                then=True
            ),
            default=False,
            output_field=models.BooleanField()
        ))


class Comment(ProjectBaseMixin):
    """Comment model."""
    text = models.TextField(verbose_name=_('Comment text'))
    mark = models.PositiveIntegerField(blank=True, null=True, default=None, verbose_name=_('Mark'))
    user = models.ForeignKey('account.User', related_name='comments', on_delete=models.CASCADE, verbose_name=_('User'))
    country = models.ForeignKey(Country, verbose_name=_('Country'), on_delete=models.SET_NULL, null=True)
    old_id = models.IntegerField(null=True, blank=True, default=None)

    content_type = models.ForeignKey(generic.ContentType, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    content_object = generic.GenericForeignKey('content_type', 'object_id')

    objects = CommentQuerySet.as_manager()

    class Meta:
        """Meta class"""
        verbose_name = _('Comment')
        verbose_name_plural = _('Comments')

    def __str__(self):
        """String representation"""
        return str(self.user)