feature(#134): update API for get contacts by ids in firebase database

Brain/logs/activity.log

@@ -146,3 +146,5 @@
 2023-07-03 15:34:55,373 WARNING firebase init this app name: firebase_admin_45qrtaer345tret does not exist
 2023-07-03 16:21:15,252 WARNING firebase init this app name: firebase_admin_45qrtaer345tret does not exist
 2023-07-03 17:01:09,415 WARNING firebase init this app name: firebase_admin_45qrtaer345tret does not exist
+2023-07-03 17:45:56,553 WARNING firebase init this app name: firebase_admin_45qrtaer345tret does not exist
+2023-07-04 05:52:58,591 WARNING firebase init this app name: firebase_admin_45qrtaer345tret does not exist

Brain/src/model/requests/request_model.py

@@ -15,10 +15,10 @@ class ClientInfo:
 
     def is_browser(self) -> bool:
         if (
-            self.browser == "Chrome"
-            or self.browser == "Firefox"
-            or self.browser == "Safari"
-            or self.browser == "Edge"
+                self.browser == "Chrome"
+                or self.browser == "Firefox"
+                or self.browser == "Safari"
+                or self.browser == "Edge"
         ):
             return True
         return False
@@ -215,3 +215,10 @@ class AutoTaskDelete(BasicReq):
         reference_link: str
 
     data: Body
+
+
+"""endpoint : /contact/getbyIds"""
+
+
+class GetContactsByIds(BasicReq):
+    contactIds: list[str]

Brain/src/router/api.py

@@ -18,7 +18,7 @@ from Brain.src.model.requests.request_model import (
     BasicReq,
     ClientInfo,
     get_client_info,
-    AutoTaskDelete,
+    AutoTaskDelete, GetContactsByIds,
 )
 from Brain.src.rising_plugin.risingplugin import (
     getCompletion,
@@ -495,7 +495,10 @@ def construct_blueprint_api() -> APIRouter:
     """@generator.request_body(
             {
                 "token": "String",
-                "uuid": "String", 
+                "uuid": "String",
+                "contactIds": [
+                    String
+                ]
             }
         )
 
@@ -505,8 +508,8 @@ def construct_blueprint_api() -> APIRouter:
 
         """
 
-    @router.post("/contacts/getByUUID")
-    def getContactsByUUID(data: BasicReq):
+    @router.post("/contacts/getByIds")
+    def getContactsByIds(data: GetContactsByIds):
         try:
             setting, firebase_app = firebase_admin_with_setting(data)
         except BrainException as ex:
@@ -515,7 +518,8 @@ def construct_blueprint_api() -> APIRouter:
         token: str = setting.token
         uuid: str = setting.uuid
 
-        result = ContactsService(firebase_app=firebase_app, setting=setting).getContactsByUUID(uuid=uuid)
+        result = ContactsService(firebase_app=firebase_app, setting=setting).getContactsByIds(uuid=uuid,
+                                                                                              contactIds=data.contactIds)
 
         return assembler.to_response(
             200, "Success to get contacts by uuid", result

Brain/src/service/contact_service.py

@@ -103,11 +103,13 @@ class ContactsService:
         return get_pinecone_index_namespace(f"{uuid}-contacts")
 
     """init firestore to save user's contacts"""
+
     def init_firestore(self):
         self.db = firestore.client(app=self.firebase_app)
         self.phones_ref = self.db.collection("phones")
 
     """create a contact into document which name is uuid in phone collections in firestore"""
+
     def create_one_contact(self, uuid: str, contact: ContactModel):
         assembler = Assembler()
         data = assembler.to_contact_result_format(contact)
@@ -116,6 +118,7 @@ class ContactsService:
         contacts_doc_ref.set(data)
 
     """update a contact into document which name is uuid in phone collections in firestore"""
+
     def update_one_contact(self, uuid: str, contact: ContactModel):
         assembler = Assembler()
         data = assembler.to_contact_result_format(contact)
@@ -124,19 +127,18 @@ class ContactsService:
         contacts_doc_ref.update(data)
 
     """delete a contact into document which name is uuid in phone collections in firestore"""
+
     def delete_one_contact(self, uuid: str, contact: ContactModel):
         phones_doc_ref = self.phones_ref.document(uuid)
         contacts_doc_ref = phones_doc_ref.collection("contacts").document(contact.contact_id)
         contacts_doc_ref.delete()
 
-
-
-    def getContactsByUUID(self, uuid: str) -> []:
+    def getContactsByIds(self, uuid: str, contactIds: list[str]) -> []:
         phones_doc_ref = self.phones_ref.document(uuid)
         contacts_ref = phones_doc_ref.collection("contacts")
 
         # Retrieve all documents in the 'contacts' sub-collection
-        contacts = contacts_ref.stream()
+        contacts = contacts_ref.where("contactId", "in", contactIds).stream()
 
         result = []
         # Iterate through the documents and print out their data