Added image api

Author: MySecondLanguage

nxtbn/admin_schema.py

@@ -3,6 +3,7 @@
 from nxtbn.cart.admin_query import AdminCartQuery
 from nxtbn.core.admin_mutation import CoreMutation
 from nxtbn.core.admin_queries import AdminCoreQuery
+from nxtbn.filemanager.admin_queries import ImageQuery
 from nxtbn.order.admin_mutation import AdminOrderMutation
 from nxtbn.payment.admin_queries import AdminPaymentQuery
 from nxtbn.product.admin_mutations import ProductMutation
@@ -15,7 +16,7 @@
 
 
 
-class Query(ProductQuery, AdminOrderQuery, AdminCoreQuery, WarehouseQuery, AdminCartQuery, UserAdminQuery, PurchaseQuery, AdminPaymentQuery):
+class Query(ProductQuery, AdminOrderQuery, AdminCoreQuery, WarehouseQuery, AdminCartQuery, UserAdminQuery, PurchaseQuery, AdminPaymentQuery, ImageQuery):
     pass
 
 class Mutation(AdminUserMutation, ProductMutation, CoreMutation, AdminOrderMutation):

nxtbn/filemanager/admin_queries.py

@@ -0,0 +1,20 @@
+from django.conf import settings
+import graphene
+from graphql import GraphQLError
+from graphene_django.filter import DjangoFilterConnectionField
+from nxtbn.filemanager.models import Image
+from nxtbn.filemanager.admin_types import ImageType
+
+
+class ImageQuery(graphene.ObjectType):
+    images = DjangoFilterConnectionField(ImageType)
+    image = graphene.Field(ImageType, id=graphene.ID(required=True))
+
+    def resolve_images(self, info, **kwargs):
+        return Image.objects.all().order_by('-created_at')
+    
+    def resolve_image(self, info, id):
+        try:
+            return Image.objects.get(id=id)
+        except Image.DoesNotExist:
+            return None

nxtbn/filemanager/admin_types.py

@@ -0,0 +1,32 @@
+from django.conf import settings
+import graphene
+from graphene_django import DjangoObjectType
+from graphene import relay
+from nxtbn.filemanager.models import Image
+
+
+class ImageType(DjangoObjectType):
+    db_id = graphene.Int(source='id')
+    image = graphene.String()
+    image_xs = graphene.String()
+
+    def resolve_image(self, info):
+        return self.get_image_url(info.context)
+    
+    def resolve_image_xs(self, info):
+        return self.get_image_xs_url(info.context)
+
+    class Meta:
+        model = Image
+        fields = (
+            'id',
+            'name',
+            'image',
+            'image_xs',
+            'created_at',
+            'last_modified',
+        )
+        interfaces = (relay.Node, )
+        filter_fields = {
+            'name': ['exact', 'icontains'],
+        }

nxtbn/filemanager/models.py

@@ -12,6 +12,20 @@ class Image(AbstractBaseModel):
     image_xs = models.ImageField(null=True, blank=True)
     image_alt_text = models.CharField(max_length=255)
 
+    def get_image_url(self,request):
+        if self.image:
+            return request.build_absolute_uri(self.image.url)
+        return None
+
+    def get_image_xs_url(self,request):
+        if self.image_xs:
+            return request.build_absolute_uri(self.image_xs.url)
+        
+        if self.image:
+            return request.build_absolute_uri(self.image.url)
+
+        return None
+
 
 class Document(AbstractBaseModel):
     created_by = models.ForeignKey(User, on_delete=models.PROTECT, related_name='document_created')