Lambdaでのtarファイルの解凍、解凍後のファイルの保存

Question

こちらのソースコードを参考にしながら、tarファイルがS3にアップロードされたら解凍するソースコードを書いているのですが、zfile.read(filename)と同じことをするtarfileのコマンドがなく難儀しています。

##参考にしているソースコード

import boto3
import zipfile
import traceback
import os

print('Loading function')

s3 = boto3.resource('s3')
s3_client = boto3.client('s3')

def lambda_handler(event, context):

    # Get the object from the event and show its content type
    bucket = event['Records'][0]['s3']['bucket']['name']
    key = event['Records'][0]['s3']['object']['key']

    try: 
        s3_client.download_file(bucket, key, '/tmp/file.zip')
        print('download')
        zfile = zipfile.ZipFile('/tmp/file.zip')
        namelist = zfile.namelist()
        print(namelist)

        for filename in namelist:
            if not os.path.basename('/tmp/'+filename):
                os.mkdir('/tmp/'+filename)
            else:
                f = open('/tmp/' + str(filename), 'wb')
                data = zfile.read(filename)
                f.write(data)
                f.close()
        for filename in namelist:
            s3_client.upload_file('/tmp/'+filename, bucket, filename)

    except Exception as e:
        print(e)
        print(traceback.format_exc())

以下が、参考にしながら書き換えたソースコードです。ここからどうすればいいでしょうか。

import boto3
import tarfile
import traceback
import os

print('Loading function')

s3 = boto3.resource('s3')
s3_client = boto3.client('s3')

def lambda_handler(event, context):

    # Get the object from the event and show its content type
    bucket = event['Records'][0]['s3']['bucket']['name']
    key = event['Records'][0]['s3']['object']['key']

    try: 
        s3_client.download_file(bucket, key, '/tmp/file.tar')
        print('download')
        zfile = tarfile.TarFile('/tmp/file.tar')
        namelist = zfile.getnames()
        print(zfile)

        for filename in namelist:
            if not os.path.basename('/tmp/'+filename):
                os.mkdir('/tmp/'+filename)
            else:
                f = open('/tmp/' + str(filename), 'r')
                zfile.extractall()
                data = zfile.read(filename)
                f.write(data)
                f.close()
        for filename in namelist:
            s3_client.upload_file('/tmp/'+filename, bucket, filename)

    except Exception as e:
        print(e)
        print(traceback.format_exc())

Answer 1

11/20追記

こちらのようにすれば解決しました。回答ありがとうございました。

import boto3
import tarfile
import traceback
import os

print('Loading function')

s3 = boto3.resource('s3')
s3_client = boto3.client('s3')

def lambda_handler(event, context):

    # Get the object from the event and show its content type
    bucket = event['Records'][0]['s3']['bucket']['name']
    key = event['Records'][0]['s3']['object']['key']

    try: 
        s3_client.download_file(bucket, key, '/tmp/file.tar')
        print('download')
        zfile = tarfile.TarFile('/tmp/file.tar')
        namelist = zfile.getnames()
        print(namelist)
        for filename in namelist:
            if not os.path.basename('/tmp/'+filename):
                os.mkdir('/tmp/'+filename)
            else:
                f = open('/tmp/' + str(filename), 'wb')


                #修正箇所
                data = zfile.extractfile(filename).read()
                
                f.write(data)
                f.close()
                
        for filename in namelist:
            s3_client.upload_file('/tmp/'+filename, bucket, filename)                

                


    except Exception as e:
        print(e)
        print(traceback.format_exc())

ありがとうございました。

Answer 2

zfile.extractfile(filename).read()

で読めると思いますが、extractallだと何がまずいんでしょうか？

Comments

[skyhigh]

ありがとうございます。無事解決しました。