imsciences 0.6.0.5__py3-none-any.whl → 0.6.0.7__py3-none-any.whl

imsciences/datafunctions.py

@@ -2108,11 +2108,17 @@ class datapull:
         return df_final_combined
     
     def pull_weather(self, week_commencing, country) -> pd.DataFrame:
+        import pandas as pd
+        import urllib.request
+        from datetime import datetime
+        import requests
+        from geopy.geocoders import Nominatim
+
         # Week commencing dictionary
         day_dict = {"mon": 0, "tue": 1, "wed": 2, "thur": 3, "fri": 4, "sat": 5, "sun": 6}
 
         # Country dictionary
-        country_dict = {"AUS": "AU__ASOS", "GBR": "GB__ASOS", "USA": "USCRN", "DEU": "DE__ASOS", "CAN": "Canada"}
+        country_dict = {"AUS": "AU__ASOS", "GBR": "GB__ASOS", "USA": "USCRN", "DEU": "DE__ASOS", "CAN": "Canada", "ZAF": "ZA__ASOS"}
 
         # Function to flatten a list of nested lists into a list
         def flatten_list(nested_list):
@@ -2180,8 +2186,17 @@ class datapull:
                                     "&stations=CWIL", "&stations=CXWB", "&stations=CYZS",
                                     "&stations=CWJC", "&stations=CYFB", "&stations=CWUW"])
 
+        elif country == "ZA__ASOS":
+            cities = ["Johannesburg", "Cape Town", "Durban", "Pretoria"]
+            stations = []
+
+            for city in cities:
+                geolocator = Nominatim(user_agent="MyApp")
+                location = geolocator.geocode(city)
+                stations.append(f"&latitude={location.latitude}&longitude={location.longitude}")
+
         # Temperature
-        if country == "GB__ASOS" or country == "AU__ASOS" or country == "DE__ASOS" or country == "FR__ASOS":
+        if country in ["GB__ASOS", "AU__ASOS", "DE__ASOS", "FR__ASOS"]:
             # We start by making a data frame of the following weather stations
             station_query = ''.join(stations)
 
@@ -2290,6 +2305,76 @@ class datapull:
                                                 "min_temp_c": "avg_min_temp_c",
                                                 "mean_temp_c": "avg_mean_temp_c",
                                                 "precip_in": "avg_mean_perc"}, inplace=True)
+        elif country == "ZA__ASOS":
+            weather_data_list = []
+
+            for city in cities:
+                geolocator = Nominatim(user_agent="MyApp")
+                location = geolocator.geocode(city)
+                url = "https://archive-api.open-meteo.com/v1/archive"
+
+                params = {
+                    "latitude": location.latitude,
+                    "longitude": location.longitude,
+                    "start_date": formatted_date,
+                    "end_date": today.strftime("%Y-%m-%d"),
+                    "daily": "temperature_2m_max,temperature_2m_min,precipitation_sum",
+                    "timezone": "auto"
+                }
+
+                response = requests.get(url, params=params)
+                response_data = response.json()
+
+                daily_data = response_data["daily"]
+                dates = daily_data["time"]
+
+                data = pd.DataFrame({
+                    "day": dates,
+                    "max_temp_f": daily_data["temperature_2m_max"],
+                    "min_temp_f": daily_data["temperature_2m_min"],
+                    "precip_in": daily_data["precipitation_sum"]
+                })
+                data["city"] = city
+                weather_data_list.append(data)
+
+            weather = pd.concat(weather_data_list)
+
+            # Convert the date column to a Date type
+            weather["day"] = pd.to_datetime(weather["day"])
+
+            # Replace None values
+            weather["max_temp_f"].replace("None", 0, inplace=True)
+            weather["min_temp_f"].replace("None", 0, inplace=True)
+            weather["precip_in"].replace("None", 0, inplace=True)
+
+            weather[["max_temp_f", "min_temp_f", "precip_in"]] = weather[["max_temp_f", "min_temp_f", "precip_in"]].apply(pd.to_numeric)
+
+            # Estimate mean temperature
+            weather["mean_temp_f"] = (weather["max_temp_f"] + weather["min_temp_f"]) / 2
+
+            # Convert Fahrenheit to Celsius for max_temp_f
+            weather["max_temp_c"] = (weather["max_temp_f"] - 32) * 5 / 9
+
+            # Convert Fahrenheit to Celsius for min_temp_f
+            weather["min_temp_c"] = (weather["min_temp_f"] - 32) * 5 / 9
+
+            # Convert Fahrenheit to Celsius for mean_temp_f
+            weather["mean_temp_c"] = (weather["mean_temp_f"] - 32) * 5 / 9
+
+            # Determine the starting chosen day for each date
+            weather['week_starting'] = weather["day"].apply(lambda x: x - pd.Timedelta(days=(x.weekday() - day_dict[week_commencing]) % 7))
+
+            # Group by week_starting and summarize
+            numeric_columns = weather.select_dtypes(include='number').columns
+            weekly_avg_temp = weather.groupby("week_starting")[numeric_columns].mean()
+            weekly_avg_temp.rename(columns={"max_temp_f": "avg_max_temp_f",
+                                            "min_temp_f": "avg_min_temp_f",
+                                            "mean_temp_f": "avg_mean_temp_f",
+                                            "max_temp_c": "avg_max_temp_c",
+                                            "min_temp_c": "avg_min_temp_c",
+                                            "mean_temp_c": "avg_mean_temp_c",
+                                            "precip_in": "avg_mean_perc"}, inplace=True)
+
         else:
             # We start by making a data frame of the following weather stations
             station_query = ''.join(stations)
@@ -2563,8 +2648,57 @@ class datapull:
             # Change index to datetime
             weekly_avg_rain.index = pd.to_datetime(weekly_avg_rain.index)
 
