baseapp-files: initial package

Author: nossila

baseapp_core/graphql/__init__.py

@@ -1,7 +1,7 @@
 from .connections import CountedConnection  # noqa
 from .decorators import login_required, user_passes_test  # noqa
 from .errors import Errors, ErrorType  # noqa
-from .fields import File, ThumbnailField  # noqa
+from .fields import get_file_object_type, ThumbnailField  # noqa
 from .middlewares import (  # noqa
     JWTAuthentication,
     LogExceptionMiddleware,

baseapp_core/graphql/fields.py

@@ -1,4 +1,5 @@
 import graphene
+from django.apps import apps
 from django.conf import settings
 from django.core.cache import InvalidCacheBackendError, caches
 from easy_thumbnails.files import get_thumbnailer
@@ -8,15 +9,23 @@
 except InvalidCacheBackendError:
     cache = None
 
+def get_file_object_type():
+    if apps.is_installed("baseapp_files"):
+        from baseapp_files.graphql.object_types import FileObjectType
+    else:
+        class FileObjectType(graphene.ObjectType):
+            url = graphene.String(required=True)
+            # contentType = graphene.String()
+            # bytes = graphene.Int()
 
-class File(graphene.ObjectType):
-    url = graphene.String(required=True)
-    # contentType = graphene.String()
-    # bytes = graphene.Int()
+            class Meta:
+                name = "File"
+
+    return FileObjectType
 
 
 class ThumbnailField(graphene.Field):
-    def __init__(self, type=File, **kwargs):
+    def __init__(self, type=get_file_object_type, **kwargs):
         kwargs.update(
             {
                 "args": {
@@ -29,6 +38,7 @@ def __init__(self, type=File, **kwargs):
 
     def get_resolver(self, parent_resolver):
         resolver = self.resolver or parent_resolver
+        FileObjectType = get_file_object_type()
 
         def built_thumbnail(instance, info, width, height, **kwargs):
             instance = resolver(instance, info, **kwargs)
@@ -40,7 +50,7 @@ def built_thumbnail(instance, info, width, height, **kwargs):
                 cache_key = self._get_cache_key(instance, width, height)
                 value_from_cache = cache.get(cache_key)
                 if value_from_cache:
-                    return File(url=value_from_cache)
+                    return FileObjectType(url=value_from_cache)
 
             thumbnailer = get_thumbnailer(instance)
             url = thumbnailer.get_thumbnail({"size": (width, height)}).url
@@ -49,7 +59,7 @@ def built_thumbnail(instance, info, width, height, **kwargs):
             if cache:
                 cache.set(cache_key, absolute_url, timeout=None)
 
-            return File(url=absolute_url)
+            return FileObjectType(url=absolute_url)
 
         return built_thumbnail
 

baseapp_core/graphql/views.py

@@ -1,4 +1,4 @@
-import logging
+import json, logging
 from contextlib import contextmanager
 
 import pghistory
@@ -7,6 +7,8 @@
 from graphene_django.views import HttpError
 from graphql import get_operation_ast, parse
 from graphql.execution import ExecutionResult
+# from graphene_file_upload.utils import place_files_in_operations
+
 
 try:
     import sentry_sdk
@@ -60,3 +62,19 @@ def execute_graphql_request(
                 return super().execute_graphql_request(
                     request, data, query, variables, operation_name, show_graphiql
                 )
+
+    def parse_body(self, request):
+        """Handle multipart request spec for multipart/form-data"""
+        content_type = self.get_content_type(request)
+        # logging.info('content_type: %s' % content_type)
+        # import pdb; pdb.set_trace()
+        if content_type == 'multipart/form-data' and 'operations' in request.POST:
+            operations = json.loads(request.POST.get('operations', '{}'))
+            # import pdb; pdb.set_trace()
+            files_map = json.loads(request.POST.get('map', '{}'))
+            return place_files_in_operations(
+                operations,
+                files_map,
+                request.FILES
+            )
+        return super(GraphQLView, self).parse_body(request)