Lambda，AWS和Python的自動(dòng)化管理操作 - 自動(dòng)Resize S3上傳的圖片

這節(jié)來(lái)學(xué)習(xí)一個(gè)經(jīng)典的案例，這個(gè)案例我在三個(gè)不同的培訓(xùn)視頻里面都看見(jiàn)過(guò)，不知道最初的原創(chuàng)者是誰(shuí) :)

和前面的幾個(gè)例子比較起來(lái)，思路其實(shí)是大同小異的，唯一的區(qū)別在于兩點(diǎn)：

1. 通過(guò)S3來(lái)觸發(fā)Lambda
2. 對(duì)于AWS默認(rèn)不包括的Python模塊，我們需要手動(dòng)上傳zip文件或者通過(guò)S3來(lái)導(dǎo)入

這個(gè)例子的流程簡(jiǎn)單的說(shuō)就是創(chuàng)建2個(gè)S3的bucket，在其中一個(gè)上傳照片，他會(huì)自動(dòng)壓縮尺寸并保存在另外一個(gè)bucket里面。 下面來(lái)看看如何實(shí)現(xiàn)。

首先創(chuàng)建2個(gè)bucket，一個(gè)source， 一個(gè)destination

然后創(chuàng)建對(duì)應(yīng)的role

然后創(chuàng)建一個(gè)Lambda function，選擇上面配置的role

添加一個(gè)觸發(fā)器，這里我們指定S3的source bucket

接下來(lái)，是配置對(duì)應(yīng)的函數(shù)

import os
import tempfile

import boto3
from PIL import Image

s3 = boto3.client('s3')
DEST_BUCKET = os.environ['DEST_BUCKET']
SIZE = 128, 128

def lambda_handler(event, context):

    for record in event['Records']:
        source_bucket = record['s3']['bucket']['name']
        key = record['s3']['object']['key']
        thumb = 'thumb-' + key
        with tempfile.TemporaryDirectory() as tmpdir:
            download_path = os.path.join(tmpdir, key)
            upload_path = os.path.join(tmpdir, thumb)
            s3.download_file(source_bucket, key, download_path)
            generate_thumbnail(download_path, upload_path)
            s3.upload_file(upload_path, DEST_BUCKET, thumb)

        print('Thumbnail image saved at {}/{}'.format(DEST_BUCKET, thumb))

def generate_thumbnail(source_path, dest_path):
    print('Generating thumbnail from:', source_path)
    with Image.open(source_path) as image:
        image.thumbnail(SIZE)
        image.save(dest_path)

注意!!!這個(gè)函數(shù)里面我們調(diào)用了Pillow這個(gè)圖片的模塊，但是這個(gè)模塊默認(rèn)在aws的運(yùn)行環(huán)境里面是沒(méi)有的，因此我們需要手動(dòng)上傳。

首先從 https://pypi.org/project/Pillow/#files 上面下載對(duì)應(yīng)的linux whl 文件，whl文件其實(shí)是一個(gè)壓縮包，Windows 下面我們可以用 wheel unpack 解壓。如果在Linux環(huán)境下，可以直接用unzip打開(kāi)。

然后把這個(gè)PIL文件夾和我們的python文件一起zip

然后上傳到Lambda的控制臺(tái)

注意py文件和入口函數(shù)的名字要和handler匹配

也別忘記了配置環(huán)境變量

我們可以通過(guò)測(cè)試模板來(lái)查看對(duì)應(yīng)的事件的json格式

最后來(lái)看看運(yùn)行效果

上傳幾個(gè)圖片

自動(dòng)壓縮保存在另外一個(gè)bucket里面

實(shí)驗(yàn)成功