+        elif country == "ZA__ASOS":
+            cities = ["Johannesburg", "Cape Town", "Durban", "Pretoria"]
+            start_date = formatted_date
+            end_date = today.strftime("%Y-%m-%d")
+
+            weather_data_list = []
+
+            for city in cities:
+                geolocator = Nominatim(user_agent="MyApp")
+                location = geolocator.geocode(city)
+                url = "https://archive-api.open-meteo.com/v1/archive"
+
+                params = {
+                    "latitude": location.latitude,
+                    "longitude": location.longitude,
+                    "start_date": start_date,
+                    "end_date": end_date,
+                    "daily": "precipitation_sum",
+                    "timezone": "auto"
+                }
+
+                response = requests.get(url, params=params)
+                response_data = response.json()
+
+                daily_data = response_data["daily"]["precipitation_sum"]
+                dates = response_data["daily"]["time"]
+
+                data = pd.DataFrame({"date": dates, "rainfall": daily_data})
+                data["city"] = city
+
+                weather_data_list.append(data)
+
+            # Combine all city data into a single data frame
+            all_weather_data = pd.concat(weather_data_list)
+
+            # Convert the date column to a Date type
+            all_weather_data["date"] = pd.to_datetime(all_weather_data["date"])
+
+            # Set week commencing col up
+            all_weather_data['week_starting'] = all_weather_data["date"].apply(lambda x: x - pd.Timedelta(days=(x.weekday() - day_dict[week_commencing]) % 7))
+
+            # Group by week_starting and summarize
+            numeric_columns = all_weather_data.select_dtypes(include='number').columns
+            weekly_avg_rain = all_weather_data.groupby("week_starting")[numeric_columns].mean()
+            weekly_avg_rain.rename(columns={"rainfall": "avg_rainfall"}, inplace=True)
+
+            # Change index to datetime
+            weekly_avg_rain.index = pd.to_datetime(weekly_avg_rain.index)
+
         # Merge the dataframes
-        if country == "AU__ASOS" or country == "DE__ASOS" or country == "FR__ASOS" or country == "GB__ASOS":
+        if country in ["AU__ASOS", "DE__ASOS", "FR__ASOS", "GB__ASOS", "ZA__ASOS"]:
             merged_df = weekly_avg_rain.merge(weekly_avg_temp, on="week_starting")
         else:
             merged_df = weekly_avg_temp
@@ -2575,3 +2709,4 @@ class datapull:
         final_weather = ims_proc.rename_cols(merged_df, 'seas_')
 
         return final_weather
+

{imsciences-0.6.0.5.dist-info → imsciences-0.6.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imsciences
-Version: 0.6.0.5
+Version: 0.6.0.7
 Summary: IMS Data Processing Package
 Author: IMS
 Author-email: cam@im-sciences.com

{imsciences-0.6.0.5.dist-info → imsciences-0.6.0.7.dist-info}/RECORD

@@ -2,13 +2,13 @@ dataprocessing/__init__.py,sha256=quSwsLs6IuLoA5Rzi0ZD40xZaQudwDteF7_ai9JfTPk,32
 dataprocessing/data-processing-functions.py,sha256=vE1vsZ8xOSbR9Bwlp9SWXwEHXQ0nFydwGkvzHXf2f1Y,41
 dataprocessing/datafunctions.py,sha256=vE1vsZ8xOSbR9Bwlp9SWXwEHXQ0nFydwGkvzHXf2f1Y,41
 imsciences/__init__.py,sha256=GIPbLmWc06sVcOySWwNvMNUr6XGOHqPLryFIWgtpHh8,78
-imsciences/datafunctions.py,sha256=v3dyZYr7Bo_tiqxiIxu9-wy43cwAkcCgySt3iKwhk4s,135389
+imsciences/datafunctions.py,sha256=9aL25wdMwxOsE5ee173NaRBCx6zQVuXI_ZGS_Hmf29w,141390
 imsciences/datapull.py,sha256=TPY0LDgOkcKTBk8OekbD0Grg5x0SomAK2dZ7MuT6X1E,19000
 imsciencesdataprocessing/__init__.py,sha256=quSwsLs6IuLoA5Rzi0ZD40xZaQudwDteF7_ai9JfTPk,32
 imsciencesdataprocessing/datafunctions.py,sha256=vE1vsZ8xOSbR9Bwlp9SWXwEHXQ0nFydwGkvzHXf2f1Y,41
 imsdataprocessing/__init__.py,sha256=quSwsLs6IuLoA5Rzi0ZD40xZaQudwDteF7_ai9JfTPk,32
 imsdataprocessing/datafunctions.py,sha256=vE1vsZ8xOSbR9Bwlp9SWXwEHXQ0nFydwGkvzHXf2f1Y,41
-imsciences-0.6.0.5.dist-info/METADATA,sha256=JdIOSMeD-hgHudpBJURanXcIeTLV5pUcqoB5yPOyZaw,854
-imsciences-0.6.0.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-imsciences-0.6.0.5.dist-info/top_level.txt,sha256=hsENS-AlDVRh8tQJ6-426iUQlla9bPcGc0-UlFF0_iU,11
-imsciences-0.6.0.5.dist-info/RECORD,,
+imsciences-0.6.0.7.dist-info/METADATA,sha256=-u5hFYdnRM5zE5Buit8tX-2zu-YbmUgoU-51jqA7wMo,854
+imsciences-0.6.0.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+imsciences-0.6.0.7.dist-info/top_level.txt,sha256=hsENS-AlDVRh8tQJ6-426iUQlla9bPcGc0-UlFF0_iU,11
+imsciences-0.6.0.7.dist-info/RECORD,,