ФУНКЦИИ - МЕЖПРОЕКТНАЯ КОММУНИКАЦИЯ: blob-дедупликация файлов

plane-src/apps/api/plane/app/views/asset/base.py

@@ -11,6 +11,7 @@ from rest_framework.parsers import MultiPartParser, FormParser, JSONParser
 from ..base import BaseAPIView, BaseViewSet
 from plane.db.models import FileAsset, Workspace
 from plane.app.serializers import FileAssetSerializer
+from plane.utils.file_dedup import UploadedObjectMissing, confirm_uploaded_file_asset, release_file_asset_blob
 
 
 class FileAssetEndpoint(BaseAPIView):
@@ -37,15 +38,24 @@ class FileAssetEndpoint(BaseAPIView):
         if serializer.is_valid():
             # Get the workspace
             workspace = Workspace.objects.get(slug=slug)
-            serializer.save(workspace_id=workspace.id)
+            asset = serializer.save(workspace_id=workspace.id)
+            try:
+                confirm_uploaded_file_asset(asset, request=request)
+            except UploadedObjectMissing:
+                return Response(
+                    {"error": "The uploaded asset object was not found.", "status": False},
+                    status=status.HTTP_400_BAD_REQUEST,
+                )
             return Response(serializer.data, status=status.HTTP_201_CREATED)
         return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
 
     def delete(self, request, workspace_id, asset_key):
         asset_key = str(workspace_id) + "/" + asset_key
         file_asset = FileAsset.objects.get(asset=asset_key)
+        if not file_asset.is_uploaded:
+            release_file_asset_blob(file_asset, request=request, delete_untracked_object=True)
         file_asset.is_deleted = True
-        file_asset.save(update_fields=["is_deleted"])
+        file_asset.save(update_fields=["is_deleted", "updated_at"])
         return Response(status=status.HTTP_204_NO_CONTENT)
 
 
@@ -75,12 +85,21 @@ class UserAssetsEndpoint(BaseAPIView):
     def post(self, request):
         serializer = FileAssetSerializer(data=request.data)
         if serializer.is_valid():
-            serializer.save()
+            asset = serializer.save()
+            try:
+                confirm_uploaded_file_asset(asset, request=request)
+            except UploadedObjectMissing:
+                return Response(
+                    {"error": "The uploaded asset object was not found.", "status": False},
+                    status=status.HTTP_400_BAD_REQUEST,
+                )
             return Response(serializer.data, status=status.HTTP_201_CREATED)
         return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
 
     def delete(self, request, asset_key):
         file_asset = FileAsset.objects.get(asset=asset_key, created_by=request.user)
+        if not file_asset.is_uploaded:
+            release_file_asset_blob(file_asset, request=request, delete_untracked_object=True)
         file_asset.is_deleted = True
-        file_asset.save(update_fields=["is_deleted"])
+        file_asset.save(update_fields=["is_deleted", "updated_at"])
         return Response(status=status.HTTP_204_NO_CONTENT)