从私有AWS S3存储桶中在Flask中提供静态文件

时间:2018-09-15 08:20:16

标签: heroku amazon-s3 flask boto3 static-files

我正在开发在Heroku上运行的Flask应用,该应用允许用户上传图像。该应用程序有一个页面,用于在表格中显示用户的图像。

出于开发目的,我将上传的文件保存到Heroku的ephemeral文件系统中,并且一切正常:图像已正确加载和显示(我使用的是here中显示的最后一种方法,使用send_from_directory())。现在,我已将存储移至S3,并且正在尝试修改代码。我使用boto3将文件上传到存储桶:工作正常。 我的疑问与下载以用用户的图片填充用户页面有关

here所述,我可以将文件设置为“公开读取”并使用URL(我认为这是Flask-S3的作用),但我宁愿不要随意使用访问文件。 因此,我的解决方案尝试是将文件下载到Heroku的文件系统中,并再次使用send_from_directory()来提供图像,如下所示:

app.py

@app.route('/download/<resource>')
def download_image(resource):
    """ resource: name of the file to download"""
    s3 = boto3.client('s3',
                      aws_access_key_id=current_app.config['S3_ACCESS_KEY'],
                      aws_secret_access_key=current_app.config['S3_SECRET_KEY'])

    s3.download_file(current_app.config['S3_BUCKET_NAME'],
                     resource,
                     os.path.join('tmp',
                                  resource))

    return send_from_directory('tmp',  # Heroku's filesystem
                               resource,
                               as_attachment=False)

然后,我在模板中生成图像的URL,如下所示:

...
<img src="{{ url_for('app.download_image',
                     resource=resource) }}" height="120" width="120">
...

它可以工作,但是由于某些原因,我认为这不是正确的方法:其中,我应该管理Heroku的文件系统,以避免耗尽dynos重新启动之间的所有空间(我应该从文件系统中删除图像)

哪种是最佳/首选方式,同时还要考虑性能? 非常感谢

1 个答案:

答案 0 :(得分:7)

首选方法是简单地为图像create a pre-signed URL,然后将重定向返回到该URL。这样可以在S3中将文件保密,但会生成一个受时间限制的临时URL,该URL可用于直接从S3下载文件。这将大大减少服务器上发生的工作量,以及服务器消耗的数据传输量。像这样:

@app.route('/download/<resource>')
def download_image(resource):
    """ resource: name of the file to download"""
    s3 = boto3.client('s3',
                      aws_access_key_id=current_app.config['S3_ACCESS_KEY'],
                      aws_secret_access_key=current_app.config['S3_SECRET_KEY'])

    url = s3.generate_presigned_url('get_object', Params = {'Bucket': 'S3_BUCKET_NAME', 'Key': resource}, ExpiresIn = 100)
    return redirect(url, code=302)

如果您不喜欢该解决方案,则至少应考虑从S3流式传输文件内容,而不是将其写入文件系统。