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, )) for row in cursor.fetchall(): loaded_message = Message() loaded_message._id = row[0] loaded_message.from_user = row[1] loaded_message.to_user = row[2] loaded_message.context = row[3] loaded_message.created_at = row[4] ret.append(loaded_message) return ret
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 = MessageService.find_by_recipient(cursor, args.username) print('Wrong data provided') if args.send == 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 is_user_logged == True: to_user = UserService.find_by_username(cursor, args.username) if to_user is not None: context = args.message if context: message = Message() message.from_user = user.id message.to_user = to_user.id message.context = args.message message.created_at = datetime.now() message.save(cursor) print("Message successfully sent") else: print("Message is empty!") else: print("Wrong username") else: print('Please login') else: print('Invalid username or password') cursor.close() connection.close()
else: print('Username or password invalid') if args.send == 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 is_user_logged == True: to_user = UserService.find_by_email(cursor, args.email) if to_user is not None: context = args.message if context: msg = Message() msg.from_user = user.id msg.to_user = to_user.id msg.context = args.message msg.created_at = datetime.now() msg.save(cursor) print("Message send") else: print("No message provided") else: print("No user with that email") else: print('Username or password invalid') else: print('Username or password invalid') cursor.close() connection.close()