forked from Jackiebibili/ticket_tracker_api
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbase.py
More file actions
57 lines (43 loc) · 1.8 KB
/
base.py
File metadata and controls
57 lines (43 loc) · 1.8 KB
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
from pymongo import collection
from uuid import uuid4
from datetime import datetime
def insert_one__(coll: collection.Collection, doc: dict):
return coll.insert_one(doc)
def insert_many__(coll: collection.Collection, docs: list):
return coll.insert_many(docs)
def delete_one__(coll: collection.Collection, filter: dict):
return coll.delete_one(filter)
def delete_many__(coll: collection.Collection, filter: dict):
return coll.delete_many(filter)
def find_one_and_replace__(coll: collection.Collection, filter: dict, new_doc: dict):
new_doc.update(gen_additional_attrs())
return coll.find_one_and_replace(filter, new_doc)
def find_one_and_update__(coll: collection.Collection, filter: dict, update=dict):
meta_attrs = get_meta_attrs()
key = "$set"
if key in update:
update[key].update(meta_attrs)
else:
update[key] = meta_attrs
return coll.find_one_and_update(filter, update)
def find_one_and_delete__(coll: collection.Collection, filter: dict):
return coll.find_one_and_delete(filter)
def find_one__(coll: collection.Collection, filter: dict, projection, **kwargs):
return coll.find_one(filter=filter, projection=projection, **kwargs)
def find_many__(coll: collection.Collection, filter: dict, projection, **kwargs):
return coll.find(filter=filter, projection=projection, **kwargs)
def count_docs__(coll: collection.Collection, filter: dict):
return coll.count_documents(filter=filter)
def estimated_count_docs__(coll: collection.Collection):
return coll.estimated_document_count()
def watch__(coll: collection.Collection, **kwargs):
return coll.watch(**kwargs)
def gen_additional_attrs():
return {
"_id": str(uuid4()),
**get_meta_attrs()
}
def get_meta_attrs():
return {
"last_modified": datetime.today().replace(microsecond=0)
}