Add content management queries and mutations (#91)

* Add content management queries and mutations
* Fix pre-commit fail issue
* Fix merge issue on schema file
* Update schema

Author: sudip-khanal

.pre-commit-config.yaml

@@ -17,6 +17,8 @@ repos:
     rev: 24.4.2
     hooks:
       - id: black
+        exclude: ^.*migrations.*
+
         # args: ["--check"]
 
   - repo: https://github.com/PyCQA/isort

common/management/commands/generate_schema.py

@@ -0,0 +1,23 @@
+import argparse
+
+from django.core.management.base import BaseCommand
+from strawberry.printer import print_schema
+
+from main.graphql.schema import schema
+
+
+class Command(BaseCommand):
+    help = "Create schema.graphql file"
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--out",
+            type=argparse.FileType("w"),
+            default="schema.graphql",
+        )
+
+    def handle(self, *args, **options):
+        file = options["out"]
+        file.write(print_schema(schema))
+        file.close()
+        self.stdout.write(self.style.SUCCESS(f"{file.name} file generated"))

content/enums.py

@@ -0,0 +1,17 @@
+import strawberry
+
+from content.models import Content
+from utils.strawberry.enums import get_enum_name_from_django_field
+
+DocumentStatusTypeEnum = strawberry.enum(Content.DocumentStatus, name="DocumentStatusTypeEnum")
+
+DocumentTypeEnum = strawberry.enum(Content.DocumentType, name="DocumentTypeEnum")
+
+
+enum_map = {
+    get_enum_name_from_django_field(field): enum
+    for field, enum in (
+        (Content.document_status, DocumentStatusTypeEnum),
+        (Content.document_type, DocumentTypeEnum),
+    )
+}

content/migrations/0003_content_document_url_alter_content_document_type.py

@@ -0,0 +1,23 @@
+# Generated by Django 5.1.6 on 2025-02-23 08:35
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('content', '0002_alter_content_document_type'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='content',
+            name='document_url',
+            field=models.URLField(blank=True, null=True),
+        ),
+        migrations.AlterField(
+            model_name='content',
+            name='document_type',
+            field=models.IntegerField(choices=[(1, 'WEB_URl'), (2, 'PDF'), (3, 'Text')], default=3),
+        ),
+    ]

content/models.py

@@ -17,7 +17,7 @@ def __str__(self):
 
 class Content(UserResource):
     class DocumentType(models.IntegerChoices):
-        WORD = 1, _("Word")
+        WEB_URl = 1, _("WEB_URl")
         PDF = 2, _("PDF")
         TEXT = 3, _("Text")
 
@@ -31,6 +31,7 @@ class DocumentStatus(models.IntegerChoices):
     title = models.CharField(max_length=100)
     document_type = models.IntegerField(choices=DocumentType.choices, default=DocumentType.TEXT)
     document_file = models.FileField(upload_to="documents")
+    document_url = models.URLField(null=True, blank=True)
     extracted_file = models.FileField(upload_to="documents-extracts", null=True, blank=True)
     content_id = models.UUIDField(default=uuid.uuid4, editable=False)
     document_status = models.PositiveSmallIntegerField(choices=DocumentStatus.choices, default=DocumentStatus.PENDING)

content/mutations.py

