main.py

import shutil
import sys
import scan
import normalize
from pathlib import Path
#from files_generator import file_generator



def hande_file(path, root_folder, dist):
    target_folder = root_folder / dist
    target_folder.mkdir(exist_ok=True)
    path.replace(target_folder/normalize.normalize(path.name))


def handle_archive(path, root_folder, dist):
    target_folder = root_folder / dist
    target_folder.mkdir(exist_ok=True)

    new_name = normalize.normalize(path.name.replace(".zip", '').replace(".tar", '').replace(".gz", ''))

    archive_folder = target_folder / new_name
    archive_folder.mkdir(exist_ok=True)

    # def _normalize_archive(folder_name):
        
    #     for item in folder_name.iterdir():
    #         if item.is_dir():
    #             _normalize_archive(item)
    #         else:
    #             normalize.normalize(item)

    try:
        shutil.unpack_archive(path, archive_folder)

       # _normalize_archive(archive_folder)

    except shutil.ReadError:
        archive_folder.rmdir()
        return
    except FileNotFoundError:
        archive_folder.rmdir()
        return
    path.unlink()


def remove_empty_folders(path):
    for item in path.iterdir():
        if item.is_dir():
            remove_empty_folders(item)
            try:
                item.rmdir()
            except OSError:
                pass


def get_folder_objects(root_path):
    for folder in root_path.iterdir():
        if folder.is_dir():
            remove_empty_folders(folder)
            try:
                folder.rmdir()
            except OSError:
                pass

def main(folder_path):
    scan.scan(folder_path)



    for file in scan.images:
        hande_file(file, folder_path, "IMAGES")

    for file in scan.documents:
        hande_file(file, folder_path, "DOCUMENTS")

    for file in scan.audio:
        hande_file(file, folder_path, "AUDIO")

    for file in scan.video:
        hande_file(file, folder_path, "VIDEO")

    for file in scan.others:
        hande_file(file, folder_path, "OTHERS")

    for file in scan.archives:
        handle_archive(file, folder_path, "ARCHIVE")



    get_folder_objects(folder_path)

if __name__ == '__main__':
    path = sys.argv[1]
    print(f"Start in {path}")

    arg = Path(path)
    #file_generator(arg)
    main(arg.resolve())