61 lines
2 KiB
Python
61 lines
2 KiB
Python
import boto
|
|
import os
|
|
import logging
|
|
|
|
from boto.s3.key import Key
|
|
from uuid import uuid4
|
|
import hmac
|
|
import time
|
|
import urllib
|
|
import base64
|
|
import sha
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class S3FileWriteException(Exception):
|
|
pass
|
|
|
|
|
|
class UserRequestFiles(object):
|
|
def __init__(self, s3_access_key, s3_secret_key, bucket_name):
|
|
self._s3_conn = boto.s3.connection.S3Connection(s3_access_key,
|
|
s3_secret_key,
|
|
is_secure=False)
|
|
self._bucket_name = bucket_name
|
|
self._bucket = self._s3_conn.get_bucket(bucket_name)
|
|
self._access_key = s3_access_key
|
|
self._secret_key = s3_secret_key
|
|
self._prefix = 'userfiles'
|
|
|
|
def prepare_for_drop(self, mimeType):
|
|
""" Returns a signed URL to upload a file to our bucket. """
|
|
file_id = str(self._prefix + '/' + str(uuid4()))
|
|
|
|
expires = str(int(time.time() + 300))
|
|
signingString = "PUT\n\n" + mimeType + "\n" + expires + "\n/" + self._bucket_name + "/" + file_id;
|
|
|
|
hmac_signer = hmac.new(self._secret_key, signingString, sha)
|
|
signature = base64.b64encode(hmac_signer.digest())
|
|
|
|
url = "http://s3.amazonaws.com/" + self._bucket_name + "/" + file_id + "?AWSAccessKeyId=" + self._access_key + "&Expires=" + expires + "&Signature=" + urllib.quote(signature);
|
|
return (url, file_id)
|
|
|
|
def store_file(self, flask_file):
|
|
file_id = str(uuid4())
|
|
full_key = os.path.join(self._prefix, file_id)
|
|
k = Key(self._bucket, full_key)
|
|
logger.debug('Setting s3 content type to: %s' % flask_file.content_type)
|
|
k.set_metadata('Content-Type', flask_file.content_type)
|
|
bytes_written = k.set_contents_from_file(flask_file)
|
|
|
|
if bytes_written == 0:
|
|
raise S3FileWriteException('Unable to write file to S3')
|
|
|
|
return file_id
|
|
|
|
def get_file_url(self, file_id, expires_in=300):
|
|
full_key = os.path.join(self._prefix, file_id)
|
|
k = Key(self._bucket, full_key)
|
|
return k.generate_url(expires_in)
|