Junaidb commited on
Commit
9e8d747
·
verified ·
1 Parent(s): fd32229

Create getbreakdown.py

Browse files
Files changed (1) hide show
  1. components/getbreakdown.py +50 -0
components/getbreakdown.py ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from component.mongoconnection import provideClient
2
+
3
+ mc=provideClient()
4
+
5
+ db=mc["askmydoc_db"]
6
+ coll=db["askmydoc_col"]
7
+
8
+ def GetBreakdown(owner,typeofmsg,files):
9
+
10
+ match typeofmsg:
11
+ case "legal":
12
+
13
+ document = coll.find_one({"owner": owner})
14
+
15
+ if document is None or "breakdown" not in document or len(document["breakdown"]) == 0:
16
+ return {"breakdown":[]}
17
+ else:
18
+ filtered=[]
19
+ for msg in document["breakdown"]:
20
+ if msg["type"] == typeofmsg and any(f in files for f in msg["files"]):
21
+ filtered.append(msg)
22
+
23
+ return {"breakdown":filtered}
24
+
25
+ case "medical":
26
+ document = coll.find_one({"owner": owner})
27
+
28
+ if document is None or "breakdown" not in document or len(document["breakdown"]) == 0:
29
+ return {"breakdown":[]}
30
+ else:
31
+ filtered=[]
32
+ for msg in document["breakdown"]:
33
+ if msg["type"] == typeofmsg and any(f in files for f in msg["files"]):
34
+ filtered.append(msg)
35
+ return {"breakdown":filtered}
36
+
37
+
38
+ case "research":
39
+ document = coll.find_one({"owner": owner})
40
+
41
+ if document is None or "breakdown" not in document or len(document["breakdown"]) == 0:
42
+ return {"breakdown":[]}
43
+ else:
44
+ filtered=[]
45
+ for msg in document["breakdown"]:
46
+ if msg["type"] == typeofmsg and any(f in files for f in msg["files"]):
47
+ if msg["type"]==typeofmsg:
48
+ filtered.append(msg)
49
+
50
+ return {"breakdown":filtered}