Kangsubeen

delete file in media

...@@ -5,7 +5,7 @@ from rest_framework.response import Response ...@@ -5,7 +5,7 @@ from rest_framework.response import Response
5 from rest_framework import status 5 from rest_framework import status
6 from rest_framework.authentication import SessionAuthentication, BasicAuthentication 6 from rest_framework.authentication import SessionAuthentication, BasicAuthentication
7 from rest_framework.permissions import IsAuthenticated 7 from rest_framework.permissions import IsAuthenticated
8 - 8 +import os
9 from restful import s3_interface 9 from restful import s3_interface
10 10
11 from restful.models import File 11 from restful.models import File
...@@ -39,6 +39,8 @@ class FileList(APIView): ...@@ -39,6 +39,8 @@ class FileList(APIView):
39 file_path = '.' + file_serializer.data.get('file') 39 file_path = '.' + file_serializer.data.get('file')
40 user = request.user 40 user = request.user
41 data = s3_interface.upload_file(s3_interface.BUCKET, user.username, file_path, path+file_path.split('/')[-1]) 41 data = s3_interface.upload_file(s3_interface.BUCKET, user.username, file_path, path+file_path.split('/')[-1])
42 + if os.path.exists(file_path):
43 + os.remove(file_path)
42 # TODO upload check 44 # TODO upload check
43 # TODO remove local file 45 # TODO remove local file
44 return Response(file_serializer.data, status=status.HTTP_201_CREATED) 46 return Response(file_serializer.data, status=status.HTTP_201_CREATED)
...@@ -66,6 +68,8 @@ class FileDetail(APIView): ...@@ -66,6 +68,8 @@ class FileDetail(APIView):
66 file = 'media/'+path.split('/')[-1] 68 file = 'media/'+path.split('/')[-1]
67 user = request.user 69 user = request.user
68 s3_interface.download_file(s3_interface.BUCKET, user.username, file, path) 70 s3_interface.download_file(s3_interface.BUCKET, user.username, file, path)
71 + if os.path.exists(file):
72 + os.remove(file)
69 # TODO error 73 # TODO error
70 return Response({'file': file}) 74 return Response({'file': file})
71 75
......