Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class User:
- def __init__(self, name):
- self.name = name
- self.messages = []
- class Message:
- def __init__(self, sender, content):
- self.sender = sender
- self.content = content
- def filter_messages(sender, messages):
- messages_to_return = []
- for current_message in messages:
- if message.sender == sender:
- messages_to_return.append(current_message)
- return messages_to_return
- all_users = {}
- while True:
- data = input().split()
- if data[0] == 'exit':
- break
- if len(data) == 2:
- all_users[data[1]] = User(data[1])
- else:
- if data[0] in all_users.keys() and data[2] in all_users.keys():
- all_users[data[2]].messages.append(Message(data[0], data[3]))
- user_names = input().split()
- user_1 = all_users[user_names[0]]
- user_1.messages = filter_messages(user_names[1], user_1.messages)
- user_2 = all_users[user_names[1]]
- user_2.messages = filter_messages(user_names[0], user_2.messages)
- if len(user_1.messages) == 0 and len(user_2.messages) == 0:
- print("No messages")
- else:
- while len(user_1.messages) != 0 or len(user_2.messages) != 0:
- if len(user_2.messages) != 0:
- message = user_2.messages[0]
- print(user_1.name + ": " + message.content)
- user_2.messages.remove(message)
- if len(user_1.messages) != 0:
- message = user_1.messages[0]
- print(message.content + " :" + user_2.name)
- user_1.messages.remove(message)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement