【Python3】S3のファイルをメモリ上でZIP化しSFTPでファイルを転送する。

import os
import traceback
from io import BytesIO
import zipfile
import socket

import paramiko
import boto3

AWS_PROFILE = 'XXXXXXXX'

host = 'XXX.XXX.X.X'
port = 22
username = 'XXXXXX'
sshkey_path = 'XXXXX.pem'


def s3_object_to_zip_obj(bucket, key):
    session = boto3.Session(profile_name=AWS_PROFILE)
    s3 = session.resource('s3')
    s3_bucket = s3.Bucket(bucket)
    filecols = s3_bucket.objects.filter(Prefix=key).all()

    archive = BytesIO()

    with zipfile.ZipFile(archive, 'w', zipfile.ZIP_DEFLATED) as zip_archive:
        for s3_obj in filecols:
            with zip_archive.open(s3_obj.key, 'w') as zip_f:
                zip_f.write(s3_obj.get()['Body'].read())

    archive.seek(0)
    return archive



if __name__ == '__main__':
    try:
        host_key = paramiko.RSAKey.from_private_key_file(sshkey_path)
        sshcon = paramiko.SSHClient()
        sshcon.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        sshcon.connect(host, port, username, pkey=host_key)
        sftp = sshcon.open_sftp()

        bucket = 'XXXX'
        key = 'XXXX/XXX.txt'
        zip_obj = s3_object_to_zip_obj(bucket, key)

        sftp_to = 'XXXX/XXXX.zip'
        sftp.putfo(zip_obj, sftp_to)
        zip_obj.close()

        sftp.close()
        sshcon.close()
    except:
        print('# error')
        print(traceback.format_exc())