import tarfile
import time
import hashlib
import pickle
import os
def hash_md5(fname):
? ? m = hashlib.md5()
? ? with open(fname,'rb') as fobj:
? ? ? ? while 1:
? ? ? ? ? ? data =fobj.read(4096)
? ? ? ? ? ? if not data:
? ? ? ? ? ? ? ? break
? ? ? ? ? ? m.update(data)
? ? return m.hexdigest()
def full_backup(src,dst,md5file):
? ? '全備份'
? ? # 拼接出備份文件的文件名
? ? fname = f'{os.path.basename(src)}_full_{time.strftime("%Y-%m-%d")}.tar.gz'
? ? fname = os.path.join(dst,fname)
? ? #備份，打tar包
? ? with tarfile.open(fname, 'w:gz') as tar:
? ? ? ? tar.add(src)
? ? # 計算當(dāng)前文件的md5值
? ? md5dir = {}
? ? for path,folders,files in os.walk(src):
? ? ? ? for file in files:
? ? ? ? ? ? key = os.path.join(path,file)
? ? ? ? ? ? md5dir[key] = hash_md5(key)
? ? # 保存md5字典到md5file
? ? with open(md5file,'wb') as fobj:
? ? ? ? pickle.dump(md5dir,fobj)
def incre_backup(src,dst,md5file):
? ? '增量備份'
? ? # 拼接出備份文件的文件名
? ? fname = f'{os.path.basename(src)}_incre_{time.strftime("%Y-%m-%d")}.tar.gz'
? ? fname = os.path.join(dst, fname)
? ? # 計算當(dāng)前文件的md5值
? ? md5dir = {}
? ? for path,folders,files in os.walk(src):
? ? ? ? for file in files:
? ? ? ? ? ? key = os.path.join(path,file)
? ? ? ? ? ? md5dir[key] = hash_md5(key)
? ? # 取出前一天的md5值
? ? with open (md5file,'rb') as fobj:
? ? ? ? old_md5 = pickle.load(fobj)
? ? # 備份新文件和改動的文件
? ? with tarfile.open(fname,'w:gz') as tar:
? ? ? ? for key in md5dir:
? ? ? ? ? ? if old_md5 != md5dir[key]:
? ? ? ? ? ? ? ? tar.add(key)
? ? # 更新md5文件
? ? with open(md5file,'wb') as fobj:
? ? ? ? pickle.dump(md5dir,fobj)
if __name__ == '__main__':
? ? src = '/tmp/demo/security' #自定義
? ? dst = '/tmp/demo/backup' #自定義
? ? md5file = '/tmp/demo/backup/md5.data'#自定義
? ? #周一全備，其他增量備
? ? if (time.strftime('%a') == 'Mon') or (not os.path.exists(md5file)):
? ? ? ? full_backup(src,dst,md5file)
? ? else:
? ? ? ? incre_backup(src,dst,md5file)
