def get_messages(id_chat): """This function get a messages from a chat.""" response = Response(json.dumps( json_error(ResponsesREST.INVALID_INPUT.value)), status=ResponsesREST.INVALID_INPUT.value, mimetype="application/json") if validator_id.is_valid({"id": id_chat}): get_message = Message() get_message.id_chat = id_chat result = get_message.get_messages_list() if result in (ResponsesREST.NOT_FOUND.value, ResponsesREST.SERVER_ERROR.value): response = Response(json.dumps(json_error(result)), status=result, mimetype="application/json") else: list_chat = [] for message_found in result: list_chat.append(message_found.json_message()) response = Response(json.dumps(list_chat), status=ResponsesREST.SUCCESSFUL.value, mimetype="application/json") return response
def add_message(): """This function add a message to chat.""" json_values = request.json values_required = {"message", "idChat", "memberType"} response = Response(json.dumps( json_error(ResponsesREST.INVALID_INPUT.value)), status=ResponsesREST.INVALID_INPUT.value, mimetype="application/json") if all(key in json_values for key in values_required): if validator_message.is_valid(json_values): message_add = Message() message_add.message = json_values["message"] message_add.id_chat = json_values["idChat"] message_add.member_type = json_values["memberType"] result = message_add.add_message() if result == ResponsesREST.CREATED.value: response = Response(json.dumps(message_add.json_message()), status=ResponsesREST.CREATED.value, mimetype="application/json") else: response = Response(json.dumps(json_error(result)), status=result, mimetype="application/json") return response