python 批量scp

import os
import pexpect
import argparse

# host = ''
# user = ''
# password = '.'
scp_cmd = 'scp {} {}@{}:{}'
# dir_path = ''
# target_path = ''
files = []


def login(file, user, host, target_path, password):
    scp_handle = pexpect.spawn(scp_cmd.format(file, user, host, target_path))
    scp_handle.expect('password:')
    scp_handle.sendline(password)
    scp_handle.read()
    print('发送完毕：{}'.format(file))


def get_files(dir_path):
    files_name_list = os.listdir(dir_path)
    for i in files_name_list:
        file_path = os.path.join(dir_path, i)
        if os.path.isfile(file_path):
            files.append(file_path)


def main(args):
    get_files(args.dir)
    for file in files:
        login(file, args.user, args.ip, args.target, args.password)


if __name__ == '__main__':
    parse = argparse.ArgumentParser(description='scp scrapy')
    parse.add_argument('-i', '--ip', required=True, type=str, help='scp的ip地址')
    parse.add_argument('-u', '--user', required=True, type=str, help='scp用户')
    parse.add_argument('-p', '--password', required=True, type=str, help='scp的密码')
    parse.add_argument('-d', '--dir', required=True, type=str, help='需要传输的目录')
    parse.add_argument('-t', '--target', required=True, type=str, help='传送到的目录')
    args = parse.parse_args()
    main(args)
    print('end')