"""
         DirDeleter.py v0.2
         by Alessio Saltarin

         Usage: python dirdeleter.py [initial_directory] [string]
        
         Walks over all sub-directories starting from
         [initial_directory] and deletes every directory that has
         its name equal to [string] or containing [string]
        

        
"""

import os, sys, string
from stat import *

def walktree(dir, initdir):
    for f in os.listdir(dir):
        pathname = '%s\%s' % (dir, f)
        mode = os.stat(pathname)[ST_MODE]
        if S_ISDIR(mode):
            walktree(pathname, initdir)
            deletedir(pathname, initdir)
        elif S_ISREG(mode):
            pass
        else:
            print 'Attention: unknown %s' % pathname

def deletedir(nome, init):
    if (string.rfind(nome,init)>0):
        prunefiles(nome)
        os.rmdir(nome)
        print 'deleted: %s' % nome
    else:
        print 'visited: %s' % nome

def prunefiles(nome):
    for f in os.listdir(nome):
           path = '%s\%s' % (nome,f)
           os.remove(path)
           print ' --- deleted '+path
 
def printusage():
    print """

Usage: python dirdeleter.py initial_dir string

where
    initial_dir:
            Initial directory to search for subdirectories to delete.
    string:
            The subdirectory containing 'string' in its name, or with a
            name equals to 'string' will be deleted.

"""   
   
if __name__ == '__main__':
    if (len(sys.argv) == 3):
        walktree(sys.argv[1], sys.argv[2])
    else:
        printusage()