Chaliceでの画像を返却するAPIの作り方

AWS製のPythonフレームワーク Chaliceを使って、画像を取得するAPIを作成しようと思ったときに、情報が出てこなかったので実装方法をメモしておきます。

一般的な、画像の返却方法を知っていたらそんなにはまらなかったかも。。。

Chaliceを使って画像を返却するAPI

返却したいファイルを開きバイナリデータを取得し、それを返却する際のbody部分にセットしてあげればOK

from chalice import Chalice, Response

app = Chalice(app_name='your-app-name')

@app.route('/', methods=['GET'])
def index():
    f = open('image-file-path', 'rb')
    image = f.read()
    f.close()
    return Response(
        image,
        status_code=200,
        headers={'content-type': 'image/png'}
    )