set log source

db/DbManager.py

@@ -14,6 +14,8 @@ TIMEOUT = "timeout_items"
 
 
 class DataManager:
+    batch_size = 20
+
     def __init__(self):
         cred = credentials.Certificate("appointment.json")
         self._app = firebase_admin.initialize_app(cred)
@@ -50,25 +52,44 @@ class DataManager:
         doc_ref = self._db.collection(TIMEOUT).document(str(contact.phone))
         doc_ref.set(contact.to_firestore_dict())
 
+    def _delete_collection(self, coll_ref, batch_size):
+        docs = coll_ref.limit(batch_size).stream()
+        deleted = 0
+
+        for doc in docs:
+            print(f'Deleting doc {doc.id} => {doc.to_dict()}')
+            doc.reference.delete()
+            deleted = deleted + 1
+
+        if deleted >= batch_size:
+            return self._delete_collection(coll_ref, batch_size)
+
+    def clear_all_sim_info(self):
+        coll_ref = self._db.collection(SIM_INFOS)
+        self._delete_collection(coll_ref, self.batch_size)
+
+    def save_to_excel(self):
+        # Start from the first cell. Rows and columns are zero indexed.
+        sim_info_list = []
+
+        for sim in self.get_all_sim_infos().stream():
+            print(sim)
+            sim_pojo = SimInfoPojo.from_firestore_dict(sim.to_dict())
+            sim_info_list.append(sim_pojo)
+        row = 0
+        col = 0
+        # Create a workbook and add a worksheet.
+        workbook = xlsxwriter.Workbook('sim_infos.xlsx')
+        worksheet = workbook.add_worksheet()
+        for info in sim_info_list:
+            # Iterate over the data and write it out row by row.
+            worksheet.write(row, col, info.phone[2:len(info.phone)])
+            worksheet.write(row, col + 1, info.ccid)
+            worksheet.write(row, col + 2, info.position)
+            row += 1
+        workbook.close()
+
 
 if __name__ == '__main__':
-
-    # Start from the first cell. Rows and columns are zero indexed.
-    sim_info_list = []
-
-    for sim in params.firebase_store_manager.get_all_sim_infos().stream():
-        print(sim)
-        sim_pojo = SimInfoPojo.from_firestore_dict(sim.to_dict())
-        sim_info_list.append(sim_pojo)
-    row = 0
-    col = 0
-    # Create a workbook and add a worksheet.
-    workbook = xlsxwriter.Workbook('sim_infos.xlsx')
-    worksheet = workbook.add_worksheet()
-    for info in sim_info_list:
-        # Iterate over the data and write it out row by row.
-        worksheet.write(row, col, info.phone[2:len(info.phone)])
-        worksheet.write(row, col + 1, info.ccid)
-        worksheet.write(row, col + 2, info.position)
-        row += 1
-    workbook.close()
+    # params.firebase_store_manager.save_to_excel()
+    params.firebase_store_manager.clear_all_sim_info()