| 1234567891011121314151617181920212223242526272829303132 |
- from rest_framework import generics, status
- from rest_framework.response import Response
- from rest_framework.parsers import MultiPartParser
- from datetime import datetime
- from boto3.session import Session
- import os
- import environ
- from .serializers import FileSerializer
- env = environ.Env(DEBUG=(bool, False))
- environ.Env.read_env()
- class FileUploadView(generics.GenericAPIView):
- serializer_class = FileSerializer
- parser_classes = (MultiPartParser, )
- def post(self, request, format=None):
- serializer = self.get_serializer(data=request.FILES)
- serializer.is_valid(raise_exception=True)
- file_extension = os.path.splitext(str(request.FILES['file']))[1]
- filename = datetime.now().strftime("%d-%m-%YT%H:%M:%S") + file_extension
- session = Session(region_name=env('REGION_NAME'),
- aws_access_key_id=env('AWS_ACCESS_KEY_ID'),
- aws_secret_access_key=env('AWS_SECRET_ACCESS_KEY'))
- s3 = session.resource('s3')
- s3.Bucket(env('BUCKET')).put_object(Key=filename, Body=request.FILES['file'])
- return Response({"message": "Upload Successful"}, status=status.HTTP_200_OK)
|
