29 KiB
29 KiB
1. 在10.99.9.75机器配置同步中泰L2数据
1.1 创建脚本/home/admin/sftp/sync_L2_files.sh
#!/bin/bash
# 获取当前日期并格式化
YEAR=$(date +%Y)
DATE_DIR=$(date +%Y%m%d)
DATE_PATH=$(date +%m%d)
# 清理之前的数据
rm -rf /home/admin/sftp/L2/${YEAR}*
# 循环同步直到两个MD5.txt文件都存在
while true; do
# LFTP自动同步脚本
lftp -u boyinwhis,'TWhbso%&*107296398YWHbhs' -p 1121 10.99.6.45 << EOF
set cmd:fail-exit yes
mirror -c /SZL2/${YEAR}/${DATE_PATH}/ /home/admin/sftp/L2/${DATE_DIR}SZ
mirror -c /SHJYS/${YEAR}/${DATE_DIR}/ /home/admin/sftp/L2/${DATE_DIR}SH
quit
EOF
# 检查两个MD5.txt文件是否存在
if [ -f "/home/admin/sftp/L2/${DATE_DIR}SZ/MD5.txt" ] && [ -f "/home/admin/sftp/L2/${DATE_DIR}SH/MD5.txt" ]; then
# 两个MD5.txt文件均已存在,同步成功
# 写入complete文件
echo "${DATE_DIR}" > /home/admin/sftp/L2/complete
echo "已写入complete文件"
break
else
# MD5.txt文件不存在或不完整,重新同步...
# 清理已同步的数据以便重新同步
rm -rf "/home/admin/sftp/L2/${DATE_DIR}SZ" "/home/admin/sftp/L2/${DATE_DIR}SH"
sleep 60
fi
done
1.2 添加计划任务
crontab -e
# sync level2
0 20 * * 1-5 /home/admin/sftp/sync_L2_files.sh >/dev/null 2>&1
2. 在datacenter机器配置同步9.75L2数据
2.1 创建脚本sftp_sync_boyigo.py
# weiyongfu
# sftp sync
import paramiko
import os
import stat
from datetime import datetime
import sys
import threading
from concurrent.futures import ThreadPoolExecutor, as_completed
import time
import hashlib
import re
def check_remote_complete_file(sftp, current_date):
"""
检查远程complete文件中是否包含当前日期的时间戳
"""
remote_complete_path = "/list/10.99.9.75_sftp/L2/complete"
try:
# 尝试读取远程complete文件
with sftp.open(remote_complete_path, 'r') as remote_file:
content = remote_file.read().decode('utf-8')
# 检查是否包含当前日期
if current_date in content:
print(f"远程complete文件已包含当前日期 {current_date},开始同步")
return True
else:
print(f"远程complete文件未包含当前日期 {current_date},等待中...")
return False
except FileNotFoundError:
print(f"远程complete文件不存在: {remote_complete_path},等待中...")
return False
except Exception as e:
print(f"检查远程complete文件时出错: {e},等待中...")
return False
def create_local_complete_file(current_date, verification_passed=True, error_messages=None):
"""
在本地创建complete文件
只有在验证成功时才创建文件
"""
local_complete_dir = r"E:\成功"
local_complete_path = os.path.join(local_complete_dir, current_date)
try:
# 确保目录存在
os.makedirs(local_complete_dir, exist_ok=True)
# 只有在验证通过时才创建文件
if verification_passed:
# 创建文件并写入验证状态
with open(local_complete_path, 'w') as f:
timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
f.write(f"COMPLETE - {timestamp} - MD5 verification PASSED\n")
print(f"本地complete文件创建成功: {local_complete_path}")
print(f"MD5验证状态: 通过")
return True
else:
# 验证失败时,如果文件已存在,删除它
if os.path.exists(local_complete_path):
os.remove(local_complete_path)
print(f"删除失败的complete文件: {local_complete_path}")
return False
except Exception as e:
print(f"操作本地complete文件失败: {e}")
return False
def wait_for_remote_complete(hostname, port, username, private_key_path, current_date):
"""
等待远程complete文件包含当前日期的时间戳
"""
print(f"等待远程complete文件包含日期: {current_date}")
while True:
ssh_client = paramiko.SSHClient()
ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
# 加载私钥
private_key = load_private_key(private_key_path)
if not private_key:
print("私钥加载失败,1分钟后重试...")
time.sleep(60)
continue
# 建立连接
ssh_client.connect(hostname, port, username, pkey=private_key, timeout=30)
sftp = ssh_client.open_sftp()
# 检查complete文件
if check_remote_complete_file(sftp, current_date):
sftp.close()
ssh_client.close()
return True
sftp.close()
ssh_client.close()
except Exception as e:
print(f"连接检查失败: {e}")
# 等待1分钟
print("等待60秒后重试...")
time.sleep(60)
def load_private_key(private_key_path):
"""
加载私钥文件
"""
try:
# 首先尝试作为RSA密钥加载
return paramiko.RSAKey.from_private_key_file(private_key_path)
except paramiko.ssh_exception.PasswordRequiredException:
# 如果私钥有密码保护,需要输入密码
key_password = input("请输入私钥密码: ")
return paramiko.RSAKey.from_private_key_file(private_key_path, password=key_password)
except Exception as e1:
print(f"RSA密钥加载失败: {e1}")
try:
# 尝试作为ECDSA密钥加载
return paramiko.ECDSAKey.from_private_key_file(private_key_path)
except Exception as e2:
print(f"ECDSA密钥加载失败: {e2}")
try:
# 尝试作为Ed25519密钥加载
return paramiko.Ed25519Key.from_private_key_file(private_key_path)
except Exception as e3:
print(f"所有密钥格式尝试失败: {e3}")
return None
def calculate_file_md5(file_path, chunk_size=8192):
"""
计算文件的MD5值
"""
md5_hash = hashlib.md5()
try:
with open(file_path, 'rb') as f:
# 逐块读取文件以节省内存
for chunk in iter(lambda: f.read(chunk_size), b""):
md5_hash.update(chunk)
return md5_hash.hexdigest().lower() # 返回小写十六进制字符串
except Exception as e:
print(f"计算文件MD5时出错 {file_path}: {e}")
return None
def parse_md5_file(md5_file_path):
"""
解析MD5.txt文件,提取文件名和MD5值
支持多种编码格式
"""
md5_dict = {}
# 尝试多种编码格式
encodings = ['utf-8', 'gbk', 'gb2312', 'gb18030', 'latin1', 'iso-8859-1']
for encoding in encodings:
try:
with open(md5_file_path, 'r', encoding=encoding) as f:
content = f.read()
# 使用正则表达式提取文件名和MD5值
# 格式: MD5 哈希(文件 StockTick.7z.065):
# 17 9a a5 ff f3 a4 8b 76 ee 2e 45 84 68 77 1b 8e
pattern = r'MD5\s*哈希\s*\(\s*文件\s*([^)]+)\s*\)\s*:\s*([0-9a-fA-F ]+)'
matches = re.findall(pattern, content, re.MULTILINE)
if not matches:
# 尝试另一种可能的格式
pattern2 = r'MD5\s*哈希\s*\(\s*文件\s*([^)]+)\s*\)\s*:\s*([0-9a-fA-F ]+)'
matches = re.findall(pattern2, content, re.MULTILINE)
if matches:
for filename, md5_hex in matches:
# 清理MD5值:移除空格并转换为小写
cleaned_md5 = md5_hex.replace(' ', '').lower()
md5_dict[filename.strip()] = cleaned_md5
print(f"使用编码 {encoding} 从 {md5_file_path} 中解析出 {len(md5_dict)} 个MD5记录")
return md5_dict
except UnicodeDecodeError:
continue # 尝试下一种编码
except Exception as e:
print(f"使用编码 {encoding} 解析MD5文件失败 {md5_file_path}: {e}")
continue
# 如果所有编码都失败,尝试二进制读取
try:
with open(md5_file_path, 'rb') as f:
content = f.read()
# 将二进制内容转换为字符串,使用errors='ignore'忽略无法解码的字符
content_str = content.decode('utf-8', errors='ignore')
pattern = r'MD5\s*哈希\s*\(\s*文件\s*([^)]+)\s*\)\s*:\s*([0-9a-fA-F ]+)'
matches = re.findall(pattern, content_str, re.MULTILINE)
if not matches:
pattern2 = r'MD5\s*哈希\s*\(\s*文件\s*([^)]+)\s*\)\s*:\s*([0-9a-fA-F ]+)'
matches = re.findall(pattern2, content_str, re.MULTILINE)
if matches:
for filename, md5_hex in matches:
cleaned_md5 = md5_hex.replace(' ', '').lower()
md5_dict[filename.strip()] = cleaned_md5
print(f"使用二进制方式从 {md5_file_path} 中解析出 {len(md5_dict)} 个MD5记录")
return md5_dict
except Exception as e:
print(f"二进制方式解析MD5文件失败 {md5_file_path}: {e}")
print(f"所有编码尝试都失败,无法解析MD5文件: {md5_file_path}")
return None
def verify_files(target_directory, md5_dict):
"""
验证目录中的文件MD5值
"""
results = []
verified_count = 0
total_files = len(md5_dict)
for filename, expected_md5 in md5_dict.items():
file_path = os.path.join(target_directory, filename)
if not os.path.exists(file_path):
error_msg = f"文件不存在: {filename}"
print(f" ✗ {error_msg}")
results.append(error_msg)
continue
# 计算实际MD5值
actual_md5 = calculate_file_md5(file_path)
if actual_md5 is None:
error_msg = f"MD5计算失败: {filename}"
print(f" ✗ {error_msg}")
results.append(error_msg)
elif actual_md5 == expected_md5:
print(f" ✓ {filename}: MD5验证通过")
verified_count += 1
else:
error_msg = f"{filename}: MD5不匹配 (期望: {expected_md5}, 实际: {actual_md5})"
print(f" ✗ {error_msg}")
results.append(error_msg)
return results, verified_count, total_files
def test_all_file_md5(filepath):
"""
验证指定目录下的所有文件MD5值
返回: (是否全部验证通过, 错误消息列表)
"""
md5_file_path = os.path.join(filepath, "MD5.txt")
target_directory = filepath
print(f"\n开始验证目录: {filepath}")
# 检查MD5文件是否存在
if not os.path.exists(md5_file_path):
error_msg = f"MD5.txt文件不存在: {md5_file_path}"
print(f" ✗ {error_msg}")
return False, [error_msg]
print("解析MD5文件...")
try:
md5_dict = parse_md5_file(md5_file_path)
if not md5_dict:
error_msg = f"解析MD5文件失败或文件为空: {md5_file_path}"
print(f" ✗ {error_msg}")
return False, [error_msg]
print(f"成功解析 {len(md5_dict)} 个文件的MD5记录")
except Exception as e:
error_msg = f"解析MD5文件失败: {str(e)}"
print(f" ✗ {error_msg}")
return False, [error_msg]
print("\n开始验证文件...")
results, verified_count, total_files = verify_files(target_directory, md5_dict)
print(f"\n验证完成: {filepath}")
print(f"总文件数: {total_files}")
print(f"通过验证: {verified_count}")
print(f"问题文件: {len(results)}")
if results:
print("\n发现问题:")
for msg in results:
print(f" - {msg}")
return False, results
else:
print("\n所有文件验证通过!")
return True, []
def get_failed_files_from_results(filepath, results):
"""
从验证结果中提取失败的文件名
"""
failed_files = []
for result in results:
# 从错误消息中提取文件名
# 格式可能是: "文件不存在: StockTick.7z.001" 或 "StockTick.7z.001: MD5不匹配..."
if "文件不存在:" in result:
filename = result.split("文件不存在: ")[1].strip()
elif "MD5不匹配" in result or "MD5计算失败" in result:
filename = result.split(":")[0].strip()
else:
continue
failed_files.append(filename)
return failed_files
def retry_failed_files(hostname, port, username, private_key, sync_paths, date_ymd, date_md, failed_files_dict):
"""
重新下载验证失败的文件
"""
print(f"\n{'=' * 60}")
print("开始重新下载验证失败的文件")
print(f"{'=' * 60}")
# 统计信息
stats = {
'total_files': 0,
'downloaded': 0,
'skipped': 0,
'failed': 0
}
stats_lock = threading.Lock()
# 为每个失败的文件找到对应的远程路径
all_files_to_retry = []
for path_info in sync_paths:
remote_path = path_info["remote"]
local_base_path = path_info["local"]
name = path_info["name"]
if name not in failed_files_dict or not failed_files_dict[name]:
continue
# 获取该路径的失败文件列表
failed_files_in_path = failed_files_dict[name]
print(f"\n重新下载 {name} 的 {len(failed_files_in_path)} 个失败文件")
# 为每个失败文件构建文件信息
for filename in failed_files_in_path:
remote_file_path = os.path.join(remote_path, filename).replace('\\', '/')
local_file_path = os.path.join(local_base_path, filename)
# 获取远程文件属性
try:
ssh_client = paramiko.SSHClient()
ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh_client.connect(hostname, port, username, pkey=private_key, timeout=30)
sftp = ssh_client.open_sftp()
remote_attr = sftp.stat(remote_file_path)
all_files_to_retry.append((remote_file_path, local_file_path, remote_attr))
sftp.close()
ssh_client.close()
except Exception as e:
print(f" 获取远程文件信息失败 {filename}: {e}")
if not all_files_to_retry:
print("没有需要重新下载的文件")
return stats
stats['total_files'] = len(all_files_to_retry)
print(f"\n总共需要重新下载 {stats['total_files']} 个文件")
# 使用多线程重新下载文件
max_workers = 10
print(f"开始多线程重新下载 (线程数: {max_workers})...")
with ThreadPoolExecutor(max_workers=max_workers) as executor:
# 提交所有下载任务
future_to_file = {}
for file_info in all_files_to_retry:
future = executor.submit(
download_file_with_new_connection,
file_info, stats, stats_lock,
hostname, port, username, private_key
)
future_to_file[future] = file_info
# 处理完成的任务
completed = 0
for future in as_completed(future_to_file):
result = future.result()
completed += 1
if completed % 5 == 0: # 每完成5个文件打印一次进度
print(f" 重试进度: {completed}/{stats['total_files']}")
# 打印重试统计信息
print(f"\n重试完成!")
print("=" * 50)
print(f"重试文件数: {stats['total_files']}")
print(f"成功下载: {stats['downloaded']}")
print(f"跳过文件: {stats['skipped']}")
print(f"失败文件: {stats['failed']}")
print("=" * 50)
return stats
def verify_and_retry(hostname, port, username, private_key, sync_paths, date_ymd, date_md, max_retries=10):
"""
验证所有目录的MD5,并在失败时重试下载
"""
print(f"\n{'=' * 60}")
print(f"开始MD5验证与重试 - 日期: {date_ymd}")
print(f"{'=' * 60}")
all_passed = False
all_error_messages = []
failed_files_dict = {} # 按目录存储失败文件
for retry_count in range(max_retries + 1):
if retry_count > 0:
print(f"\n{'=' * 60}")
print(f"第 {retry_count} 次重试下载验证失败的文件")
print(f"{'=' * 60}")
# 重新下载失败的文件
retry_stats = retry_failed_files(
hostname, port, username, private_key,
sync_paths, date_ymd, date_md, failed_files_dict
)
# 如果有文件重试失败,直接退出
if retry_stats['failed'] > 0:
print(f"重试下载失败,无法继续验证")
all_error_messages.append(f"第{retry_count}次重试下载失败")
break
# 清空之前的失败记录
all_error_messages.clear()
failed_files_dict.clear()
current_failed = False
# 验证每个目录
for path_info in sync_paths:
local_path = path_info["local"]
name = path_info["name"]
if not os.path.exists(local_path):
error_msg = f"目录不存在: {local_path}"
print(f" ✗ {error_msg}")
all_error_messages.append(error_msg)
current_failed = True
failed_files_dict[name] = [] # 目录不存在,无法获取失败文件列表
continue
# 验证该目录的MD5
passed, error_msgs = test_all_file_md5(local_path)
if not passed:
current_failed = True
all_error_messages.extend([f"{name}: {msg}" for msg in error_msgs])
# 提取失败的文件名
failed_files = get_failed_files_from_results(local_path, error_msgs)
failed_files_dict[name] = failed_files
else:
failed_files_dict[name] = []
# 检查是否全部通过
if not current_failed:
all_passed = True
print(f"\n所有MD5验证通过!")
break
elif retry_count < max_retries:
print(f"\n第 {retry_count + 1} 次验证失败,准备重试...")
time.sleep(2) # 等待2秒再进行重试
else:
print(f"\n已达到最大重试次数 ({max_retries}),停止重试")
return all_passed, all_error_messages
def sync_remote_to_local():
# 服务器连接信息
hostname = "59.36.23.197"
port = 16888
username = "boyigo"
# 私钥文件路径
private_key_path = r"C:\Users\admin\Desktop\boyigo"
# 获取当天日期
today = datetime.now()
year = today.strftime("%Y")
month = today.strftime("%m")
day = today.strftime("%d")
date_ymd = today.strftime("%Y%m%d") # 格式: 20251114
date_md = today.strftime("%m%d") # 格式: 1114
# 等待远程complete文件包含当前日期
if not wait_for_remote_complete(hostname, port, username, private_key_path, date_ymd):
print("等待远程complete文件超时或失败")
return False
# 根据日期动态生成同步路径
sync_paths = [
{
"remote": f"/list/10.99.9.75_sftp/L2/{date_ymd}SH",
"local": fr"E:\data\SHL2\{year}\{date_ymd}",
"name": "SHL2"
},
{
"remote": f"/list/10.99.9.75_sftp/L2/{date_ymd}SZ",
"local": fr"E:\data\SZL2\{year}\{date_md}",
"name": "SZL2"
}
]
# 统计信息
stats = {
'total_files': 0,
'downloaded': 0,
'skipped': 0,
'failed': 0
}
stats_lock = threading.Lock()
# 创建SSH客户端
ssh_client = paramiko.SSHClient()
ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
print(f"{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - 开始连接服务器进行同步...")
print(f"主机: {hostname}:{port}, 用户: {username}")
# 加载私钥文件
private_key = load_private_key(private_key_path)
if not private_key:
print("私钥加载失败")
return False
# 使用密钥建立连接
ssh_client.connect(hostname, port, username, pkey=private_key, timeout=30)
sftp_main = ssh_client.open_sftp()
print("连接成功,开始同步文件...")
print(f"同步日期: {date_ymd}")
# 收集所有需要同步的文件
all_files = []
for path_info in sync_paths:
remote_path = path_info["remote"]
local_base_path = path_info["local"]
name = path_info["name"]
print(f"\n扫描目录: {remote_path} ({name})")
files_in_path = scan_remote_directory(sftp_main, remote_path, local_base_path)
print(f" 找到 {len(files_in_path)} 个文件")
all_files.extend(files_in_path)
stats['total_files'] = len(all_files)
print(f"\n总共发现 {stats['total_files']} 个文件需要同步")
if stats['total_files'] == 0:
print("没有文件需要同步")
# 不创建complete文件
return False
# 使用多线程下载文件 - 10个线程
max_workers = 10
print(f"开始多线程下载 (线程数: {max_workers})...")
with ThreadPoolExecutor(max_workers=max_workers) as executor:
# 提交所有下载任务
future_to_file = {}
for file_info in all_files:
future = executor.submit(
download_file_with_new_connection,
file_info, stats, stats_lock,
hostname, port, username, private_key
)
future_to_file[future] = file_info
# 处理完成的任务
completed = 0
for future in as_completed(future_to_file):
result = future.result()
completed += 1
if completed % 10 == 0: # 每完成10个文件打印一次进度
print(f" 进度: {completed}/{stats['total_files']}")
# 打印统计信息
print(f"\n{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - 文件同步完成!")
print("=" * 50)
print(f"总文件数: {stats['total_files']}")
print(f"成功下载: {stats['downloaded']}")
print(f"跳过文件: {stats['skipped']}")
print(f"失败文件: {stats['failed']}")
print("=" * 50)
# 检查是否有失败的文件
if stats['failed'] > 0:
print(f"警告: 有 {stats['failed']} 个文件下载失败,MD5验证可能不完整")
# 同步完成后进行MD5验证,并在失败时重试
md5_verification_passed, error_messages = verify_and_retry(
hostname, port, username, private_key,
sync_paths, date_ymd, date_md, max_retries=10
)
# 根据验证结果创建complete文件
if md5_verification_passed:
print(f"\n所有MD5验证通过,创建成功标记文件")
# 只在成功时创建文件
create_local_complete_file(date_ymd, verification_passed=True)
return True
else:
print(f"\nMD5验证失败,不创建成功标记文件")
# 验证失败时不创建文件
create_local_complete_file(date_ymd, verification_passed=False, error_messages=error_messages)
return False
except Exception as e:
error_msg = f"同步过程中出现错误: {e}"
print(error_msg)
import traceback
traceback.print_exc()
# 发生异常时不创建文件
return False
finally:
# 关闭连接
try:
sftp_main.close()
except:
pass
try:
ssh_client.close()
except:
pass
return True
def scan_remote_directory(sftp, remote_path, local_base_path):
"""扫描远程目录,返回所有需要同步的文件列表"""
files_list = []
try:
remote_items = sftp.listdir_attr(remote_path)
except Exception as e:
print(f" 无法访问远程目录 {remote_path}: {e}")
return files_list
for item in remote_items:
remote_item_path = os.path.join(remote_path, item.filename).replace('\\', '/')
if stat.S_ISDIR(item.st_mode):
# 如果是目录,递归扫描
sub_local_path = os.path.join(local_base_path, item.filename)
sub_files = scan_remote_directory(sftp, remote_item_path, sub_local_path)
files_list.extend(sub_files)
else:
# 如果是文件,添加到列表
local_file_path = os.path.join(local_base_path, item.filename)
files_list.append((remote_item_path, local_file_path, item))
return files_list
def download_file(sftp, file_info, stats, stats_lock):
"""下载单个文件"""
remote_file_path, local_file_path, remote_attr = file_info
try:
# 确保本地目录存在
local_dir = os.path.dirname(local_file_path)
os.makedirs(local_dir, exist_ok=True)
# 检查本地文件是否存在
if os.path.exists(local_file_path):
local_stat = os.stat(local_file_path)
local_size = local_stat.st_size
local_mtime = local_stat.st_mtime
# 获取远程文件修改时间
remote_mtime = remote_attr.st_mtime
# 如果文件大小和修改时间都相同,则跳过
if local_size == remote_attr.st_size and abs(local_mtime - remote_mtime) < 2:
with stats_lock:
stats['skipped'] += 1
return "skipped"
else:
print(f" 更新: {os.path.basename(remote_file_path)} (大小: {local_size} -> {remote_attr.st_size})")
else:
print(f" 下载: {os.path.basename(remote_file_path)} (大小: {remote_attr.st_size} bytes)")
# 下载文件
sftp.get(remote_file_path, local_file_path)
# 验证下载的文件大小
downloaded_size = os.path.getsize(local_file_path)
if downloaded_size != remote_attr.st_size:
# 如果大小不匹配,删除文件并重试一次
os.remove(local_file_path)
sftp.get(remote_file_path, local_file_path)
downloaded_size = os.path.getsize(local_file_path)
if downloaded_size != remote_attr.st_size:
raise Exception(f"文件大小不匹配: 期望 {remote_attr.st_size}, 实际 {downloaded_size}")
# 保持文件修改时间
os.utime(local_file_path, (remote_attr.st_atime, remote_attr.st_mtime))
with stats_lock:
stats['downloaded'] += 1
return "downloaded"
except Exception as e:
print(f" 下载失败 {os.path.basename(remote_file_path)}: {e}")
with stats_lock:
stats['failed'] += 1
return "failed"
def download_file_with_new_connection(file_info, stats, stats_lock,
hostname, port, username, private_key):
"""为多线程创建的下载函数,每个线程有自己的SFTP连接"""
remote_file_path, local_file_path, remote_attr = file_info
# 为这个线程创建独立的SFTP连接
ssh_client = paramiko.SSHClient()
ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
sftp = None
try:
ssh_client.connect(hostname, port, username, pkey=private_key, timeout=30)
sftp = ssh_client.open_sftp()
# 调用原有的下载逻辑
return download_file(sftp, file_info, stats, stats_lock)
except Exception as e:
print(f" 连接失败 {os.path.basename(remote_file_path)}: {e}")
with stats_lock:
stats['failed'] += 1
return "failed"
finally:
# 关闭连接
try:
if sftp:
sftp.close()
except:
pass
try:
ssh_client.close()
except:
pass
def main():
"""主函数"""
try:
print("=" * 60)
print("远程文件同步工具 (10线程SFTP + 密钥认证 + MD5验证 + 失败重试)")
print("=" * 60)
success = sync_remote_to_local()
if success:
print("\n同步成功且MD5验证通过!")
return 0
else:
print("\n同步失败或MD5验证未通过!")
return 1
except Exception as e:
print(f"程序发生未捕获异常: {e}")
import traceback
traceback.print_exc()
return 1
if __name__ == "__main__":
try:
exit_code = main()
except KeyboardInterrupt:
print("\n程序被用户中断")
exit_code = 1
except Exception as e:
print(f"程序发生严重错误: {e}")
import traceback
traceback.print_exc()
exit_code = 1
# 添加延时,确保所有输出都能显示
time.sleep(1)
print("\n程序执行完成,即将退出...")
input("按Enter键退出...")
sys.exit(exit_code)
2.2 添加计划任务
每周一至周五 执行 C:\ProgramData\anaconda3\python.exe C:\Users\admin\Desktop\sftp_sync_boyigo.py