Modified main.py

main.py

@@ -300,10 +300,10 @@ def nuc_monitor(rte_data, mongo_json_data, usr_start_date, usr_end_date, photo_d
 
 
 
-    mongo_data = mongo_json_data.json()
+    # mongo_data = mongo_json_data.json()
 
-        # Specify the file path
-    file_path = "/Users/diegomarroquin/HayaEnergy/Nucmonitor_MVP/NucPy_v0.2/testing/test_data3.txt"
+    #     # Specify the file path
+    # file_path = "/Users/diegomarroquin/HayaEnergy/Nucmonitor_MVP/NucPy_v0.2/testing/test_data3.txt"
 
     # Open the file in write mode
     with open(file_path, 'w') as file:
@@ -322,9 +322,9 @@ def nuc_monitor(rte_data, mongo_json_data, usr_start_date, usr_end_date, photo_d
     # Drop the original column
     mongo_df_result.drop(columns=['generation_unavailabilities'], inplace=True)
     mongo_df_columns = mongo_df_result.columns
-    print(mongo_df_columns)
-    print(mongo_df_result)
-    print(mongo_df_result["values"])
+    # print(mongo_df_columns)
+    # print(mongo_df_result)
+    # print(mongo_df_result["values"])
     # # Unpack values column
     # # mongo_df2 = mongo_df_result.copy().apply(unpack_values, axis=1)
     # mongo_df_values_unpacked = pd.json_normalize(mongo_df_result['values'])
@@ -342,8 +342,8 @@ def nuc_monitor(rte_data, mongo_json_data, usr_start_date, usr_end_date, photo_d
     mongo_df_result['end_date'] = mongo_df_result['values'].apply(lambda x: x[0]['end_date'])
     mongo_df_result['available_capacity'] = mongo_df_result['values'].apply(lambda x: x[0]['available_capacity'])
     mongo_df_result['unavailable_capacity'] = mongo_df_result['values'].apply(lambda x: x[0]['unavailable_capacity'])
-    print(mongo_df_result)
-    print(mongo_df_result.columns)
+    # print(mongo_df_result)
+    # print(mongo_df_result.columns)
     # Drop the original 'values' column
     mongo_df_result.drop('values', axis=1, inplace=True)
     mongo_df2 = mongo_df_result
@@ -362,7 +362,7 @@ def nuc_monitor(rte_data, mongo_json_data, usr_start_date, usr_end_date, photo_d
 
     # Merge dataframes
     column_order = mongo_unavs.columns
-    print(column_order)
+    # print(column_order)
     merged_df = pd.concat([mongo_unavs[column_order], rte_unavs[column_order]], ignore_index=True)
 
 # --------------------------- HERE IS THE CHANGE TO GET ONLY ACTIVE OR ACTIVE AND INACTIVE --------------------------- #
@@ -417,12 +417,12 @@ def nuc_monitor(rte_data, mongo_json_data, usr_start_date, usr_end_date, photo_d
     # Turn df into dict for json processing
     filtered_unavs = filtered_df3.copy().to_dict(orient='records')
 
-    file_path = "/Users/diegomarroquin/HayaEnergy/Nucmonitor_MVP/NucPy_v0.2/testing/test_data4.txt"
+    # file_path = "/Users/diegomarroquin/HayaEnergy/Nucmonitor_MVP/NucPy_v0.2/testing/test_data4.txt"
 
-    # Open the file in write mode
-    with open(file_path, 'w') as file:
-        for item in filtered_unavs:
-            file.write("%s" % item)
+    # # Open the file in write mode
+    # with open(file_path, 'w') as file:
+    #     for item in filtered_unavs:
+    #         file.write("%s" % item)
     results = {}
 
     for unav in filtered_unavs:
@@ -553,11 +553,11 @@ def nuc_monitor(rte_data, mongo_json_data, usr_start_date, usr_end_date, photo_d
 
     # print(output_results)
     add_total(output_results)
-    print("Done")
+    # print("Done")
     # print(results_plants)
     # Convert datetime key to string to store in mongodb
     output_results = {plant: {str(date): power for date, power in plant_data.items()} for plant, plant_data in output_results.items()}
-    print(output_results)
+    # print(output_results)
     # -------------------------------------------------
     if photo_date == False:
         # Store the results_plants in MongoDB
@@ -570,7 +570,7 @@ def nuc_monitor(rte_data, mongo_json_data, usr_start_date, usr_end_date, photo_d
         
         # json_data = json.dumps(convert_to_json(output_results))
         json_data = json.dumps(output_results)
-        print(json_data)
+        # print(json_data)
         return json_data
     else:
         database_name = "data"  # Specify your database name
@@ -579,7 +579,7 @@ def nuc_monitor(rte_data, mongo_json_data, usr_start_date, usr_end_date, photo_d
 
         # json_data = json.dumps(convert_to_json(output_results))
         json_data = json.dumps(output_results)
-        print(json_data)
+        # print(json_data)
         return json_data
     # -------------------------------------------------
     return