archive

import os
import time
import zipfile

# Откуда берем файлы

source = input('Введите расположение файла: ')

# Куда будем сохранять

input_dir = input('Введите директорию, куда будет сохранен архив: ')

# Если директории нет, то создаем ее

if not os.path.exists(input_dir):
    os.mkdir(input_dir)
    print('Каталог ', input_dir, ' успешно создан')

# Запрашиваем имя файла

comment = input('Введите имя файла: ')
if len(comment) == 0: # проверяем, введён ли комментарий
    name = time.strftime('photos %d.%m.%Y %H.%M.%S') + '.zip'
    target_dir = input_dir + name
else:
    name = comment + ' ' + time.strftime('%d.%m.%Y %H.%M.%S') + '.zip'
    target_dir = input_dir + name


# Архивируем файлики через костыли

target = zipfile.ZipFile(target_dir, 'w')

for folder, subfolders, files in os.walk(source):

    for file in files:
        if file.endswith('.jpg'):
            target.write(os.path.join(folder, file), os.path.relpath(os.path.join(folder,file), source), compress_type = zipfile.ZIP_DEFLATED)

target.close()

# Проверочка, сохранился ли архив

for folder, subfolders, files in os.walk(input_dir):
    for file in files:
        if file == name:
            x=1
            break
        else:
            x=0
if x==1:
    print('Архив успешно создан в', input_dir)
else:
    print("Не архивировано")