sort.py

"""
Copyright 2023 nikallow

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. 
"""

import os
import shutil
from threading import Thread


def sort_files(folder_location):
    def sorting(folder_name, file_types):
        for file_name in os.listdir(folder_location):

            # Check if there are files with the received extensions
            if file_name.endswith(file_types): 

                # Creating a Sorted folder if it doesn't exist
                sorted_folder = os.path.join(folder_location, "Sorted")
                os.makedirs(sorted_folder, exist_ok=True)   

                # Creating a folder for a category if it does not exist
                folder = os.path.join(sorted_folder, folder_name)
                os.makedirs(folder, exist_ok=True)

                # Get the base name and extension of the file
                base_name, extension = os.path.splitext(file_name)

                # Check if the file already exists in the destination folder
                new_file_name = file_name
                suffix = 0
                while os.path.exists(os.path.join(folder, new_file_name)):
                    suffix += 1
                    new_file_name = f"{base_name} ({suffix}){extension}"

                # Moving files to the folder with the updated file name
                file_path = os.path.join(folder_location, file_name)
                new_file_path = os.path.join(folder, new_file_name)
                shutil.move(file_path, new_file_path)

    images_types = (".png", ".jpg", ".bmp", ".gif", ".tif", ".kra", ".svg",
                    ".raw", ".tiff", ".psd", ".bmp", ".jpeg", ".webp")

    documents_types = (".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx",
                       ".pdf", ".djvu", ".fb2", ".epub", ".mobi", ".txt",
                       ".odt", ".ods", ".odp", ".odg", ".odf", ".odb")

    archives_types = (".zip", ".7z", ".rar", ".tar", ".gz", ".tar.gz")

    videos_types = (".mp4", ".mkv", ".mov", ".hevc")

    music_types = (".mp3", ".ogg", ".wav", ".aiff", ".ape", ".flac")

    Images = Thread(target=sorting("Images", images_types))
    Images.start()
    
    Documents = Thread(target=sorting("Documents", documents_types))
    Documents.start()

    Archives = Thread(target=sorting("Archives", archives_types))
    Archives.start()

    Videos = Thread(target=sorting("Videos", videos_types))
    Videos.start()

    Music = Thread(target=sorting("Music", music_types))
    Music.start()