Python Comment.find方法代码示例

# 需要导入模块: from models.comment import Comment [as 别名]
# 或者: from models.comment.Comment import find [as 别名]
class PostModel(Model):

    def __init__(self, db, collection, obj):
        super(PostModel, self).__init__(db, collection, obj)
        self.user = obj['user']
        self.title = obj['title']
        self.body = obj['body']
        self.tags = obj['tags']
        self.upvotes = obj['upvotes']
        self.comments = Comment()

    # Increases the upvotes on a post
    def vote_up(self):
        self.upvotes += 1
        self.collection.objects.update({'_id': self.get_id()},
                {'$inc': {'upvotes': 1}})

    # Adds a comment on this post
    def add_comment(self, **kwargs):
        return self.comments.insert(post_id=self.get_id(), **kwargs)

    # Gets comments on this post
    def get_comments(self):
        return self.comments.find(post_id=self.get_id())

    # Use for removing comments by the user who created them, or by the poster
    def remove_comments(self, **kwargs):
        self.comments.remove(**kwargs)

    # Removes self and the comments
    def remove(self):
        self.remove_comments(post_id=self.get_id())
        super(PostModel, self).remove()

# 需要导入模块: from models.comment import Comment [as 别名]
# 或者: from models.comment.Comment import find [as 别名]
class UserModel(Model):

    def __init__(self, db, collection, obj):
        super(UserModel, self).__init__(db, collection, obj)
        self.username = obj['username']
        self.password = obj['password']
        self.voted = obj['voted']
        self.posts = Post()
        self.comments = Comment()

    # Change password with authentication--username auth to be done 
    # in middleware
    def change_password(self, oldpass, newpass):
        if oldpass == self.password:
            self.password = newpass
            self.collection.update({'_id': self.get_id()}, password=newpass)
            return True
        return False

    # Vote a post up, cannot do this more than once for any given post
    def vote_up(self, post_id):
        if not post_id in self.voted:
            p = self.posts.find_one(_id=post_id)
            p.vote_up()
            self.voted.append(post_id)
            self.collection.update({'_id': self.get_id()}, voted=self.voted)

    # Adds a post under the users page
    def add_post(self, **kwargs):
        return self.posts.insert(user=self.username, **kwargs)

    # Get blog posts made by this user, and with other arguments
    def get_posts(self, **kwargs):
        return self.posts.find(user=self.username, **kwargs)

    # Get comments made by the user, with other parameters
    def get_comments(self, **kwargs):
        return self.comments.find(user=self.username, **kwargs)

# 需要导入模块: from models.comment import Comment [as 别名]
# 或者: from models.comment.Comment import find [as 别名]
 def get_comments(self, **kwargs):
     comments = Comment()
     return comments.find(img_id=self.get_id(), **kwargs)