@@ -0,0 +1,95 @@
+import typing
+
+import strawberry
+from asgiref.sync import sync_to_async
+from strawberry.file_uploads import Upload
+
+from content.models import Content
+from content.serializers import (
+    ArchiveContentSerializer,
+    ContentSerializer,
+    TagSerializer,
+    UpdateContentSerializer,
+)
+from content.types import ContentType, TagType
+from main.graphql.context import Info
+from utils.strawberry.mutations import (
+    ModelMutation,
+    MutationEmptyResponseType,
+    MutationResponseType,
+    mutation_is_not_valid,
+    process_input_data,
+)
+
+
+@strawberry.input
+class FolderInput:
+    files: typing.List[Upload]
+
+
+CreateContentMutation = ModelMutation("Content", ContentSerializer)
+UpdateMutation = ModelMutation("UpdateContent", UpdateContentSerializer)
+DeleteContent = ModelMutation("archive", ArchiveContentSerializer)
+CreateTagMutation = ModelMutation("CreateTag", TagSerializer)
+
+
+@strawberry.type
+class PrivateMutation:
+    @strawberry.mutation
+    async def create_content(
+        self,
+        data: CreateContentMutation.InputType,  # type: ignore[reportInvalidTypeForm]
+        info: Info,
+    ) -> MutationResponseType[ContentType]:
+        return await CreateContentMutation.handle_create_mutation(data, info, None)
+
+    @strawberry.mutation
+    def read_file(self, file: Upload) -> str:
+        return file.read().decode("utf-8")
+
+    @strawberry.mutation
+    async def update_content_title(
+        self,
+        id: strawberry.ID,
+        data: UpdateMutation.PartialInputType,  # type: ignore[reportInvalidTypeForm]
+        info: Info,
+    ) -> MutationResponseType[ContentType]:
+        try:
+            instance = await Content.objects.aget(id=id)
+        except Content.DoesNotExist:
+            return MutationResponseType(ok=False, errors=["Content not found"])
+        serializer = UpdateContentSerializer(
+            instance, data=process_input_data(data), context={"request": info.context.request}, partial=True
+        )
+        if errors := mutation_is_not_valid(serializer):
+            return MutationResponseType(ok=False, errors=errors)
+        await sync_to_async(serializer.save)()
+        return MutationResponseType()
+
+    @strawberry.mutation
+    async def archive_content(
+        self, id: strawberry.ID, info: Info, data: DeleteContent.PartialInputType  # type: ignore[reportInvalidTypeForm]
+    ) -> MutationEmptyResponseType:
+        try:
+            instance = await Content.objects.aget(id=id)
+        except Content.DoesNotExist:
+            return MutationEmptyResponseType(ok=False, errors=["Content not found"])
+
+        serializer = ArchiveContentSerializer(
+            instance, data=process_input_data(data), context={"request": info.context.request}, partial=True
+        )
+
+        if errors := mutation_is_not_valid(serializer):
+            return MutationEmptyResponseType(ok=False, errors=errors)
+
+        await sync_to_async(serializer.save)()
+
+        return MutationEmptyResponseType(ok=True)
+
+    @strawberry.mutation
+    async def create_tag(
+        self,
+        data: CreateTagMutation.InputType,  # type: ignore[reportInvalidTypeForm]
+        info: Info,
+    ) -> MutationResponseType[TagType]:
+        return await CreateTagMutation.handle_create_mutation(data, info, None)

content/queries.py

@@ -0,0 +1,25 @@
+import strawberry
+import strawberry_django
+
+from content.types import ContentType, TagType
+from main.graphql.context import Info
+from utils.strawberry.paginations import CountList, pagination_field
+
+
+@strawberry.type
+class PrivateQuery:
+    content: CountList[ContentType] = pagination_field(
+        pagination=True,
+    )
+
+    @strawberry_django.field(description="Return all content")
+    async def all_content(self, info: Info) -> list[ContentType]:
+        return [content async for content in ContentType.get_queryset(None, None, info)]
+
+    tag: CountList[ContentType] = pagination_field(
+        pagination=True,
+    )
+
+    @strawberry_django.field()
+    async def all_tags(self, info: Info) -> list[TagType]:
+        return [tag async for tag in TagType.get_queryset(None, None, info)]

content/serializers.py

@@ -1,9 +1,73 @@
 from rest_framework import serializers
 
 from chat.models import UserChatSession
+from content.models import Content, Tag
+from utils.file_check import validate_document_size, validate_file_type
 
 
 class UserQuerySerializer(serializers.Serializer):
     query = serializers.CharField(required=True, allow_null=False, allow_blank=False)
     user_id = serializers.UUIDField(required=True)
     platform = serializers.IntegerField(default=UserChatSession.Platform.STREAMLIT)
+
+
+class TagSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Tag
+        fields = ["name", "description"]
+
+
+class ContentSerializer(serializers.ModelSerializer):
+    tag = serializers.PrimaryKeyRelatedField(queryset=Tag.objects.all(), many=True, required=False)
+    document_file = serializers.FileField(required=False)
+
+    class Meta:
+        model = Content
+        fields = ["title", "document_file", "tag", "document_type", "document_url"]
+        read_only_fields = ["created_by", "modified_by"]
+
+    def validate_document_file(self, file):
+        validate_document_size(file)
+        validate_file_type(file)
+        return file
+
+    def create(self, validated_data):
+        tags = validated_data.pop("tag", [])
+        validated_data["created_by"] = self.context["request"].user
+        validated_data["modified_by"] = self.context["request"].user
+        content = super().create(validated_data)
+        content.tag.set(tags)
+        return content
+
+
+class UpdateContentSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Content
+        fields = ["title"]
+        read_only_fields = ["modified_by"]
+
+    def update(self, instance, validated_data):
+        validated_data["modified_by"] = self.context["request"].user
+
+        return super().update(instance, validated_data)
+
+
+class ArchiveContentSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Content
+        fields = [
+            "is_deleted",
+        ]
+        read_only_fields = ["deleted_at", "deleted_by"]
+
+    def validate(self, data):
+        instance = self.instance
+        if instance.is_deleted:
+            raise serializers.ValidationError("Content is already deleted.")
+        return data
+
+    def update(self, instance, validated_data):
+        instance.is_deleted = True
+        instance.deleted_by = self.context["request"].user
+        instance.save(update_fields=("is_deleted", "deleted_by"))
+        return instance

