Cleanup, more typing for queries

This commit is contained in:
Thomas Sileo
2019-08-01 22:24:18 +02:00
parent d4590996e0
commit 5b9a0cb286
5 changed files with 43 additions and 28 deletions

19
core/db.py Normal file
View File

@@ -0,0 +1,19 @@
from enum import Enum
from enum import unique
from typing import Dict
from typing import Any
from typing import Optional
from config import DB
_Q = Dict[str, Any]
_Doc = Optional[Dict[str, Any]]
@unique
class CollectionName(Enum):
ACTIVITIES = "activities"
def find_one_activity(q: _Q) -> _Doc:
return DB[CollectionName.ACTIVITIES.value].find_one(q)