import subprocess
import os
import schedule
import time
import logging
from pathlib import Path
def encrypt_folder_with_winrar(folder_path, password):
# WinRAR命令
winrar_command = r'C:\Program Files \WinRAR\WinRAR.exe'
# 获取当前日期
current_date = time.strftime("%Y-%m-%d", time.localtime())
# 获取桌面BD文件夹路径
desktop_path = str(Path.home() / "Desktop")
bd_folder_path = os.path.join(desktop_path, "DB")
# 创建DB文件夹（如果不存在）
if not os.path.exists(bd_folder_path):
os.mkdir(bd_folder_path)
# 压缩包文件名
compressed_filename = f'压缩包-{current_date}.rar'
# 压缩包路径
compressed_filepath = os.path.join(bd_folder_path, compressed_filename)
# 加密命令
encryption_command = [winrar_command, 'a', '-r', '-hp' + password, compressed_filepath, folder_path]
# 执行加密命令
subprocess.run(encryption_command)
return compressed_filepath
def encrypt_and_delete_folder(folder_path, password):
try:
# 压缩并加密文件夹
compressed_file = encrypt_folder_with_winrar(folder_path, password)
# 等待一段时间，确保压缩文件完成
time.sleep(10)
# 删除源文件夹中的文件
for root, dirs, files in os.walk(folder_path):
for file in files:
os.remove(os.path.join(root, file))
logger.info("加密打包操作成功")
except Exception as e:
logger.error(f"加密打包操作失败：{e}")
# 出现错误时，删除已创建的压缩文件
if os.path.exists(compressed_file):
os.remove(compressed_file)
logger.info("压缩文件已删除.")
# 创建日志记录器
logger = logging.getLogger("encrypt_and_delete")
logger.setLevel(logging.INFO)
# 创建文件处理器
log_file = "encrypt_and_delete.log"
file_handler = logging.FileHandler(log_file)
file_handler.setLevel(logging.INFO)
# 创建控制台处理器
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.INFO)
# 创建日志格式
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
file_handler.setFormatter(formatter)
console_handler.setFormatter(formatter)
# 添加处理器到记录器
logger.addHandler(file_handler)
logger.addHandler(console_handler)
# 文件夹路径
folder_path = r'D:\BACK'
# 密码
password = '123456'
# 每天定时执行加密打包操作
schedule.every().day.at("12:20").do(encrypt_and_delete_folder, folder_path=folder_path, password=password)
# 循环执行任务，并写入日志
while True:
schedule.run_pending()
time.sleep(60)
等保要求加密传输 奈何数据库比较老 不支持加密 所以暂时想到了这个办法 有需要的可以自行打包
