[Lambda]Dockerコンテナ上でLambdaを動作(PDFファイル作成、ダウンロード機能)

import json
import base64
import createpdf
import os

def handler(event, context):
    # PDF作成クラスインスタンス取得
    ss = createpdf.ReportlabView()

    # PDF作成処理依頼
    ss._create_pdf( '/tmp/sample.pdf', 'template.pdf' )
    
    # 作成されたファイルを読み込み、Base64に変換して返す
    with open("/tmp/sample.pdf", "rb") as file_data:
        text = file_data.read()
    
    text = base64.b64encode( text ).decode('utf-8')

    # 返す値はJSON形式の文字列で返す。ファイルダウンロードする際は以下の様な記述にする。
    return {
        "statusCode": 200,
        "headers": {
            'Content-Length': len(text),
            'Content-Type': 'application/pdf',
            'Content-disposition': 'attachment;filename=output.pdf'
        },
        "isBase64Encoded": True,
        "body": text
    }

from PyPDF2 import PdfFileWriter, PdfFileReader
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import A4, portrait
from reportlab.lib.units import inch, mm, cm
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.pdfbase.cidfonts import UnicodeCIDFont
from reportlab.platypus import Table, TableStyle
from reportlab.lib import colors

class ReportlabView():

    def _create_pdf(self, output_file, template_file):
        # ファイルの指定
        tmp_file = '/tmp/tmp.pdf' # 一時ファイル

        # Canvasを作成 
        w, h = portrait(A4)
        cv = canvas.Canvas(tmp_file, pagesize=(w, h))

        # フォントを登録しCanvasに設定
        font_size = 10
        font_name = 'HeiseiKakuGo-W5'
        pdfmetrics.registerFont(UnicodeCIDFont(font_name))
        
        cv.setFont(font_name, font_size)

        # 文字列を描画
        cv.drawString(
            20 * mm,
            264 * mm,
            "AAA株式会社御中"
        )
        
        cv.drawString(
            80 * mm,
            257 * mm,
            "AA案件"
        )
        
        data = [
            ["8800円"], #合計金額
        ]

        table = Table(data, colWidths=(60 * mm), rowHeights=(7 * mm))
        table.setStyle(
            TableStyle(
                [
                    ("FONT", (0, 0), (-1, -1), font_name, 10),
                    # ("BOX", (0, 0), (-1, -1), 1, colors.red),
                    # ("INNERGRID", (0, 0), (-1, -1), 1, colors.black),
                    ("VALIGN", (0, 0), (-1, -1), "MIDDLE"),
                    ("ALIGN", (0, 0), (-1, -1), "RIGHT"),
                ]
            )
        )
        table.wrapOn(
            cv,
            70 * mm,
            218 * mm,
        )
        table.drawOn(
            cv,
            70 * mm,
            218 * mm,
        )
        
        data = []

        data.append(["A案件", "1000", "2", "2000", "", ""])
        data.append(["B案件", "2000", "3", "6000", "", ""])

        
        table = Table(
            data,
            colWidths=(70 * mm, 25 * mm, 25 * mm, 20 * mm, 20 * mm, 20 * mm),
            rowHeights=6 * mm,
        )
        table.setStyle(
            TableStyle(
                [
                    ("FONT", (0, 0), (-1, -1), font_name, 8),
                    ("VALIGN", (0, 0), (-1, -1), "MIDDLE"),
                    ("ALIGN", (1, 0), (-1, -1), "RIGHT"),
                ]
            )
        )
        table.wrapOn(cv, 17 * mm, 184 * mm)
        table.drawOn(cv, 17 * mm, 184 * mm)

        # 一時ファイルに保存
        cv.showPage()
        cv.save()

        # テンプレートとなるPDFを読む
        template_pdf = PdfFileReader(template_file)
        template_page = template_pdf.getPage(0)

        # 一時ファイルを読んでマージする
        tmp_pdf = PdfFileReader(tmp_file)
        template_page.mergePage(tmp_pdf.getPage(0))

        # 出力用PDFを用意し書き込む
        output = PdfFileWriter()
        output.addPage(template_page)
        with open(output_file, "wb") as fp:
            output.write(fp)
            

FROM public.ecr.aws/lambda/python:3.8

RUN pip install pypdf2
RUN pip install reportlab

COPY template.pdf   ./
COPY app.py   ./
COPY createpdf.py   ./

CMD ["app.handler"]

docker build -t イメージ名 . 

docker run --rm -p 9000:8080 イメージ名:latest 

docker ps

CONTAINER ID   IMAGE          COMMAND                  CREATED      STATUS      PORTS                    NAMES
a9218fac6c25   イメージ名:latest   "/lambda-entrypoint.…"   2 days ago   Up 2 days   0.0.0.0:9000->8080/tcp   hardcore_feistel

Invoke-WebRequest http://localhost:9000/2015-03-31/functions/function/invocations -Method POST -Body '{}'

curl -X POST http://localhost:9000/2015-03-31/functions/function/invocations -d {}

StatusCode        : 200
StatusDescription : OK
Content           : {"statusCode": 200, "headers": {"Content-Length": 18448, "Content-Type": "application/pdf", "Conten
                    t-disposition": "attachment;filename=output.pdf"}, "isBase64Encoded": true, "body": "JVBERi0xLjQKJe
                    Lj...
RawContent        : HTTP/1.1 200 OK
                    Transfer-Encoding: chunked
                    Content-Type: text/plain; charset=utf-8
                    Date: Thu, 27 Oct 2022 07:42:15 GMT

                    {"statusCode": 200, "headers": {"Content-Length": 18448, "Content-Type": "a...
Forms             : {}
Headers           : {[Transfer-Encoding, chunked], [Content-Type, text/plain; charset=utf-8], [Date, Thu, 27 Oct 2022 0
                    7:42:15 GMT]}
Images            : {}
InputFields       : {}
Links             : {}
ParsedHtml        : mshtml.HTMLDocumentClass
RawContentLength  : 18634

import requests
import json
import base64

response = requests.post('http://localhost:9000/2015-03-31/functions/function/invocations', '{}')

json_str = json.loads(response.text)

#取得したJSONデータのbody部分はBase64デコードをしてファイルに保存します。
data = base64.b64decode( json_str['body'].encode() )
f = open('sample.pdf', 'wb')
f.write( data )

docker tag イメージID XXXXXXXX.dkr.ecr.ap-northeast-1.amazonaws.com/lambda-docker-sample:tag

aws ecr get-login-password --region  ap-northeast-1 | docker login --username AWS --password-stdin  XXXXXXXX.dkr.ecr.ap-northeast-1.amazonaws.com

docker push XXXXXXXX.dkr.ecr.ap-northeast-1.amazonaws.com/lambda-docker-sample:tag