Skript mis otsib kõik pildi,video,muusika,dokumendi failid ja jagab need kaustadesse - Python

Skript mis otsib failid faili tüübi järgi ja kopeerib failitüübile määratud kausta. Uue faili tüübi lisamisel tuleb täiendada kataloogide nimekirja ja i väärtust.

#!/usr/bin/python
# -*- coding: utf-8 -*-


import shutil
import glob
import os

# Failitüübid
failA = ["*.mp3", "*.wav", "*.doc", "*.pdf", "*.avi", "*.jpg"]
# Kaustad kuhu failid failitüüdi järgi kopeeritakse
kaustA = ["/home/Muusika/", "/home/Muusika/", "/home/Dokument/", "/home/Dokument/", "/home/Video/", "/home/Pilt/"]
for i in range (0, 6):
    if os.path.isdir( kaustA[i] ):
        print failA[i]
        for f in  glob.glob(failA[i]):
            print f
            dir = kaustA[i] + f
            shutil.copy(f, dir)
    else:
        loo = "mkdir " + kaustA[i]
        os.system(loo)