Update magnetic.py

magnetic.py

@@ -899,9 +899,19 @@ def batch_requests(stations, dataset, lon, lat, date, distance=100):
                     'S': test_.get('S', []),
                 })
                 
-                if not plotted[['X', 'Y', 'Z', 'S']].any().any():
+                if plotted[['X', 'Y', 'Z', 'S']].any().any():
+                    all_data.append(plotted)
+                    all_event_times.append(pd.to_datetime(date_))
+                    additional_data = f"Date: {date_}\nLat/Lon: {lat_}, {lon_}\nClosest station: {station_name.values[0]}\nDistance: {round(station_distance.values[0], 2)} km"
+                    fig = plot_data_custom(plotted, date=pd.to_datetime(date_), save_path=None, subtitle=additional_data)
+                    with st.status(f'Magnetic Data: {date_}', expanded=False) as status:
+                        st.pyplot(fig)
+                        status.update(f'Magnetic Data: {date_} - Finished!')
+                else:
                     print(f"No data for X, Y, Z, or S for date: {date_}")
-                    return None
+        except Exception as e:
+            #print(f"An error occurred: {e}")
+            pass
 
             # if test_:
             #     results["station"].append(new_dataset.iloc[0]['IagaCode'])
@@ -913,17 +923,17 @@ def batch_requests(stations, dataset, lon, lat, date, distance=100):
             #         'Z': test_['Z'],
             #         'S': test_['S'],
             #     })
-                all_data.append(plotted)
-                all_event_times.append(pd.to_datetime(date_))
-                # print(date_)
-                additional_data = f"Date: {date_}\nLat/Lon: {lat_}, {lon_}\nClosest station: {station_name.values[0]}\n Distance:{round(station_distance.values[0],2)} km"
-                fig = plot_data_custom(plotted, date=pd.to_datetime(date_), save_path=None, subtitle =additional_data)
-                with st.status(f'Magnetic Data: {date_}', expanded=False) as status:
-                    st.pyplot(fig)
-                    status.update(f'Magnetic Data: {date_} - Finished!')
-        except Exception as e:
-            #print(f"An error occurred: {e}")
-            pass
+        #         all_data.append(plotted)
+        #         all_event_times.append(pd.to_datetime(date_))
+        #         # print(date_)
+        #         additional_data = f"Date: {date_}\nLat/Lon: {lat_}, {lon_}\nClosest station: {station_name.values[0]}\n Distance:{round(station_distance.values[0],2)} km"
+        #         fig = plot_data_custom(plotted, date=pd.to_datetime(date_), save_path=None, subtitle =additional_data)
+        #         with st.status(f'Magnetic Data: {date_}', expanded=False) as status:
+        #             st.pyplot(fig)
+        #             status.update(f'Magnetic Data: {date_} - Finished!')
+        # except Exception as e:
+        #     #print(f"An error occurred: {e}")
+        #     pass
 
     if all_data:
         fig_overlapped = plot_overlapped_timeseries(all_data, all_event_times)