clean_dirs

import os
import argparse


'''
Этот код содержит функцию для очистки директорий, удаляя либо печатая на экране имена файлов, 
которые уже присутствуют в эталонной директории. Эта функция может быть использована 
с опцией -V для проверки размера файлов и удаления только тех, которые равны по размеру. 
Функция запускается через командную строку с параметрами etalon_dir и clear_dir. 
Дополнительно можно использовать опции -R для удаления файлов и -V для проверки их размеров.
'''


def list_dirs(directory):
    file_list = []
    if not os.path.exists(directory):
        return file_list

    for root, sub_folders, files in os.walk(directory):
        for file in files:
            filename = os.path.join(root, file)
            file_list.append(filename)
    return list(set(file_list))


def verify_files(f1, f2, verify):
    if not os.path.exists(f1) or not os.path.exists(f2):
        return False
    if not verify:
        return True
    if not (os.path.getsize(f1) == os.path.getsize(f2)):
        return False
    return True


def clean_dirs(*, save_dir, toclean_dir, remove=False, verify=False):
    save_filelist = list_dirs(save_dir)
    for file in save_filelist:
        clean_filename = file.replace(save_dir, toclean_dir)
        if verify_files(file, clean_filename, verify):
            if remove == False:
                print(f'try to rm {clean_filename}')
            else:
                os.remove(clean_filename)


class DirAction(argparse.Action):
    def __init__(self, option_strings, dest, nargs=None, **kwargs):
        if nargs is not None:
            raise ValueError("nargs not allowed")
        super().__init__(option_strings, dest, **kwargs)
    def __call__(self, parser, namespace, values, option_string=None):
        values = os.path.realpath(values)
        if os.path.exists(values):
            setattr(namespace, self.dest, values)
        else:
            raise Exception(f'path: {values} not exists')


if __name__ == "__main__":
    from pprint import pprint
    parser = argparse.ArgumentParser(description="Help", formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument("-e", "--etalon_dir", type=str, action=DirAction, default="/tmp")
    parser.add_argument("-c", "--clear_dir", type=str, action=DirAction, default="/tmp")
    parser.add_argument("-R", "--remove", action="count", default=0)
    parser.add_argument("-V", "--verify", action="count", default=0)
    config = parser.parse_args()
    pprint(vars(config))
    clean_dirs(save_dir=os.path.realpath(config.etalon_dir), toclean_dir=config.clear_dir, remove=config.remove, verify=config.verify)