1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
import contextlib
import tempfile
import boto3
from flask import request, send_file
from .base import BaseStorage
class Storage(BaseStorage):
def __init__(self, app):
super().__init__(app)
self.client = boto3.resource('s3', **self.app.config['S3_CONFIG'])
def _get_object_key(self, file_hash, user_id):
return '{}_{}'.format(file_hash, user_id)
def get_object_key(self, f):
return self._get_object_key(f.hash, f.user_id if f.user_id else 0)
def store_file(self, uploaded_file, file_hash, user, ip):
bucket = self.client.Bucket(self.app.config['S3_BUCKET'])
key = self._get_object_key(file_hash, user.id if user else 0)
obj = bucket.upload_fileobj(Fileobj=uploaded_file.stream, Key=key)
size = uploaded_file.content_length
if not size:
obj = self.client.ObjectSummary(self.app.config['S3_BUCKET'], key)
size = obj.size
return self.add_file(file_hash, uploaded_file.filename, size, user, ip)
def get_file(self, f):
obj = self.client.Object(self.app.config['S3_BUCKET'], self.get_object_key(f))
kwargs = {}
if 'Range' in request.headers:
kwargs['Range'] = request.headers['Range']
data = obj.get(**kwargs)
rv = send_file(data['Body'], attachment_filename=f.filename)
rv.headers['Content-Length'] = data['ContentLength']
rv.headers['Accept-Ranges'] = data['AcceptRanges']
if 'ContentRange' in data:
rv.headers['Content-Range'] = data['ContentRange']
rv.status_code = 206
return rv
def delete_file(self, f):
obj = self.client.Object(self.app.config['S3_BUCKET'], self.get_object_key(f))
obj.delete()
@contextlib.contextmanager
def temp_file(self, f):
obj = self.client.Object(self.app.config['S3_BUCKET'], self.get_object_key(f))
with tempfile.NamedTemporaryFile() as f:
obj.download_fileobj(f)
f.seek(0)
yield f
|
