-
Notifications
You must be signed in to change notification settings - Fork 172
/
Copy pathtask.py
41 lines (30 loc) · 1 KB
/
task.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# coding=utf-8
import pymongo
import json
from bson import json_util
from bson.objectid import ObjectId
class TaskDAO:
def __init__(self, database):
self.database = database
self.task_collection = self.database.task
def create(self, data):
task = {
'title': data.get('title'),
'description': data.get('description'),
'done': False
}
inserted_id = self.task_collection.insert_one(task).inserted_id
task = self.task_collection.find_one({ '_id': ObjectId(inserted_id) })
return self.to_json(task)
def list(self):
tasks = self.task_collection.find().sort('done', pymongo.ASCENDING)
return self.to_json(tasks)
def read(self, object_id):
task = self.task_collection.find_one({ '_id': ObjectId(object_id)})
return self.to_json(task)
def update(self):
pass
def delete(self):
pass
def to_json(self, data):
return json.loads(json_util.dumps(data))