def print_user_messages(cursor, user): messages = Messages.load_all_messages(cursor, user.id) for message in messages: from_ = User.load_user_by_id(cursor, message.from_id) print(20 * "-") print(f"from: {from_.username}") print(f"data: {message.creation_date}") print(message.text) print(20 * "-")
def list_messages(cursor, username, password): user = User.load_user_by_name(cursor, username) if not user: print("User does not exist.") elif check_password(password, user.hashed_password): messages = Messages.load_all_messages(cursor, user.id) for message in messages: sender = User.load_user_by_id(cursor, message.from_id) print( f"From: {sender.username}\n{message.text}\nDate: {message.creation_date}\n-----------------------------------------------------------------------------" ) else: print("Incorrect password.")
while True: try: option = int( input( 'What you want to do?\n1: See your messages\n2: Send message\nYour choice: ' )) except: print('Try again') continue if option == 1: user = login() if user: a = Messages.load_all_messages(to_id=user.id) for item in a: user_from = Users.load_user_by_id(item.from_id) print( f'\nMessage from {user_from.username}:\n{item.msg}\nSent {item.creation_date}\n' ) if len(a) == 0: print('No messages for you') elif option == 2: user = login() if user: receiver = input('To whom you want to send your message') receiver_user = Users.load_user_by_username(receiver) msg = input('Write your message: ') try: new_msg = Messages(user.id, receiver_user.id, msg)