content/types.py

@@ -0,0 +1,38 @@
+import strawberry
+import strawberry_django
+from django.db import models
+
+from content.models import Content, Tag
+from main.graphql.context import Info
+from utils.common import get_queryset_for_model
+from utils.strawberry.enums import enum_field
+
+
+@strawberry_django.type(Tag)
+class TagType:
+    id: strawberry.ID
+    name: strawberry.auto
+    description: strawberry.auto
+
+    @staticmethod
+    def get_queryset(_, queryset: models.QuerySet | None, info: Info) -> models.QuerySet:
+        return get_queryset_for_model(Tag, queryset)
+
+
+@strawberry_django.type(Tag)
+class TagNameType:
+    name: strawberry.auto
+
+
+@strawberry_django.type(Content)
+class ContentType:
+    id: strawberry.ID
+    title: strawberry.auto
+    extracted_file: strawberry.auto
+    tag: list[TagNameType]
+    document_status = enum_field(Content.document_status)
+    document_type = enum_field(Content.document_type)
+
+    @staticmethod
+    def get_queryset(_, queryset: models.QuerySet | None, info: Info):
+        return get_queryset_for_model(Content, queryset)

main/graphql/enums.py

@@ -2,10 +2,12 @@
 
 import strawberry
 
+from content.enums import enum_map as content_enum_map
 from user.enums import enum_map as user_enum_map
 
 ENUM_TO_STRAWBERRY_ENUM_MAP: dict[str, type] = {
     **user_enum_map,
+    **content_enum_map,
 }
 
 

main/graphql/schema.py

@@ -1,6 +1,8 @@
 import strawberry
 from strawberry.django.views import AsyncGraphQLView
 
+from content import mutations as content_mutations
+from content import queries as content_queries
 from user import mutations as user_mutations
 from user import queries as user_queries
 
@@ -22,13 +24,15 @@ async def get_context(self, *args, **kwargs) -> GraphQLContext:
 @strawberry.type
 class PublicQuery(
     user_queries.PublicQuery,
+    content_queries.PrivateQuery,
 ):
     id: strawberry.ID = strawberry.ID("public")
 
 
 @strawberry.type
 class PrivateQuery(
     user_queries.PrivateQuery,
+    content_queries.PrivateQuery,
 ):
     id: strawberry.ID = strawberry.ID("private")
 
@@ -41,7 +45,7 @@ class PublicMutation(
 
 
 @strawberry.type
-class PrivateMutation(user_mutations.PrivateMutation):
+class PrivateMutation(user_mutations.PrivateMutation, content_mutations.PrivateMutation):
     id: strawberry.ID = strawberry.ID("private")
 
 

main/urls.py

@@ -25,9 +25,15 @@
 from main.graphql.schema import CustomAsyncGraphQLView
 from main.graphql.schema import schema as graphql_schema
 
-urlpatterns = [path("admin/", admin.site.urls), path("chat_message", UserQuery.as_view())]
+urlpatterns = [
+    path("admin/", admin.site.urls),
+    path("chat_message", UserQuery.as_view()),
+    path("graphql/", csrf_exempt(CustomAsyncGraphQLView.as_view(schema=graphql_schema, multipart_uploads_enabled=True))),
+]
 if settings.DEBUG:
-    urlpatterns.append(path("graphiql/", csrf_exempt(CustomAsyncGraphQLView.as_view(schema=graphql_schema))))
+    urlpatterns.append(
+        path("graphiql/", csrf_exempt(CustomAsyncGraphQLView.as_view(schema=graphql_schema, multipart_uploads_enabled=True)))
+    )
 
     # Static and media file URLs
     urlpatterns += static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT)