80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
from pymongo import MongoClient
|
|
import re
|
|
from bson import ObjectId
|
|
import json
|
|
from app_fb.models.user import MongoUserDao
|
|
|
|
mongo_user_dao = MongoUserDao()
|
|
|
|
class Post:
|
|
def __init__(self, user, post_type, post_content, updated_time, post_size, likes, comments):
|
|
self.user = user
|
|
self.post_type = post_type
|
|
self.post_content = post_content
|
|
self.updated_time = updated_time
|
|
self.post_size = post_size
|
|
self.likes = likes
|
|
self.comments = comments
|
|
|
|
|
|
class MongoPostDao:
|
|
def __init__(self):
|
|
config = json.load(open('./config.json', 'r'))
|
|
client = MongoClient(config['mongo_host'], config['mongo_port'])
|
|
self.db = client[config['db_name']]
|
|
|
|
def save(self, post):
|
|
self.db.posts.insert_one(post)
|
|
|
|
def search_by_content(self, post_content):
|
|
result_cursor = self.db.posts.find({'post_content': re.compile(post_content, re.IGNORECASE)})
|
|
matches = []
|
|
for post in result_cursor:
|
|
matches.append(post)
|
|
return matches
|
|
|
|
def delete_by_id(self, _id):
|
|
self.db.posts.delete_one({'_id': ObjectId(_id)})
|
|
|
|
def update_by_id(self, _id, post):
|
|
self.db.posts.update_one({'_id': ObjectId(_id)}, {'$set': post})
|
|
|
|
def list_all_posts_for_user(self, user_id):
|
|
user = mongo_user_dao.get_by_id(user_id)
|
|
if 'posts' in user:
|
|
return user['posts']
|
|
else:
|
|
return []
|
|
|
|
def find_posts_by_type(self, post_type):
|
|
a = 3 if True else 1
|
|
result_cursor = self.db.posts.find({'post_type': post_type})
|
|
matches = []
|
|
for post in result_cursor:
|
|
matches.append(post)
|
|
return matches[0]['first_name'] + " " + matches[0]['last_name'] if len(matches) > 0 else None
|
|
|
|
def get_by_id(self, _id):
|
|
query = {
|
|
'_id': ObjectId(_id)
|
|
}
|
|
cursor = self.db.posts.find(query)
|
|
post = cursor[0] if cursor.count() > 0 else None
|
|
return post
|
|
|
|
def get_id_by_post_content(self, post_content):
|
|
cursor = self.db.posts.find({'post_content': post_content})
|
|
post_data = cursor[0] if cursor.count() > 0 else None
|
|
if post_data is None:
|
|
return "No posts found"
|
|
else:
|
|
matches = []
|
|
for post in cursor:
|
|
matches.append(post)
|
|
return matches[0]['_id']
|
|
|
|
def get_comments_by_postid(self, post_id):
|
|
post = self.get_by_id(post_id)
|
|
return post['comments']
|
|
|