git-sync/crawlab
1
0
Fork 0
mirror of https://github.com/crawlab-team/crawlab.git synced 2026-01-26 17:49:15 +01:00
Code Issues Packages Projects Releases Wiki Activity

added base class for api handling

Browse Source
This commit is contained in:
Yeqing Zhang
2019-02-12 06:58:54 +08:00
parent 2610f76719
commit 04c5d69c9f
12 changed files with 136 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
db/manager.py
View File

@@ -1,6 +1,10 @@
from bson import ObjectId
from mongoengine import connect
from pymongo import MongoClient
from config.db import MONGO_HOST, MONGO_PORT, MONGO_DB
from utils import is_object_id
connect(db=MONGO_DB, host=MONGO_HOST, port=MONGO_PORT)
class DbManager(object):
@@ -32,8 +36,12 @@ class DbManager(object):
return data
def get(self, col_name: str, id: str):
if is_object_id(id):
_id = ObjectId(id)
else:
_id = id
col = self.db[col_name]
return col.find_one({'_id': ObjectId(id)})
return col.find_one({'_id': _id})
def count(self, col_name: str, cond):
col = self.db[col_name]