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

added log file for task

Browse Source
This commit is contained in:
Marvin Zhang
2019-02-19 22:55:52 +08:00
parent a4a0fdd67b
commit 6b13975af4
4 changed files with 78 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
db/manager.py
View File

@@ -26,7 +26,10 @@ class DbManager(object):
def update_one(self, col_name: str, id: str, values: dict, **kwargs):
col = self.db[col_name]
col.find_one_and_update({'_id': ObjectId(id)}, {'$set': values})
_id = id
if is_object_id(id):
_id = ObjectId(id)
col.find_one_and_update({'_id': _id}, {'$set': values})
def remove_one(self, col_name: str, id: str, **kwargs):
col = self.db[col_name]
@@ -39,13 +42,14 @@ class DbManager(object):
data.append(item)
return data
def get(self, col_name: str, id: str):
if is_object_id(id):
def get(self, col_name: str, id):
if type(id) == ObjectId:
_id = id
elif is_object_id(id):
_id = ObjectId(id)
else:
_id = id
col = self.db[col_name]
print(_id)
return col.find_one({'_id': _id})
def count(self, col_name: str, cond):