Python--机器人自动备份文件到自动递增ZIP

import zipfile
import os

  

def backupToZip(folder):

    # 1. 确保传入的文件夹路径是绝对路径

    folder = os.path.abspath(/root/frp)
    

    # 提取文件夹的名字，比如 '/var/log/nginx' 会提取出 'nginx'

    folder_name = os.path.basename(folder)
    


    # 2. 自动判定版本号 (核心逻辑)

    number = 1

    while True:

        # 拼接即将生成的 ZIP 文件名

        zip_filename = f'{folder_name}_backup_V{number}.zip'

        # 如果这个文件不存在，说明我们找到了一个可用的新版本号！跳出循环

        if not os.path.exists(zip_filename):

            break

        # 如果存在，编号加 1，继续找下一个
import zipfile

import os

#import sys

  

def backupToZip(folder):

    """

    将整个文件夹的内容备份到一个带有递增版本号的 ZIP 文件中。

    """

    # 1. 确保传入的文件夹路径是绝对路径

    folder = os.path.abspath(folder)

    # 提取文件夹的名字，比如 '/var/log/nginx' 会提取出 'nginx'

    folder_name = os.path.basename(folder)

  

    # 2. 自动判定版本号 (核心逻辑)

    number = 1

    while True:

        # 拼接即将生成的 ZIP 文件名

        zip_filename = f'{folder_name}_backup_{number}.zip'

        # 如果这个文件不存在，说明我们找到了一个可用的新版本号！跳出循环

        if not os.path.exists(zip_filename):

            break

        # 如果存在，编号加 1，继续找下一个

        number = number + 1

  
  

    # 3. 创建 ZIP 压缩文件

    print(f'备份机器人正在创建备份归档: {zip_filename} ...')

    backup_zip = zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED)

  
  

    # 4. 遍历目录树并添加到 ZIP 文件中

    # os.walk 会像剥洋葱一样，一层层遍历文件夹里的所有内容

    for foldername, subfolders, filenames in os.walk(folder):

        print(f'正在添加文件夹: {foldername} ...')

        # 把当前文件夹本身加进压缩包

        backup_zip.write(foldername)

  

        # 把当前文件夹里的所有文件加进压缩包

        for filename in filenames:

            # 安全防范：防止脚本把刚刚生成的 ZIP 包自己也打包进去（无限套娃）

            if filename.startswith(f'{folder_name}_backup_') and filename.endswith('.zip'):

                continue

            # 获取文件的绝对路径并写入压缩包

            file_path = os.path.join(foldername, filename)

            backup_zip.write(file_path)

  

    # 5. 收工关门

    backup_zip.close()

    print(f'报告老板，备份完成！已成功生成：{zip_filename}')

  

# ================= 运行区 =================

if __name__ == '__main__':

    # if len(sys.argv) < 2:

    #     print("用法错误！请在命令后面加上你要备份的文件夹路径。")

    #     print("例如： python backup.py /var/log/nginx/error.log")

  

    # target_folder = sys.argv[1]

  
  
  
  

    # 假设我们要备份有/root/aaa这个路径文件夹

    target_folder = '/root/aaa'

    if os.path.exists(target_folder):

        backupToZip(target_folder)

    else:

        print(f"找不到文件夹: {target_folder}，请检查路径！")

python3 backup.py /var/log/nginx/error.log