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 /home kataloogi. Uue faili tüübi lisamisel tuleb täiendada kataloogide nimekirja ja i väärtust. Skripti kävitamisel kuvatakse ükshaaval failitüüp kujul "*.mp3" ning iga faili tüübi alla failinimed,mida kopeeriti.

#!/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/"]
# i´le antakse järjest numbrilisi väärtusi 
for i in range (0, 6):
    # kontroll kas kaust on loodud
    if os.path.isdir( kaustA[i] ):
        # Väljasta kopeeritav failitüüp 
        print failA[i]
        for f in  glob.glob(failA[i]):
            # Näita kopeeritava faili nime    
            print f
            # Faili uue asukoha ja nime liitmine
            dir = kaustA[i] + f
            # Kopeerimine f-failinimi, dir-kataloog kuhu kopeeritakse
            shutil.copy(f, dir)
    #kui kausta veel ei ole siis luuakse  see
    else:
        loo = "mkdir " + kaustA[i]
        os.system(loo)