仅20行代码，实现文件自动化上传。

最新推荐文章于 2024-07-06 20:34:24 发布

Wu_Candy

最新推荐文章于 2024-07-06 20:34:24 发布

阅读量407

点赞数

分类专栏： Python 文章标签：自动化服务器 python

本文链接：https://blog.csdn.net/weixin_41754309/article/details/125258025

版权

本文介绍如何在含有特殊字符的密码环境中，利用Python编写20行代码实现SFTP服务器文件的自动化上传。主要涉及yaml模块读取配置、OptionParser处理命令行参数，确保文件能成功推送到指定SFTP节点。

摘要由CSDN通过智能技术生成

1. 背景

最近接到一个产品需求是给指定的 sftp 服务器的指定目录定时推送文件数据。

因为项目组已有现成的组件可以轻松实现 sftp 服务器文件的快速上传，本来是一件很容易的事情，但是问题出现在这个指定的 sftp 服务器所指定的密码带有系统关键字和一些特殊字符，导致现在的组件在解析过程中会失败。

因此重新开发了下面的这套脚本来满足这个特殊的需求。

2. 代码实现

Python代码

#test_upload.py
import os, sys
import yaml
from optparse import OptionParser
import paramiko
import traceback


opt_parser = OptionParser()
opt_parser.add_option('--node', action='store', type='string', dest='node', help='sftp config')
opt_parser.add_option('--local_file', action='store', type='string', dest='local_file',  help='file to be transferred')


def load_config_from_param_conf(key):
    """
    :param key: 指定业务所使用的sftp服务的名称，例如sftp.yaml文件中的key: test_file_upload
    :return: 从指定路径拿到yaml文件并读取指定key，返回例如sftp.yaml文件中key为test_file_upload下的配置选项值
    包含: host,port,user,password,destination
    """
    config_file = f'{os.environ["TASK_MAIN"]}/test_conf/sftp.yaml' #获取指定路径下的yaml文件
    with open(config_file, 'r') as r: #打开yaml文件内容
        config = yaml.load(r) #转换yaml数据为字典
    return config['sftp']['upload'][key]


def upload(sftp_conf, local_file_path):
    """
    :param sftp_conf: 获取指定sftp_conf=test_file_upload下返回的字典值，这里的参数sftp_conf与load_config_from_para