# coding=gbk

import os
import time
import sys
import zipfile

# 1. 需備份的檔案與目錄將被指定在一個列表中
a=sys.argv[1]
sources = ['F:\\source']
sources.append(a)
# 此處需注意：必須在字串中使用雙引號用以括起其中包含空格的名稱。
print(sources)

# 2. 備份檔案必須儲存在一個主備份目錄中
target_dir = 'G:\\Backup'

# 3.備份檔案將打包壓縮成zip
# 4.zip的檔名由當前日期與時間構成
today = target_dir + os.sep + time.strftime('%Y%m%d')
now = time.strftime('%H%M%S')

# 新增一條來自使用者的註釋以建立zip的檔名
comment = 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(target_dir):
    os.mkdir(target_dir) #建立目錄

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


# 執行備份
print('Running:')
z = zipfile.ZipFile(target, 'w', zipfile.ZIP_DEFLATED)
for source in sources:
    for dirpath, dirnames, filenames in os.walk(source): 
        for filename in filenames:
            directory = os.path.join(dirpath,filename)
            z.write(directory)
names=z.namelist()
print(names)
z.close()