Creates directory with filetypes for all files in specified directory

filetype_sorter.py

import shutil
import os
from os.path import join
import time
import sys
"""
USAGE
python filetype_sorter.py <target directory>
"""
directory = sys.argv[1]
files_in_dir = [(f,f.split(".")[-1]) for f in os.listdir(directory) if os.path.isfile(join(directory,f))]
for fi, di in files_in_dir:
    if not os.path.isdir(join(directory, di)):
        os.mkdir(join(directory ,di))
    time.sleep(.05)
    print("Moving %s to %s" % (join(directory,fi),join(directory,di,fi)))
    shutil.move(join(directory,fi), join(directory, di,fi))