python操作MongoDB(API)

from pymongo import MongoClient
from datetime import datetime


class TestMongo(object):

    def __init__(self):
        self.client = MongoClient()   # 连接数据库
        self.db = self.client['blog']   # 使用blog数据库

    def add_one(self):
        """新增数据"""
        post = {
            'title': '新的标题',
            'content': '博客内容。。。。',
            'created_at': datetime.now(),
            'view': 10
        }
        return self.db.blog.posts.insert_one(post)    # 数据库下的集合（表）名为 blog.posts

    def get_one(self):
        """查询一条数据"""
        return self.db.blog.posts.find_one()

    def get_more(self):
        """查询所有数据"""
        return self.db.blog.posts.find()

    def get_from_oid(self, oid):
        """根据记录的ID来获取数据"""
        from bson.objectid import ObjectId    # 使用ObjectID才能查询到数据
        return self.db.blog.posts.find_one({'_id': ObjectId(oid)})

    def update(self):
        """修改数据"""
        # 修改一条数据
        rest = self.db.blog.posts.update_one({'view': 10}, {'$inc': {'view': 1}})   # $inc将view增加1，$set会将view设置成指定值
        print(rest.matched_count)     # matched_count 匹配的次数
        print(rest.modified_count)    # modified_count 修改成功的数据条数
        # 修改多条数据
        rest = self.db.blog.posts.update_many({}, {'$inc': {'view': 10}})   # 将所有的view增加10
        print(rest.matched_count)
        print(rest.modified_count)

    def delete(self):
        """删除数据"""
        # 删除一条数据
        rest = self.db.blog.posts.delete_one({'view': 10})
        print(rest.deleted_count)   # deleted_count 删除成功的数据条数
        # 删除多条数据
        rest = self.db.blog.posts.delete_many({'title': '新的标题'})
        print(rest.deleted_count)


def main():
    obj = TestMongo()
    # rest = obj.add_one()
    # rest = obj.get_more()
    # for item in rest:
    #     print(item)
    # obj.update()
    # obj.delete()


if __name__ == '__main__':
    main()