AWS lamdbaの中でファイルの中身を成形してS3に保存したい

Question

AWS lambdaで、sshで接続した先にある、テキストファイルを読み込んだあと、中身を書き換え、JSON形式の文字列に成形してS3に保存したいと考えております。

具体的には、test.txtの中身が

a
100　　※計算してintになっているもの
b
200
300 ※計算してintになっているもの

となっているものを、2,4,5行目の数字だけ取り出し、

{"数字1": 100, "数字2": 200, "数字3": 300}

という形に成形したいです。

エラーとしては以下の通りです。
エラーとなっているのは、「file_contents = '{"数字1": ' + lines[1] + ', "数字2": ' + lines[2] + ', "数字3": ' + lines[4] + '}'」のところです。
TypeError: must be str, not int

実際のコードは以下の通りになります。

import paramiko
import boto3
import datetime
import os

print('Loading testdata lambda function')
today = datetime.datetime.now(datetime.timezone(datetime.timedelta(hours=9))).strftime('%Y%m%d')
#now = datetime.datetime.now()
yesterday = (datetime.datetime.now(datetime.timezone(datetime.timedelta(hours=9))) - datetime.timedelta(days=1)).strftime('%Y%m%d')

def lambda_handler(event, context):

    #接続先情報の設定
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.WarningPolicy())
    user = os.environ['ID']
    passwd = os.environ['PASS']
    ip = os.environ['IP']
    client.connect(ip,port=22,username=user,password=passwd)

    #接続先サーバー実行コマンド
    stdin, stdout, stderr = client.exec_command('cat /home/tmp/test.txt')

    # アップロードファイルの内容格納
    lines = stdout.read()
    file_contents = '{"数字1": ' + lines[1] + ', "数字2": ' + lines[2] + ', "数字3": ' + lines[4] + '}'
    print(file_contents)
    file_contents = file_contents.replace('\n','')
    print(file_contents)
    client.close()
    bucket = os.environ['BUCKET']

    #取得ファイルの保存
    key = 'Testdata/' + 'json/' + today + '/' +'testdata_' + yesterday + '.json'
    print(key)
    s3 = boto3.resource('s3')
    obj = s3.Object(bucket,key)
    obj.put(Body=file_contents)
    return

ご教授のほどよろしくお願いいたします。

Answer 1

paramikoは使っていないので違うかもしれませんが、stdout.read()で返ってくるのはバイト列なのではないでしょうか。なので、lines[1]等が数値扱いされているんだと思います。

linesのtypeを確認してみてください。

バイト列をデコードして文字列に変換して、改行コードで分割すればいいと思います。

    lines = stdout.read().decode().split('\n')

もしかするとstdout.readlines()でもう少し簡単に書けるかもしれません。

Comments

[Shinjuku1904]

lines にしたことで無事解決できました！
ありがとうございます！