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)