python 练手程序 文件备份

#filename; backup_ver3.py

import os
import time

source = ['/home/book/Desktop/happy','/home/book/Desktop/new']

target_dir = '/home/book/Desktop/backup'

today = target_dir + time.strftime('%Y%m%d')

now = time.strftime('%H%M%S')

comment = raw_input('Enter a comment-->')
if len(comment) == 0:
    target = today + os.sep + now + '.zip'
else:
    target = today + os.sep + now + '_' + comment.replace(' ','_') + '.zip'

if not os.path.exists(today):
    os.mkdir(today)
    print 'Successfully created directory',today

zip_command = "zip -qr '%s' %s" % (target,' '.join(source))
if os.system(zip_command) == 0:
    print 'Successful backup to %s' % target
else:
    print 'Backup FAILED'