'''
Created on Oct 26, 2011

@author: rodger
'''

def processFile(file):
    '''
    The parameter file is the name of a file with one or more names
    on a line. Each name is two words separated by a blank.
    Multiple names on a line are separated by a colon
    Return a list of names
    '''
    nameList = []
    for line in file:
        listNames = line.strip().split(':')
        [nameList.append(nm.split()) for nm in listNames]
    return nameList

def processFiles(numberFiles):
    ''' Given the number of files to process, assume all have
    the name "namesNUM.txt". Process all those files
    '''
    nameList = []
    for number in range(1,numberFiles+1):
        filename = "../data/names"+  str(number) + ".txt"
        print "just checking", filename
        file = open(filename)
        moreNames = processFile(file)
        for name in moreNames:
            nameList.append(name)
        file.close()
    return nameList   
        
        
        
        
    return nameList

def uniqueFirstNames(data):
    '''
    The parameter data is a list of lists where each list is two 
    strings representing the first and last name of a person. 
    Return a list of the unique first names of all the people.
    '''
    uniqueNames = set([])
    #TODO:
    for item in data:
        uniqueNames.add(item[0])
    return list(uniqueNames)
        
    
    
    
    
    
    
    
def uniqueLastNames(data, firstName):
    '''
    The parameter data is a list of lists where each list is two  
    strings representing the first and last name of a person. 
    The string parameter firstname is the first name of a person 
    (one word).
    Return a list of the last names of all the people from 
    data with this 
    first name. Assume each person in data is unique.
    '''
    return [item[1]  for item in data  if item[0]== firstName]


    
 
    
    
def correspondingLastNames(data, firstNames): 
    '''
    The parameter data is a list of lists where each list is two  
    strings representing the first and last name of a person. 
    The parameter firstNames is a list of the unique 
    first names from data.
    return a list of lists of last names such that the nth list 
    of last names corresponds to the nth first name in firstNames.
    '''
    allLastNames = []
    #TODO:
    for name in firstNames:
        lastNamesForName = uniqueLastNames(data, name)
        allLastNames.append(lastNamesForName)
    return allLastNames
    
    
    

def printFirstWithLasts(firstNames, lastNames):
    '''
    firstNames is a list of unique first names
    lastNames is a list of lists of all the last names that
    correspond to a firstName in the same slot
    '''
    print "Print Unique First Names with all Last Names"      
    #TODO:
    for (index,first) in enumerate(firstNames):
        print first +":" + " ".join(lastNames[index])
        
     
    
    
    


def main():

    #fileWithNames = open("../data/names1.txt")
    #namelist = processFile(fileWithNames)  

    # for last problem: 
    namelist = processFiles(3)
    
    print namelist 
    firstNames = uniqueFirstNames(namelist)
    print firstNames
    lastNames = correspondingLastNames(namelist, firstNames)
    print lastNames
    print "Susan", uniqueLastNames(namelist, "Susan")
    print "Eric", uniqueLastNames(namelist, "Eric")
    
    print "corresponding last names"
    print correspondingLastNames(namelist, firstNames)

    printFirstWithLasts(firstNames, lastNames)

    maxnum = max(len(item) for item in lastNames)
    print maxnum
    lastIndex = [index for (index, v) in enumerate(lastNames) if len(v) == maxnum]
    print "first name with most last names is:"
    #TODO:
    print firstNames[lastIndex[0]]
    

if __name__ == "__main__":
    main()