from friend import Friend
from utils import doesFriendExist
allFriends = []
inputFile = open('./friends.txt', 'r')
allLines = inputFile.readlines()
for line in allLines:
names = line.split(' ')
firstFriend = Friend(names[0])
secondFriend = Friend(names[1])
firstFriend.addFriend(secondFriend)
secondFriend.addFriend(firstFriend)
if doesFriendExist(allFriends, firstFriend) == False:
allFriends.append(firstFriend)
if doesFriendExist(allFriends, secondFriend) == False:
allFriends.append(secondFriend)
# from human import Human
# ageSum = 0
# humans = []
# peopleFile = open('./people.txt', 'r')
# allLines = peopleFile.readlines()
# for line in allLines:
# humans.append(Human(line))
allLines = inputFile.read().splitlines()
for line in allLines:
names = line.split(' ')
firstFriend = Friend(names[0])
secondFriend = Friend(names[1])
firstFriendIndex = next((i for i, friend in enumerate(allFriends) if friend.name == firstFriend.name), -1)
secondFriendIndex = next((i for i, friend in enumerate(allFriends) if friend.name == secondFriend.name), -1)
if firstFriendIndex > -1:
if secondFriendIndex > -1:
allFriends[firstFriendIndex].addFriend(allFriends[secondFriendIndex])
allFriends[secondFriendIndex].addFriend(allFriends[firstFriendIndex])
else:
secondFriend.addFriend(allFriends[firstFriendIndex])
allFriends[firstFriendIndex].addFriend(secondFriend)
allFriends.append(secondFriend)
else:
if secondFriendIndex > -1:
firstFriend.addFriend(allFriends[secondFriendIndex])
allFriends[secondFriendIndex].addFriend(firstFriend)
allFriends.append(firstFriend)
else:
firstFriend.addFriend(secondFriend)
secondFriend.addFriend(firstFriend)
allFriends.append(firstFriend)
allFriends.append(secondFriend)
if firstFriendIndex > -1:
count = getFriendsCountInFriendNetwork([allFriends[firstFriendIndex]])
