/Python/Python split files into even sets of folders

Python split files into even sets of folders

import os       # Used to do path manipulations
import shutil   # Used to copy files

# Taken from https://stackoverflow.com/questions/312443/how-do-you-split-a-list-into-evenly-sized-chunks
def chunks(lst:list, n:int) -> list:
    """Yield successive n-sized chunks from lst."""
    for i in range(0, len(lst), n):
        yield lst[i:i + n]

def get_abspath_files_in_directory(directory:str) -> list:
    """Takes in a directory and returns the abspath of all the files in the directory as a list

    Parameters
    ----------
    directory : str
        The directory to get the abspaths for

    Returns
    -------
    list
        A list of the abspaths of all the files in the directory
    """
    return [os.path.abspath(os.path.join(directory,path)) for path in  os.listdir(directory)]


def split_to_subdirectories(file_paths:list, amount_per_folder:int):
    """Take in a list of file absolute paths, and copy them to folders

    Parameters
    ----------
    file_paths : list
        The list of abspaths to the file folders

    amount_per_folder : int
        The amount of files per folder to split the files into
    """
    file_paths = chunks(file_paths, amount_per_folder)

    for index, chunk in enumerate(file_paths):
        os.mkdir(str(index)) # Create a folder with a name of the current iteration index
        for file_path in chunk:
            file_name = file_path.split(os.sep)[-1]
            shutil.copy(file_path, os.path.join(str(index), file_name))

if __name__ == "__main__":
    file_paths = get_abspath_files_in_directory("original_folder") # Replace "original_folder" with the directory where your files are stored
    split_to_subdirectories(file_paths, 20)

Source: stackoverflow.com

Similar codes

0 vote292 views1 answer
Find highest correlation pairs pandas
Python
0 vote315 views1 answer
In deserialize_keras_object raise ValueError('Unknown ' + printable_module_name + ':' + object_name) ValueError: Unknown loss function:contrastive_loss
Python
9 votes299 views4 answers
Python run server
Python
0 vote298 views1 answer
Drawing arrows in tkinter
Python
6 votes307 views1 answer
Django desc order
Python
0 vote307 views1 answer
Python sleeping with a varible
Python
0 vote284 views2 answers
Numpy method to make polynomial model
Python
0 vote356 views2 answers
Django serializer method field read write
Python Django
0 vote297 views1 answer
Add favicon in django admin
Python
0 vote322 views1 answer
Python download complete web page
Python