uaveiro-leci/1ano/fp/aula05/telephones.py

# Convert a telephone number into corresponding name, if on list.
# (If not on list, just return the number itself.)


def telToName(tel, telList, nameList):
    # your code here
    index = 0
    for t in telList:
        if t == tel:
            break
        index += 1
    return tel if index == len(telList) else nameList[index]


# Return list of telephone numbers corresponding to names containing partName.
def nameToTels(partName, telList, nameList):
    # your code here
    tels = []
    index = 0
    for name in nameList:
        if partName in name:
            tels.append(telList[index])
        index += 1
    return tels

def main():
    # Lists of telephone numbers and names
    telList = ['975318642', '234000111', '777888333', '911911911']
    nameList = ['Angelina', 'Brad',      'Claudia',   'Bruna']

    # Test telToName:
    tel = input("Tel number? ")
    print( telToName(tel, telList, nameList) )
    print( telToName('234000111', telList, nameList) == "Brad" )
    print( telToName('222333444', telList, nameList) == "222333444" )

    # Test nameToTels:
    name = input("Name? ")
    print( nameToTels(name, telList, nameList) )
    print( nameToTels('Clau', telList, nameList) == ['777888333'] )
    print( nameToTels('Br', telList, nameList) == ['234000111', '911911911'] )


if __name__ == "__main__":
    main()