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)
