From afe3395d48ca693343afb179c1ffd11ae187e6f5 Mon Sep 17 00:00:00 2001 From: SecretlyTaco Date: Sat, 17 Mar 2018 01:40:28 +0000 Subject: [PATCH] Created messages.py with basic send_message function --- backend/messages.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 backend/messages.py diff --git a/backend/messages.py b/backend/messages.py new file mode 100644 index 0000000..63245c4 --- /dev/null +++ b/backend/messages.py @@ -0,0 +1,28 @@ +from pymongo import MongoClient +from bson.objectid import ObjectId +from backend import accounts + +client = MongoClient() +db = client.tweeder +accounts_db = db.accounts +messages_db = db.messages + + +def send_message(msg_from, msg_to, msg_content): + if type(msg_from) == str: + from_id = accounts.get_profile(msg_from)['_id'] + elif type(msg_from) == ObjectId: + from_id = msg_from + + if type(msg_to) == str: + to_id = accounts.get_profile(msg_to)['_id'] + elif type(msg_to) == ObjectId: + to_id = msg_to + + message = { + 'from': from_id, + 'to': to_id, + 'content': msg_content + } + + messages_db.insert_one(message) \ No newline at end of file