Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import person
- def logIn(friends_list):
- user = " "
- x = False
- while x == False:
- user_input = input("Enter username: \n")
- user_passw = input("Password: \n")
- for i in range(len(friends_list)):
- if user_input == friends_list[i].getUser() and user_passw == friends_list[i].getPassword():
- print("Username Found!\nWelcome to the CS network!")
- user = friends_list[i]
- x = True
- break
- elif i == len(friends_list) - 1:
- print("User not found! Please Try again.\n")
- return user
- def menu():
- print("1. Print all my friends")
- print("2. Print all my messages/status updates")
- print("3. Post a message/status update")
- print("4. Print all my friends' messages/status updates")
- print("5. Add a friend")
- print("6. Logout (Change user)")
- print("7. Exit")
- def printFriends(user):
- my_friends = user.getFriendList()
- print("Friends:", len(my_friends))
- for i in range(len(my_friends)):
- print(my_friends[i])
- print("\n")
- def PrintMessage(user):
- my_messages = user.getStatUpdate()
- for i in range(len(my_messages)):
- print(my_messages[i])
- print("\n")
- def postMessage(user):
- message = input("What is on your mind?\n")
- user.setStatUpdate(message)
- def printFMessage(user, friends_list):
- my_friends = user.getFriendList()
- for i in my_friends:
- for x in friends_list:
- if i == x.getUser():
- print(i,"'s messages:")
- other_user = x.getStatUpdate()
- for y in range(len(other_user)):
- print(other_user[y], "\n")
- def addFriend(user):
- new_friend = input("Who would you like to add?\n")
- user.setFriendList(new_friend)
- def Exit(user):
- def main():
- #read in data file; network.txt
- #pass the list to the readIn fucntion & open the file and read it and return the list.
- friends_list = []
- file = open("network.txt", "r")
- myfile = file.readlines()
- for line in myfile:
- line = line.split(", ")
- line[-1] = line[-1].strip(",\n")
- messages = []
- friends = []
- username = line[0]
- password = line[1]
- count = 3
- while line[count] != "friends":
- messages.append(line[count])
- count += 1
- count += 1
- while count < len(line):
- friends.append(line[count])
- count += 1
- user = person.Person(username, password, messages, friends)
- friends_list.append(user)
- #Log in
- user = logIn(friends_list)
- #Diplay Menu
- while True:
- menu()
- #Input Option
- option = int(input("Please enter a choice: \n"))
- #Carry out option selected
- if option == 1:
- printFriends(user)
- elif option == 2:
- PrintMessage(user)
- elif option == 3:
- postMessage(user)
- elif option == 4:
- printFMessage(user, friends_list)
- elif option == 5:
- addFriend(user, friends_list)
- elif option == 6:
- user = " "
- user = logIn(friends_list)
- elif option == 7:
- new_file = open("network.txt", 'w')
- for i in range(len(friends_list)):
- user = []
- user.append(friends_list[i].getUser())
- user.append(friends_list[i].getPassword())
- user.append(friends_list[i].getStatUpdate())
- user.append(friends_list[i].getFriendList())
- user = ', '.join(user)
- new_file.write(user + '\n')
- else:
- print("That is not an option please try again!")
- break
- #If option was to exit the program, write data out to the file network.txt
- main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement