from taggit.models import GenericUUIDTaggedItemBase, TagBase
from taggit.managers import TaggableManager

class Tag(TagBase):
    class Meta:
        unique_together = ("name", "category")

    category = models.CharField(
        db_index=True, default="custom", max_length=200, null=False
    )
    name = models.CharField(db_index=True, max_length=255)


class TaggedItem(GenericUUIDTaggedItemBase):
    class Meta:
        unique_together = [["content_type", "object_id", "tag"]]

    confidence = models.FloatField()


class Taggable(models.Model):
    class Meta:
        abstract = True

    tags = TaggableManager(blank=True, through=TaggedItem)
    tagged_through = TaggedItem

class CustomModel(Taggable):
    ....

import graphene

class OzuTag(graphene.ObjectType):
    category = graphene.String()
    library_id = graphene.String()
    name = graphene.String()
    user_id = graphene.String()


class OzuTaggedItem(OzuTag):
    confidence = graphene.Float()
    tag_type = graphene.String()

class CustomObjectNode((DjangoObjectType):
    tags = graphene.List(OzuTaggedItem)

    # The original code has more robustness checks and if-else clauses, but I've reproduced
    # a minimal version here for ease of demonstration
    def resolve_tags(self, info):
        res = []
        for tagged_item in self.prefetched_tagged_items:
            res.append(
                {
                    "name": tagged_item.tag.name,
                    "category": tagged_item.tag.category,
                    "user_id": tagged_item.tag.user_id,
                    "library_id": tagged_item.tag.library_id,
                    "confidence": tagged_item.confidence,
                    "tag_type": tagged_item.tag_type,
                }
            )


class Query:
    all_objects = DjangoFilterConnectionField(CustomObjectNode)

    def resolve_all_objects(self, info):
        qset = CustomObject.objects.all()
        is_querying_for_tags, _ = check_for_field(info, "tags")  # Checks incoming GQL query to see if 'tags' is queried for
        if is_querying_for_tags:
            tagged_items_prefetch = Prefetch(
                "tagged_items",
                queryset=(
                    TaggedItem.objects.filter(
                        object_id__in=qset.values_list("pk")
                    ).select_related("tag")
                ),
                to_attr="prefetched_tagged_items",
            )
            qset = qset.prefetch_related(tagged_items_prefetch)
        return qset