Python3｜ZIPファイルを作成してロリポップのサーバにFTPするスクリプトを作りました

""" My note project workflow script.
Create 3 zip files from JPG files.
Upload all JPG/ZIP files to FTPS server.
"""
from ftplib import FTP_TLS
from pathlib import Path
import zipfile


WORK_DIR = '任意の作業ディレクトリ'
FTP_SERVER = 'サーバーURL'
FTP_USER = 'ユーザID'
FTP_PASSWORD = 'パスワード'
UPLOAD_PATH = 'アップロード先ディレクトリ'

p_num = ''  # Project number


def get_file_lists():
    """ Get compressed file lists.

    Returns:
        l_list: L-size jpg file list
        m_list: M
        s_list: S
    """

    path = Path(WORK_DIR)
    l_list, m_list, s_list = [], [], []
    for file_path in path.iterdir():
        if file_path.suffix == '.jpg':
            if 'L' in file_path.name:
                l_list.append(str(file_path.name))
            elif 'M' in file_path.name:
                m_list.append(str(file_path.name))
            else:
                s_list.append(str(file_path.name))

    return l_list, m_list, s_list


def create_zip_files():
    """ Create 3 zip files per L/M/S-size files. """

    # Create targets list
    targets_list = []
    l_list, m_list, s_list = get_file_lists()
    targets_list.append(l_list)
    targets_list.append(m_list)
    targets_list.append(s_list)

    # Get project number from L-size file name
    #   Project number is 2nd & 3rd characters of file name
    #   (ex. No40-L-01.jpg -> it is 40)
    num = l_list[0]
    global p_num
    p_num = num[2:4]

    # Create 3 zip files
    for i, size in enumerate(['L', 'M', 'S']):
        zip_file_path = WORK_DIR + f'No{p_num}-{size}.zip'
        print(f'Creation start: {zip_file_path}')

        with zipfile.ZipFile(zip_file_path, 'w',
                             compression=zipfile.ZIP_DEFLATED) as new_zip:
            for file_name in targets_list[i]:
                new_zip.write(WORK_DIR + f'{file_name}', arcname=file_name)
                print(f'File combining: {file_name}')

        print('_____Creation complated_____')


def ftps_upload_files():
    """ Upload jpg/zip files to FTPS(Passive mode). """

    # Get upload file list(jpg & zip)
    path = Path(WORK_DIR)
    upload_list = [p for p in path.iterdir()
                   if p.suffix == '.jpg' or p.suffix == '.zip']

    try:
        # Login FTP server
        ftps = FTP_TLS(FTP_SERVER)
        ftps.login(FTP_USER, FTP_PASSWORD)
        ftps.prot_p()
        ftps.set_pasv('true')
        print('login OK')

        # Make project number dir
        ftps.cwd(UPLOAD_PATH)
        global p_num
        p_num_dir = 'No' + p_num
        if p_num_dir in ftps.nlst('.'):
            print('This dir already exist.')
        else:
            ftps.mkd(p_num_dir)
            print('Create project dir.')
        ftps.cwd(p_num_dir)
        print(f'Upload directory: {ftps.pwd()}')

        # Upload files
        count = 1
        total_count = len(upload_list)
        for upload_file in upload_list:
            with open(upload_file, 'rb') as f:
                ftps.storbinary('STOR ' + upload_file.name, f)
            print(f'Uploading({count}/{total_count}): {upload_file}')
            count += 1
    except ftps.all_error as e:
        print(e)
        print('_____Upload failed_____')
    else:
        print('_____Upload complated_____')
    finally:
        ftps.quit()
        print('_____Close connection_____')


if __name__ == '__main__':
    create_zip_files()
    ftps_upload_files()