cluster.py

import os
import shutil
import sys

#
# For each file in a given directory, create a directory with the first letter of the file, and then move the file
# into that directory.
#

def main():
    directory = sys.argv[1]
    os.chdir(directory)
    files = os.listdir('.')
    letters = {}
    for f in files:
        if f.startswith('.'):
            continue
        if os.path.isfile(f):
            ltr = f[0].lower()
            if not ltr in letters.keys():
                letters[ltr] = []
            letters[ltr].append(f)
    for ltr in letters.keys():
        os.mkdir(ltr)
        for f in letters[ltr]:
            shutil.move(f, os.path.join(ltr, f))
    
if __name__ == "__main__":
    main()