76 lines
3.2 KiB
Python
76 lines
3.2 KiB
Python
import os
|
||
import ftplib
|
||
import socks
|
||
import socket
|
||
from urllib.parse import urlparse
|
||
|
||
def set_proxy(proxy_url):
|
||
"""设置代理"""
|
||
if proxy_url:
|
||
parsed_proxy = urlparse(proxy_url)
|
||
proxy_host = parsed_proxy.hostname
|
||
proxy_port = parsed_proxy.port
|
||
socks.set_default_proxy(socks.SOCKS5, proxy_host, proxy_port) # 可以根据需要调整为 SOCKS5、SOCKS4 或 HTTP
|
||
socket.socket = socks.socksocket
|
||
print(f"已设置代理: {proxy_host}:{proxy_port}")
|
||
|
||
def upload_file_to_ftp(server, username, password, file_path, remote_path, proxy_url=None):
|
||
"""上传单个文件到FTP"""
|
||
try:
|
||
# 设置代理（如果有的话）
|
||
set_proxy(proxy_url)
|
||
|
||
with ftplib.FTP(server) as ftp:
|
||
ftp.login(user=username, passwd=password)
|
||
with open(file_path, 'rb') as file:
|
||
ftp.storbinary(f'STOR {remote_path}', file)
|
||
print(f"文件 {file_path} 成功上传到 {remote_path}！")
|
||
|
||
except ftplib.all_errors as e:
|
||
print(f"FTP 错误: {e}")
|
||
except Exception as e:
|
||
print(f"发生错误: {e}")
|
||
|
||
def create_remote_directory(ftp, remote_directory):
|
||
"""确保远程目录存在，如果不存在则创建"""
|
||
try:
|
||
ftp.cwd(remote_directory) # 尝试进入远程目录
|
||
except ftplib.error_perm:
|
||
ftp.mkd(remote_directory) # 如果远程目录不存在，则创建
|
||
ftp.cwd(remote_directory) # 再次进入远程目录
|
||
print(f"远程目录 {remote_directory} 已创建！")
|
||
|
||
def upload_directory_to_ftp(server, username, password, local_directory, remote_directory, proxy_url=None):
|
||
"""上传整个目录及其子目录到FTP服务器"""
|
||
try:
|
||
# 设置代理（如果有的话）
|
||
set_proxy(proxy_url)
|
||
|
||
with ftplib.FTP(server) as ftp:
|
||
ftp.login(user=username, passwd=password)
|
||
|
||
# 确保远程目录存在
|
||
create_remote_directory(ftp, remote_directory)
|
||
|
||
# 遍历本地目录树
|
||
for root, dirs, files in os.walk(local_directory):
|
||
# 上传子文件夹
|
||
for dir_name in dirs:
|
||
remote_dir_path = os.path.join(remote_directory, os.path.relpath(os.path.join(root, dir_name), local_directory)).replace("\\", "/")
|
||
try:
|
||
ftp.mkd(remote_dir_path) # 在FTP服务器上创建子目录
|
||
print(f"已创建远程目录: {remote_dir_path}")
|
||
except ftplib.error_perm:
|
||
pass # 如果目录已经存在则忽略
|
||
|
||
# 上传文件
|
||
for filename in files:
|
||
local_file_path = os.path.join(root, filename) # 本地文件的完整路径
|
||
relative_path = os.path.relpath(local_file_path, local_directory) # 计算相对路径
|
||
remote_file_path = os.path.join(remote_directory, relative_path).replace("\\", "/") # 计算远程文件路径
|
||
upload_file_to_ftp(server, username, password, local_file_path, remote_file_path, proxy_url) # 上传文件
|
||
|
||
except ftplib.all_errors as e:
|
||
print(f"上传目录时发生错误: {e}")
|
||
except Exception as e:
|
||
print(f"发生错误: {e}") |