ITS·黑客Linux技巧

《Byte of Python3》 11章 建立备份的程序

2017-05-22  本文已影响9人  某米狼

“我想写一个能给我所有重要文件建立备份的程序”。

  1. 需要备份的文件和目录由一个列表指定。
  2. 备份应该保存在主备份目录中。
  3. 文件备份成一个 zip 文件。
  4. zip 存档的名称是当前的日期和时间。
  5. 我们使用标准的 zip 命令,它通常默认地随 Linux/Unix 发行版提供。Windows 用户可以从GnuWin32项目页安装。注意你可以使用任何存档命令,只要它有命 令行界面就可以了,那样的话我们可以从我们的脚本中传递参数给它。

版本一

# !/usr/bin/env python3
# -*- coding: utf-8 -*-
# Filename: backup_ver1.py

import os
import time

# 1.待备份的文件作为一个list
# 在windows下:
# source = ['"C:\\My Documents"','C:\\Code'] #这里要转义
# 在 Mac OS X 和 Linux 下:
source = ['/Users/Lazyant/Documents/学习/Pythonlearn/练习题']

# 2. 目标存档目录,记得把路径改为自己电脑的路径
target_dir = '/Users/Lazyant/Documents/学习/Pythonlearn/backup'

# 3. 文件备份到一个zip文件里
# 以当前日期命名zip文件
# time.strftime()获取时间
# os.sep 会根据操作系统给出路径分隔符,如在 windows 中用'\\'
target = target_dir + os.sep + time.strftime('%Y%m%d%H%M%S') + '.zip'

# 4. 如果文件夹不存在则创建文件夹
if not os.path.exists(target_dir):
    os.mkdir(target_dir)

# 5. 创建zip_command字符串,其包含将要执行的zip命令
# 使用的 zip 命令有一些选项和传递的参数。-q 选项被用来表明 zip 命令应该以快 速的方式进行。-r 选项指定 zip 命令应该对每个目录重复执行,例如应该包含所有子 目录和文件。这两个选项可以用 -qr 结合到一起。
# 用 join 方法将资源列表转换成一个字符串
zip_command = "zip -r {0} {1}".format(target,' '.join(source))

# 用 os.system 函数运行命令
print('Zip command is:')
print(zip_command)
print('Running:')
if os.system(zip_command) == 0:
    print('Successful backup to',target)
else:
    print('Backup FAILED')

版本二
优化:使用时间作为文件名,而当前的 日期作为目录名,存放在主备份目录中。

...

# 3. 文件备份到一个zip文件里
# 以当前日期在主目录里命名子目录
today = target_dir + os.sep + time.strftime('%Y%m%d')
# 以当前时间命名zip
now = time.strftime('%H%M%S')
target = today + os.sep + now + '.zip'

# 4. 如果文件夹不存在则创建文件夹
if not os.path.exists(target_dir):
    os.mkdir(target_dir)
if not os.path.exists(today):
    os.mkdir(today)
    print('Successful created directory',today)
...
...

版本三:
优化:文件名添加备注

...
# 3. 文件备份到一个zip文件里
# 以当前日期在主目录里命名子目录
today = target_dir + os.sep + time.strftime('%Y%m%d')
# 以当前时间命名zip
now = time.strftime('%H%M%S')

# 加个备注
comment = input('Enter a comment -->')
#检查输入的备注
if len(comment) == 0:
    target = today + os.sep + now + '.zip'
else:
    target = today + os.sep + now + \
     '_' + comment.replace(' ','_') + '.zip'
...

对于大多数用户来说,第四个版本是一个满意的工作脚本了,但是它仍然有进一 步改进的空间。比如,你可以在程序中包含交互程度 —— 你可以用 -v 选项来使你的 程序更具交互性。
另一个可能的改进是使文件和目录能够通过命令行直接传递给脚本。我们可以通 过 sys.argv 列表来获取它们,然后我们可以使用 list 类提供的 extend 方法把它们加到 source 列表中去。
最重要的提炼就是不用 os.system 方式来创建压缩文档,而是用 zipfile 或 tarfile 内置模块来创建压缩文档。它们是标准库的一部分,没有与你计算机中外部 zip 程序 的依赖性,并且已经能够使用。
但是,在上面的例子中,我已经使用了 os.system 的方式创建备份,这纯粹是为 教学的方便,以使得例子足够简单,能被每个人理解,但这已经足够用了。
你能尝试用zipfile 模块来替代 os.system 调用写出第四版吗?

上一篇下一篇

猜你喜欢

热点阅读