[SELC-5702] Added script to delete pecNotification entity for institu…

…tion with institutionType = PT (#44)

apps/institution-migration/delete_pecNotification_PT.py

@@ -0,0 +1,46 @@
+import math
+import os
+from dotenv import load_dotenv
+import time
+from pymongo import MongoClient
+from query import *
+
+load_dotenv(dotenv_path=".env", override=True)
+HOST = os.getenv('MONGO_HOST')
+
+CORE_DB = 'selcMsCore'
+INSTITUTION_COLLECTION = 'Institution'
+PEC_NOTIFICATION_COLLECTION = 'PecNotification'
+
+BATCH_SIZE = 100
+START_PAGE = 0
+
+def delete_pec_notificatio_for_pt(client):
+ print("Starting process to delete PecNotification for PT")
+
+ institutions_size_cursor = client[CORE_DB][INSTITUTION_COLLECTION].aggregate(count_pt_institutions())
+ institutions_size = next(institutions_size_cursor)['count']
+ print("Institutions size: " + str(institutions_size))
+ pages = math.ceil(institutions_size / BATCH_SIZE)
+ for page in range(START_PAGE, pages):
+ print("Start page " + str(page + 1) + "/" + str(pages))
+
+ institutions_pages = client[CORE_DB][INSTITUTION_COLLECTION].aggregate(
+ get_pt_institutions(page, BATCH_SIZE))
+
+ for institution in institutions_pages:
+ institution_id = institution['_id']
+ pec_notifications = client[CORE_DB][PEC_NOTIFICATION_COLLECTION].find({"institutionId": institution_id})
+
+ for pec_notification in pec_notifications:
+ pec_notification_id = pec_notification['_id']
+ print(f"founded PecNotification with _id: {pec_notification_id} for institutionId: {institution_id}")
+ client[CORE_DB][PEC_NOTIFICATION_COLLECTION].delete_one({"_id": pec_notification_id})
+ print(f"Deleted PecNotification with _id: {pec_notification_id}")
+
+ print("completed")
+
+if __name__ == "__main__":
+ client = MongoClient(HOST)
+ delete_pec_notificatio_for_pt(client)
+ client.close()

apps/institution-migration/query.py

@@ -29,6 +29,28 @@ def get_institutions(page, size):
  *paging(page, size)
  ]
 
+def count_pt_institutions():
+ return [
+ {
+ '$match': {
+ 'institutionType': 'PT'
+ }
+ },
+ {
+ '$count': 'count'
+ }
+ ]
+
+def get_pt_institutions(page, size):
+ return [
+ {
+ '$match': {
+ 'institutionType': 'PT'
+ }
+ },
+ *paging(page, size)
+ ]
+
 
 def get_delegations(page, size):
  return [