def message_search():
search = request.args.get('search')
if not search:
messages = []
else:
messages = search_messages(search)
return jsonify({'count': len(messages), 'messages': messages})
def search():
if not all([x in request.args for x in ("keyword", )]) or len(
request.args.get("keyword", "")) == 0:
return render_template("search.html")
keyword = request.args["keyword"]
messages = list(
db.search_messages(session["user"]["id"],
{"author": {
"$regex": f".*{keyword}.*"
}}))
return render_template("search.html", results=messages)
def on_message(self, body, message):
response = None
if body["query"] == "all_messages":
response = list_messages()
elif body["query"] == "user_messages":
response = list_user_messages(body["user_id"])
elif body["query"] == "search_messages":
response = search_messages(body["search"])
elif body["query"] == "post_message":
response = post_message(body["user_id"], body["message"])
if response is not None:
response_prod = MessageResponseProducer(self.connection, messages_response_queue)
response_prod.put_message(
{"response": response, "message_id": body["message_id"]}, messages_response_exchange
)
message.ack()
def on_message(self, body, message):
response = None
if body['query'] == 'all_messages':
response = list_messages()
elif body['query'] == 'user_messages':
response = list_user_messages(body['user_id'])
elif body['query'] == 'search_messages':
response = search_messages(body['search'])
elif body['query'] == 'post_message':
response = post_message(body['user_id'], body['message'])
if response is not None:
response_prod = MessageResponseProducer(self.connection,
messages_response_queue)
response_prod.put_message(
{
'response': response,
'message_id': body['message_id']
}, messages_response_exchange)
message.ack()
