备份脚本二

#!/usr/bin/env python

import tarfile
import os
import re
import sys
import time

pattern = r'(.*)\.tar\.gz'
suffix = '.tar.gz'
ls = os.sep

def tar_dir(stpath,back_name,back_dir,excludedir):
    tar = tarfile.open('%s%s' % (stpath,back_name), 'w:gz')
    for name in os.listdir('%s' % back_dir):
        os.chdir('%s' % back_dir)
        if name not in excludedir:
            tar.add(name)
    tar.close()

while True:
    backupdir = raw_input('1. Enter a directroy which \
you want to back up: ').strip()
    if len(backupdir) == 0:
        print 'Input nothing, try again'
	    continue
    elif os.path.isfile(backupdir):
	     print backupdir + ' is not a directory,try again'
	     continue
    elif not os.path.isabs(backupdir):
	     print 'Please enter an absolute path, try again'
	     continue
    elif not os.path.exists(backupdir):
	     print 'There is no directory called ' + backupdir
	     continue
    elif backupdir[-1] != '/':
	    backupdir = backupdir + ls
    break

while True:
    dest_path = raw_input('2. Enter a storage path: ').strip()
    if len(dest_path) == 0:
        print 'Input nothing, try again'
	    continue
    if not os.path.exists(dest_path):
	    os.makedirs(dest_path)
    if not os.path.isabs(dest_path):
	    print 'Please enter an absolute path, try again'
	    continue
    if dest_path[-1] != '/':
	    dest_path = dest_path + ls
    break


while True:
    backupname = raw_input('3. Take a name for the backup: ').strip()
    if len(backupname) == 0:
	    print 'Input nothing, try again'
	    continue
    elif '/' in backupname:
	    print 'Please enter a name without "/".'
	    continue
    break

if 'tar.gz' in backupname:
    m = re.search(pattern,backupname)
    prefix = m.group(1).replace('.', '_').replace('-', '_')
    backupname = prefix + suffix
else:
    prefix = backupname.replace('.', '_').replace('-', '_')
    backupname = prefix + suffix

while True:
    if_continue = 0
    prompt = '''
4. Enter directorys or files which you want to exclude,if there
is no directorys or files need to be excluded, press enter directly!

(Note: absolute path is not needed)

For example:dir1,dir2 or dir1 dir2

Enter:'''

    excludes = raw_input(prompt).strip()
    if '/' in excludes:
        print '\nWarning: absolute path is not needed\n'
	    continue
    excludes = excludes.replace(' ', ',').split(',')
    for i in excludes:
        if not os.path.exists(backupdir+i):
	        print '\nWarning: there is no file or directory called ' + (backupdir + i) + '\n'
	        if_continue = 1
	        break
    if if_continue:
        continue
    else:
	    break

backabs = dest_path + backupname     
if os.path.exists(backabs):
    while True:
        anwser = raw_input('the backup %s is already exists, do you \
want to overwrite it (yes/no)? ' % backabs)
        if anwser == 'yes':
	        print 'backup start...'
	        start_time = time.time()
	        tar_dir(dest_path,backupname,backupdir,excludes)
	        print 'backup complete'
	        end_time = time.time()
	        print '#' * 60
	        print 'the backup took ' + str(round(end_time-start_time,1)) + 's.'
            print 'you can find the backup at: ' + backabs
	        print '#' * 60
	        break
        elif anwser == 'no':
	        print 'the backup for %s has been caceled' % backupdir
	        break
	        sys.exit()
	    else:
	        print 'invalid option, try again'
else:
    print 'backup start...'
    start_time = time.time()
    tar_dir(dest_path,backupname,backupdir,excludes)
    print 'backup complete'
    end_time = time.time()
    print '#' * 60
    print 'the backup took ' + str(round(end_time-start_time,1)) + 's.'
    print 'you can find the backup at: ' + backabs
    print '#' * 60