'''
Created on Oct 31, 2016

@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(':')
        print "listnames is", listNames
        [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 = "names"+str(number)+".txt"
        print "filename is", filename
        file = open(filename)
        nameList += processFile(file)     
        file.close()
       
    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 name in data: # name is a list of two strings
        uniqueNames.add(name[0])
    return list(uniqueNames)
    


    
    
    
    
    
    
def allLastNames(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. 
    '''
    lastNames = []
    for name in data:  # name is a list of two strings
        (first, last) = name
        if first == firstName:
            lastNames.append(last)
        
    return lastNames  #TODO; Replace me


    

 
    
    
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.
    '''
    lastNames = []
    for name in firstNames:
        lastNames.append(allLastNames(data,name))
         
    return lastNames
    
def printFirstWithLasts(firstNames, lastNames):
    print "Print Unique First Names with all Last Names"      
    #TODO:
    for (index,name) in enumerate(firstNames):
        print name+": ", " ".join(lastNames[index])

    
    
    


def main():

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

    # for last problem: namelist = processFiles(3)
    namelist = processFiles(3)
    print namelist 
    firstNames = uniqueFirstNames(namelist)
    print firstNames
    lastNames = correspondingLastNames(namelist, firstNames)
    print lastNames
    printFirstWithLasts(firstNames, lastNames)
    maxnum = max([len(item) for item in lastNames])
    print "max number of last names is", maxnum
    lastIndex = [index for (index, v) in enumerate(lastNames) if len(v) == maxnum][0]
    print "first name with most last names is:", firstNames[lastIndex]
    #TODO:
    print 
    

if __name__ == "__main__":
    main()