81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
from ftplib import FTP
|
|
import os
|
|
|
|
|
|
# 使用示例
|
|
ftp_hosts = ["172.31.1.11","172.31.1.12","172.31.1.13","172.31.1.14","172.31.1.15","172.31.1.16","172.31.1.17"] # FTP 服务器地址
|
|
ftp_user = 'pi' # FTP 登录用户名
|
|
ftp_passwd = 'ph2008' # FTP 登录密码
|
|
#获取当前文件上一层目录，即项目目录 make2 底下
|
|
local_file = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
|
|
def upload_file(ftp_host, ftp_user, ftp_passwd, local_file, remote_file):
|
|
ftp = FTP(ftp_host)
|
|
ftp.login(user=ftp_user, passwd=ftp_passwd)
|
|
|
|
#判断 local_file 是文件还是目录
|
|
if os.path.isdir(local_file):
|
|
if remote_file not in ftp.nlst():
|
|
ftp.mkd(remote_file)
|
|
|
|
# 切换至远程目录
|
|
ftp.cwd(remote_file)
|
|
|
|
# 遍历本地目录
|
|
for file_name in os.listdir(local_file):
|
|
local_path = os.path.join(local_file, file_name)
|
|
|
|
if os.path.isfile(local_path):
|
|
# 如果是文件，上传文件
|
|
with open(local_path, 'rb') as file:
|
|
ftp.storbinary(f'STOR {file_name}', file)
|
|
ftp.quit()
|
|
return
|
|
|
|
|
|
# 打开本地文件
|
|
with open(local_file, 'rb') as file:
|
|
# 上传文件到指定目录
|
|
ftp.storbinary(f'STOR {remote_file}', file)
|
|
|
|
ftp.quit()
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
#获取输入的文件名,有路劲需要携带路劲
|
|
local_file = input("请输入文件名/目录：例如 main_step1.py 或者 tools/ftp.py 或者目录 tools ->")
|
|
if local_file == "":
|
|
print("文件不能为空")
|
|
exit()
|
|
#本地文件的路径构造
|
|
dirPath = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))+"/" + local_file
|
|
#检测文件是否存在
|
|
if os.path.exists(dirPath) == False:
|
|
print("文件/目录 不存在")
|
|
exit()
|
|
|
|
#请选择是否上传到 apps 目录下 还是上传到 make2 下
|
|
upload_path = input("请选择是否上传到 apps 目录下 还是上传到 make2 下 ->")
|
|
if upload_path != "apps" and upload_path != "make2":
|
|
print("不支持该目录上传")
|
|
exit()
|
|
|
|
#请选择是否上传到全部的服务器 11 到 17
|
|
isIp = input("请选择是否上传到全部的服务器 11 到 17,全部上传输入yes, 指定服务上传输入IP ->")
|
|
if isIp == "":
|
|
print("输入不能为空")
|
|
exit()
|
|
#远程服务的路径
|
|
remote_file = "/" + upload_path + "/" + local_file
|
|
|
|
#上传到全部的服务器上去
|
|
if isIp == "yes":
|
|
for ftp_host in ftp_hosts:
|
|
upload_file(ftp_host, ftp_user, ftp_passwd, dirPath, remote_file)
|
|
else:
|
|
#上传到指定的服务器上去
|
|
ftp_host = isIp
|
|
upload_file(ftp_host, ftp_user, ftp_passwd, dirPath, remote_file)
|
|
|
|
print("上传成功") |