Merge remote-tracking branch 'origin/feature/mailer'

db/DbManager.py

@@ -3,7 +3,9 @@ import firebase_admin
 import xlsxwriter as xlsxwriter
 from firebase_admin import credentials, firestore
 
+import definitions
 import params
+from pojo.MailPojo import MailPojo
 from pojo.ReserveResultPojo import ReserveResultPojo, PublishType
 from pojo.SimInfoPojo import SimInfoPojo
 from pojo.contact_pojo import ContactPojo
@@ -11,6 +13,7 @@ from utils.excel_reader import ExcelHelper
 
 ERROR_COLLECTION_NAME = "error_items"
 CONTACT_COLLECTION_NAME = "contact_list"
+MAIL_COLLECTION_NAME = "mail_list"
 SIM_INFOS = "sim_infos"
 TIMEOUT = "timeout_items"
 
@@ -19,7 +22,7 @@ class DataManager:
     batch_size = 20
 
     def __init__(self):
-        cred = credentials.Certificate("appointment.json")
+        cred = credentials.Certificate(definitions.ROOT_DIR + "/appointment.json")
         self._app = firebase_admin.initialize_app(cred)
         self._db = firestore.client()
         contact_collection = self._db.collection(CONTACT_COLLECTION_NAME)
@@ -119,8 +122,17 @@ class DataManager:
         contact_collection = self._db.collection(CONTACT_COLLECTION_NAME)
         return contact_collection
 
+    def get_mail_list(self) -> list:
+        mail_collection = self._db.collection(MAIL_COLLECTION_NAME)
+        mail_list = []
+        for mail in mail_collection.stream():
+            mail_pojo = MailPojo.from_firestore_dict(mail.to_dict())
+            mail_list.append(mail_pojo)
+        return mail_list
+
 
 if __name__ == '__main__':
     # params.firebase_store_manager.upload_contact_list_to_cloud()
     params.firebase_store_manager.save_to_excel()
     # params.firebase_store_manager.clear_all_sim_info()
+    # print(params.firebase_store_manager.get_mail_list())