自定义Mongodb的操作接口
自定义Mongodb的操作接口
from pymongo import MongoClient
class MongoAPI(object):
def __init__(self, db_name, table_name, host="127.0.0.1", port=27017):
self.db_name = db_name
self.table_name = table_name
self.host = host
self.port = port
self.conn = MongoClient(host=self.host, port=self.port)
self.db = self.conn[self.db_name]
self.table = self.db[self.table_name]
def get_one(self, query):
# 获取一个指定的值,并且将_id字段不返回
return self.table.find_one(query, projection={'_id': False})
def get_query_all(self, query):
# 返回符合查询条件的所有值
return self.table.find(query)
def get_all(self):
# 返回所有数据
return self.table.find({})
def add(self, kv_dict):
# 插入一个字典值
return self.table.insert_one(kv_dict)
def delete(self, query):
# 删除指定条件的所有值
return self.table.delete_many(query)
def is_exist(self, query):
# 判断是否有值满足当前的查询条件
ret = self.table.find_one(query)
return ret is not None
def update(self, query, kv_dict):
# 更新记录,如果没有就会插入
self.table.update_one(query, {
'$set': kv_dict,
}, upsert=True)
def q_sort(self,column, order=1, query={}):
# 查询结果后,按照column进行排序通过order来进行排序,默认为升序order:1,返回所有数据 降序为-1,
return self.table.find(query).sort(column, order)
if __name__ == '__main__':
ma = MongoAPI(db_name='Student', table_name="16jixinan")
students = [{'id': 1, 'name': 'hhf', 'age': 22, 'hobby': ['足球', '篮球']},
{'id': 2, 'name': 'jyf', 'age': 26, 'hobby': ['wzry', '篮球']},
{'id': 25, 'name': 'pjl', 'age': 15, 'hobby': ['ship', '篮球']}]
for student in students:
# 去重插入
ma.update({'id': student.get('id')}, student)
students_all = ma.get_all()
for student in students_all:
print(student)
print('--'*20)
student_sort = ma.q_sort('age') # 按照age进行升序排序
for student in student_sort:
print(student)
print('--' * 20)
student_is_exist = ma.is_exist({'name': 'hhf'}) # 判读满足条件的记录是否存在
print(student_is_exist)
student_is_exist = ma.is_exist({'name': 'ww'})
print(student_is_exist)
ma.delete({'id': 2}) # 删除id为2的值
测试结果:
{'_id': ObjectId('5e96be7b8c584f65c8d0c35d'), 'id': 1, 'name': 'hhf', 'age': 22, 'hobby': ['足球', '篮球']}
{'_id': ObjectId('5e96be7b8c584f65c8d0c35f'), 'id': 25, 'name': 'pjl', 'age': 15, 'hobby': ['ship', '篮球']}
{'_id': ObjectId('5e96c44ab7333bc6ed6683be'), 'id': 2, 'age': 26, 'hobby': ['wzry', '篮球'], 'name': 'jyf'}
----------------------------------------
{'_id': ObjectId('5e96be7b8c584f65c8d0c35f'), 'id': 25, 'name': 'pjl', 'age': 15, 'hobby': ['ship', '篮球']}
{'_id': ObjectId('5e96be7b8c584f65c8d0c35d'), 'id': 1, 'name': 'hhf', 'age': 22, 'hobby': ['足球', '篮球']}
{'_id': ObjectId('5e96c44ab7333bc6ed6683be'), 'id': 2, 'age': 26, 'hobby': ['wzry', '篮球'], 'name': 'jyf'}
True
False