/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

91 votes193 views2 answers
Python 3 download
Python
35 votes213 views2 answers
How to run cmd line commands in python
Python
0 vote207 views1 answer
Python background process
Python
0 vote195 views1 answer
NSDictionary fast access
Python
0 vote212 views1 answer
Set shortcut for Qaction pyqt5
Python
0 vote242 views1 answer
Get category discord.py
Python
2 votes267 views1 answer
Convert int to byte python
Python
142 votes214 views6 answers
While loop python
Python
27 votes208 views3 answers
How to convert exe file to python file
Python
10 votes205 views6 answers
Merge two lists python
Python