def load_all_msg(cursor):
sql = "SELECT * FROM messages"
ret = []
cursor.execute(sql)
for row in cursor.fetchall():
loaded_msg = Message()
loaded_msg._id = row[0]
from_user_temp = UserService.find_by_id(cursor, row[1])
loaded_msg.from_user = from_user_temp.username
to_user_temp = UserService.find_by_id(cursor, row[2])
loaded_msg.to_user = to_user_temp.username
loaded_msg.context = row[3]
loaded_msg.created_at = row[4]
ret.append(loaded_msg)
return ret
def find_by_recipient(cursor, recipient_username):
sql = "SELECT id, from_user, to_user, context, created_at FROM messages WHERE to_user=%s"
ret = []
cursor.execute(sql, (recipient_username, ))
data = cursor.fetchall()
if data:
for row in data:
loaded_msg = Message()
loaded_msg._id = row[0]
from_user_temp = UserService.find_by_id(cursor, row[1])
loaded_msg.from_user = from_user_temp.username
to_user_temp = UserService.find_by_id(cursor, row[2])
loaded_msg.to_user = to_user_temp.username
loaded_msg.context = row[3]
loaded_msg.created_at = row[4]
ret.append(loaded_msg)
return ret
else:
return None
user.set_password('qwerty12', 'testowa-sol')
user.save(cursor)
print('User created')
if args.list == True:
print('List all users from database')
users = UserService.get_all(cursor)
for user in users:
print(user)
if args.list_msg == True:
user = UserService.find_by_username(cursor, args.username)
if user is not None:
is_user_logged = UserService.login(cursor, args.username, args.password)
if args.from_user:
from_user = UserService.find_by_id(cursor, args.from_user)
print(f"Messages send from: {from_user.username}")
msgs = MessageService.load_all_messages_by_user(cursor, args.from_user)
if msgs is not None:
for msg in msgs:
print(msg)
elif args.to_user:
to_user = UserService.find_by_id(cursor, args.to_user)
username = to_user.username
print(f"Messages send to {username}")
msgs = MessageService.find_by_recipient(cursor, args.to_user)
if msgs is not None:
for msg in msgs:
print(msg)
