#!/bin/env python2.2

from os import getcwd, listdir, chdir, rename
from os.path import getmtime, isdir
from time import localtime, strftime
from string import split
from glob import glob


def chname(apath):  
    ntlist = []  
    mlist = glob('*.mpg')
    jlist = glob('*.jpg')    
    tlist = listdir(apath)
    for item in tlist:            #look for subdirectories
        if isdir(item):
            ntlist.append(item)
	    
    for file in mlist:            #rename *.mpg files
        tlm = getmtime(file) 
	print tlm
        tlml = localtime(tlm) 
        print tlml
        tlmbl = strftime('%m%d%y_%H%M%S', tlml)
        mname = tlmbl + '.mpg'
        print 'Renaming ', file, 'to ', mname
	rename(file, mname)

    for file in jlist:	          #rename *.jpg files
        tlm = getmtime(file)	 
        tlml = localtime(tlm)	
        tlmbl = strftime('%m%d%y_%H%M%S', tlml)		    
	jname = tlmbl + '.jpg'
        print 'Renaming ', file, 'to ', jname
	rename(file, jname)
	
    print getcwd()
    print mlist
    print jlist
    print ntlist
    return ntlist
	

root = getcwd()
list1 = chname(root)
num = len(list1)
list2 = []
list2.append(list1)

for h in range(0,4):     #NOTE 1
    list1 = list2[h]
    for i in range(len(list1)):
        tdir = root + '/' + list1[i]
        chdir(tdir)		
        tlist = chname(tdir)	
        for j in range(len(tlist)):
            tlist[j] = list1[i] + '/' + tlist[j]
        list2.append(tlist)	
        chdir(root)

import os
import time

DATE_FORMAT   = '%m%d%y_%H%M%S'
NEED_RENAMING = ('jpg', 'mpg')

def needsRenaming(file_name):
    """ Return True or False depending on whether <file_name> needs renaming """

    # We only want to rename the files right?
    if os.path.isdir(file_name):
        return False

    neat_extension = file_name.lower().split('.')[-1]
    # Returns True if our file extension is in the list. False otherwise
    return neat_extension in NEED_RENAMING


def renameFile(file_path):
    """ Rename <file_name> with the modified date """

    path, file_name = os.path.split(file_path)
    file_extension  = file_name.split('.')[-1]

    modified_time = time.localtime(os.path.getmtime(file_path))

    new_filename = time.strftime(DATE_FORMAT, modified_time) + '.' + file_extension

    print 'Renaming:', file_name, 'to', new_filename

    new_filename = os.path.join(path, new_filename)

    os.rename(file_path, new_filename)



def walkerFunction(arg, path, file_list):
    """
        The function called for each directory visited.
        We'll rename all the .jpg and .mpg files we find
    """

    print 'Folder:', path

    for file_name in file_list:

        # We need the full path to locate the file
        file_path = os.path.join(path, file_name)

        if needsRenaming(file_path):
            renameFile(file_path)


if __name__ == '__main__':

    import sys

    # Call this script as;
    # rename.py <path to rename>
    # eg.
    # rename.py 'd:/test'
    if len(sys.argv) > 1:
        os.path.walk(sys.argv[1], walkerFunction, None)
    else:
        print "No path to rename specified."
        sys.exit(1)