「お役立ち Twitter Bot を作りながら学ぶ AWS ドリル」を元に作ったBotを紹介します

お前、誰よ（自己紹介）

Python大好き にっきー
- Twitter @ftnext ／ GitHub @ftnext
株式会社ユーザベースのデータサイエンティスト（NLPer）

ツイートのエンドポイント

POST https://api.twitter.com/2/tweets

ドリルに沿って、OAuth 1.0a User Contextを使っています
https://developer.twitter.com/en/docs/twitter-api/tweets/manage-tweets/api-reference/post-tweets

ツイートするプログラム（Python）

import os
from requests_oauthlib import OAuth1Session

# 環境変数の読み込み
consumer_key = os.getenv("TWITTER_API_KEY")
client_secret = os.getenv("TWITTER_API_KEY_SECRET")
access_token = os.getenv("TWITTER_API_ACCESS_TOKEN")
access_token_secret = os.getenv("TWITTER_API_ACCESS_TOKEN_SECRET")

oauth = OAuth1Session(consumer_key, client_secret, access_token, access_token_secret)

payload = {"text": "#アイの歌声を聴かせて 面白いからみんな観て"}
response = oauth.post("https://api.twitter.com/2/tweets", json=payload)

ツイートするLambda関数の例

import os
from requests_oauthlib import OAuth1Session

consumer_key = os.getenv("TWITTER_API_KEY")
client_secret = os.getenv("TWITTER_API_KEY_SECRET")
access_token = os.getenv("TWITTER_API_ACCESS_TOKEN")
access_token_secret = os.getenv("TWITTER_API_ACCESS_TOKEN_SECRET")

oauth = OAuth1Session(consumer_key, client_secret, access_token, access_token_secret)

def lambda_handler(event, context):
    payload = {"text": "#アイの歌声を聴かせて 面白いからみんな観て"}
    response = oauth.post("https://api.twitter.com/2/tweets", json=payload)

Lambda関数の引数 `event`

def lambda_handler(event, context):
    # EventBridgeで発生させたevent(JSON)がevent引数に渡る
    # eventはPythonの辞書という型（他の言語でいうMap）
    # {"キー": "値"}

このeventを受け取って実行されたら、この振る舞いをすると実装

eventのカスタマイズ

EventBridgeの 入力トランスフォーマー
Lambda関数で参照するキーと値をeventに追加する

{"bot-mode": "time-signal"}

Lambda関数の実装（イメージ）

def lambda_handler(event, context):
    mode = event.get("bot-mode")
    if mode == "information":
        # informationモードの文面を作ってツイート
    elif mode == "time-signal":
        # time-signalモードの文面を作ってツイート
    ...

入力トランスフォーマーの設定例

入力パスは変数定義。今回は使わないので {} にしています
入力テンプレートは eventのJSON。何時ならこの値と固定しています

{
  "bot-mode": "time-signal"
}

第1回ドリルのCloudShell

mkdir twitter-ainouta-bot
cd twitter-ainouta-bot
python -m pip install -t ./ -r ../requirements.lock
cp ../lambda_function.py .
zip -r twitter-ainouta-bot.zip ./

aws lambda update-function-code \
    --function-name twitter-ainouta-bot \
    --zip-file fileb://twitter-ainouta-bot.zip | \
jq 'del(.Environment, .FunctionArn, .Role)'