updating database names to run in prepod jubilant

db_script_for_location.py

@@ -3,9 +3,9 @@ from pymongo import MongoClient
 CONNECTION_STRING = "mongodb://ilensread:ilens@34.93.53.174:57017/?authMechanism=DEFAULT&directConnection=true"
 client = MongoClient(CONNECTION_STRING)
 
-mydatabase = client["project_181__ilens_assistant"]
+mydatabase = client["ilens_assistant"]
 task_instance_data = mydatabase['task_instance_data']
-task_instance=mydatabase['task_instances']
+task_instance = mydatabase['task_instances']
 
 query = {"step_id": "step_1165", "step_data.data.Location_1": {"$exists": True}}
 records = list(mydatabase.task_instance_data.find(query))
@@ -13,7 +13,7 @@ records = list(mydatabase.task_instance_data.find(query))
 data_list = []
 for record in records:
     if "step_data" in list(record.keys()) and "data" in list(record['step_data'].keys()):
-        data_list.append({"task_id":record.get("task_id"), "Location_1": record['step_data']['data'].get("Location_1")})
+        data_list.append({"task_id": record.get("task_id"), "Location_1": record['step_data']['data'].get("Location_1")})
 
 for record in data_list:
     query = {"task_id": record['task_id']}

db_script_to_update_batch_downtime_step_keys.py

@@ -20,7 +20,7 @@ data_mapping = {'bBatchNoB6': 'step_no',
                 'bBatchNoB5': 'loss_reason_comment',
                 }
 for record in records:
-    if 'step_data' in list(record.keys()) and 'data' in list(record['step_data'].keys()) and 'data_grid_non_periodic' in list(record['step_data']['data'].keys()):
+    if 'step_data' in list(record.keys()) and 'data' in list(record['step_data'].keys()) and 'data_grid_non_periodic' in list(record['step_data']['data'].keys()) and record['step_data']['data']['data_grid_non_periodic']:
         for data in record['step_data']['data']['data_grid_non_periodic']:
             for key in list(data.keys()):
                 if key in list(data_mapping.keys()):

db_script_to_update_downtime_common_step_keys.py

@@ -4,7 +4,7 @@ import psycopg2
 CONNECTION_STRING = "mongodb://@34.93.53.174:57017/?authMechanism=DEFAULT&directConnection=true"
 client = MongoClient(CONNECTION_STRING)
 
-mydatabase = client["project_181__ilens_assistant"]
+mydatabase = client["ilens_assistant"]
 mycollection = mydatabase['periodic_data']
 
 query = {"step_id": "step_1414"}
@@ -20,13 +20,13 @@ data_mapping = {'bBatchNoB6': 'step_no',
                 'bBatchNoB5': 'loss_reason_comment',
                 }
 for record in records:
-    if 'manual_data' in list(record.keys()) and 'prod_losses' in list(record['manual_data'].keys()):
+    if 'manual_data' in list(record.keys()) and 'prod_losses' in list(record['manual_data'].keys()) and record['manual_data']['prod_losses']:
         for data in record['manual_data']['prod_losses']:
             for key in list(data.keys()):
                 if key in list(data_mapping.keys()):
                     data[data_mapping[key]] = data[key]
                     del data[key]
-    if "manual_data" in list(record.keys()) and "data_grid" in list(record['manual_data'].keys()):
+    if "manual_data" in list(record.keys()) and "data_grid" in list(record['manual_data'].keys()) and record['manual_data']['data_grid']:
         for data in record['manual_data']['data_grid']:
             for key in list(data.keys()):
                 if key in list(data_mapping.keys()):
@@ -35,7 +35,7 @@ for record in records:
 
 list_of_data = []
 for record in records:
-    if "manual_data" in list(record.keys()) and "data_grid" in list(record['manual_data'].keys()):
+    if "manual_data" in list(record.keys()) and "data_grid" in list(record['manual_data'].keys()) and record['manual_data']['data_grid']:
         dictionary = {}
         for data in record['manual_data']['data_grid']:
             dictionary = {'date': record.get("date"), 'step_id': record.get("step_id")}