replace keys

replace_mongo_properties.py

 import json
 from pymongo import MongoClient
 
-input_json_file_name = 'input_json.json'
+MONGO_URI = "mongodb://ilens:ilens4321@192.168.0.220:2717/"
+database_ilens_assistant = "ilens_assistant"
+database_ilens_configuration = "ilens_configuration"
+
+input_json_file_name = 'input_json.json'  # input_json_file
 input_project_id = "project_099"
-input_logbook_id = "logbook_131"
-input_step_id = 'step_182'
+input_logbook_id = ["logbook_131"]
+input_step_id = ['step_182']
 step_type = "NON-PERIODIC"  # NON-PERIODIC,PERIODIC
 
-CONNECTION_STRING = "mongodb://ilens:ilens4321@192.168.0.220:2717/"
+CONNECTION_STRING = MONGO_URI
 client = MongoClient(CONNECTION_STRING)
 
-# add correct database names below
-ilens_assistant = client["ilens_assistant"]
-ilens_configuration = client["ilens_configuration"]
+ilens_assistant = client[database_ilens_assistant]
+ilens_configuration = client[database_ilens_configuration]
 
 task_instances = ilens_assistant['task_instances']
 task_instance_data = ilens_assistant['task_instance_data']
 periodic_data = ilens_configuration['periodic_data']
 
+# ---------------------------code execution---------------------------------------#
+
 with open(input_json_file_name) as user_file:
     input_json = user_file.read()
     input_dict = json.loads(input_json)
@@ -37,6 +42,7 @@ def replace_keys(data, replacements):
 
 if step_type == 'NON-PERIODIC':
 
+    for logbook_id in input_logbook_id:
         query = {"logbook_id": input_logbook_id, "project_id": input_project_id}
         data = []
         if logbook_data := list(ilens_assistant.task_instances.find(query)):
@@ -51,12 +57,13 @@ if step_type == 'NON-PERIODIC':
                 updated_y.append(item)
             for item in updated_y:
                 if item['task_id'] in data:
-                update_query = {"task_id": item['task_id'], "step_id": input_step_id, "project_id": input_project_id}
+                    update_query = {"task_id": item['task_id'], "step_id": input_step_id,
+                                    "project_id": input_project_id}
                     ilens_assistant.task_instance_data.update_one(update_query, {"$set": item})
 
 elif step_type == "PERIODIC":
-
-    query = {"step_id": input_step_id}
+    for step_id in input_step_id:
+        query = {"step_id": step_id}
         periodic_step_data = list(ilens_assistant.periodic_data.find(query))
         updated_y = []
         if periodic_step_data: