-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.py
92 lines (78 loc) · 2.73 KB
/
db.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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
from mongoengine import *
from model import Todo
import json
import datetime
import os
from service import sec
class DB:
def __init__(self):
db_name = os.environ.get("DB_NAME", "todos")
if os.environ.get("FLASK_ENV", "development") != "testing":
connect(db_name, host="mongo", port=27017, username="admin", password="admin", authentication_source="admin")
else:
connect(db_name, host="localhost", port=27017)
def createTodo(self, payload):
auth = sec.context.get_auth()
newTodo = Todo(
title=payload.get("title"),
description=payload.get("description"),
createdBy=auth.user_id
)
try:
newTodo.save()
except ValidationError as error:
errorMessage = {
"msg": str(error)
}
return json.dumps(errorMessage)
return newTodo.to_json()
def getAllTodos(self, limit=10):
listTodos = Todo.objects[:limit]
return listTodos.to_json()
def getTodoById(self, todoId):
try:
extistingTodo = Todo.objects.get(id=todoId)
except Todo.DoesNotExist as error:
errorMessage = {
"msg": str(error)
}
return json.dumps(errorMessage)
return extistingTodo.to_json()
def deleteTodo(self, todoId):
try:
deletedTodo = Todo.objects.get(id=todoId)
except Todo.DoesNotExist as error:
errorMessage = {
"msg": str(error)
}
return json.dumps(errorMessage)
deletedTodoTitle = deletedTodo.title
try:
deletedTodo.delete()
except Todo.DoesNotExist as error:
errorMessage = {
"msg": str(error)
}
return json.dumps(errorMessage)
message = {
"msg": "The todo with title: {0} is now deleted".format(deletedTodoTitle)
}
return json.dumps(message)
def updateTodo(self, todoId, payload):
try:
updateTodo = Todo.objects.get(id=todoId)
except Todo.DoesNotExist as error:
errorMessage = {
"msg": str(error)
}
return json.dumps(errorMessage)
if payload.get("title") is not None:
updateTodo.title = payload.get("title")
if payload.get("description") is not None:
updateTodo.description = payload.get("description")
if payload.get("done") is not None:
updateTodo.done = payload.get("done")
if updateTodo.done is True:
updateTodo.completedAt = datetime.datetime.now()
updateTodo.save()
return updateTodo.to_json()