def send_message(cursor, from_id, recipient_name, text):
if len(text) > 255:
print("Message is too long!")
return
to = User.load_user_by_username(cursor, recipient_name)
if to:
message = Messages(from_id, to.id, text=text)
message.save_to_db(cursor)
print("Message send")
else:
print("Recipient does not exist.")
def send_a_massage(username, password, to_id, text):
if User.load_user_by_username(connect1().cursor(), username) == None:
print('This username doesn\'t exists, submit a correct username')
else:
user1 = User.load_user_by_username(connect1().cursor(), username)
if check_password(str(password), user1.hashed_password):
a = User.load_user_by_id(connect1().cursor(), to_id)
if a != None:
if len(text) < 255:
from_id = user1.id
message1 = Messages(from_id, to_id, text)
message1.save_to_db(connect1().cursor())
return "Message sent"
else:
print('Your message is too long, should be maximum 255 characters')
else:
print("Receiver\'s id is incorrect")
else:
print('Wrong password')
def send_message(cursor, username, password, to_user, content):
user = User.load_user_by_name(cursor, username)
if not user:
print("User does not exist.")
elif check_password(password, user.hashed_password):
user2 = User.load_user_by_name(cursor, to_user)
if not user2:
print("That user does not exist.")
else:
if len(content) < 255:
message = Messages(from_id=user.id,
to_id=user2.id,
text=content)
message.save_to_db(cursor)
print("Message send.")
else:
print("Your message is too long, maximum 254 characters.")
else:
print("Incorrect password.")
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)
new_msg.save_to_db()
except AttributeError:
print(f'No {receiver} user')
