Showing
1 changed file
with
2 additions
and
1 deletions
... | @@ -58,7 +58,8 @@ class ItemViewSet(viewsets.ViewSet): | ... | @@ -58,7 +58,8 @@ class ItemViewSet(viewsets.ViewSet): |
58 | def destroy(self, request, pk): | 58 | def destroy(self, request, pk): |
59 | if request.method == 'DELETE': | 59 | if request.method == 'DELETE': |
60 | print(pk) | 60 | print(pk) |
61 | - item = Item.objects.get(item_id = pk) | 61 | + item = get_object_or_None(Item, item_id=pk) |
62 | + if item != None: | ||
62 | if item.is_folder == True: # 폴더는 삭제 안되도록 처리 | 63 | if item.is_folder == True: # 폴더는 삭제 안되도록 처리 |
63 | return Response({'message': 'This item is folder.'}, status=status.HTTP_200_OK) | 64 | return Response({'message': 'This item is folder.'}, status=status.HTTP_200_OK) |
64 | item.is_deleted = True | 65 | item.is_deleted = True | ... | ... |
-
Please register or login to post a comment