windborne 1.2.5__tar.gz → 1.2.7__tar.gz

{windborne-1.2.5 → windborne-1.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: windborne
-Version: 1.2.5
+Version: 1.2.7
 Summary: A Python library for interacting with WindBorne Data and Forecasts API
 Author-email: WindBorne Systems <data@windbornesystems.com>
 Classifier: Programming Language :: Python :: 3

{windborne-1.2.5 → windborne-1.2.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "windborne"
-version = "1.2.5"
+version = "1.2.7"
 description = "A Python library for interacting with WindBorne Data and Forecasts API"
 readme = {file = "README.md", content-type = "text/markdown"}
 authors = [

{windborne-1.2.5 → windborne-1.2.7}/windborne/__init__.py

@@ -22,6 +22,7 @@ from .data_api import (
 from .forecasts_api import (
     get_point_forecasts,
     get_initialization_times,
+    get_historical_initialization_times,
     get_forecast_hours,
     get_generation_times,
 
@@ -41,7 +42,8 @@ from .forecasts_api import (
 
     get_tropical_cyclones,
 
-    get_population_weighted_hdd
+    get_population_weighted_hdd,
+    get_population_weighted_cdd
 )
 
 # Define what should be available when users import *
@@ -63,6 +65,7 @@ __all__ = [
 
     "get_point_forecasts",
     "get_initialization_times",
+    "get_historical_initialization_times",
     "get_forecast_hours",
     "get_generation_times",
     
@@ -86,5 +89,6 @@ __all__ = [
     "get_historical_500hpa_wind_v",
     "get_tropical_cyclones",
 
-    "get_population_weighted_hdd"
+    "get_population_weighted_hdd",
+    "get_population_weighted_cdd"
 ]

{windborne-1.2.5 → windborne-1.2.7}/windborne/cli.py

@@ -19,12 +19,14 @@ from . import (
 
     get_point_forecasts,
     get_initialization_times,
+    get_historical_initialization_times,
     get_forecast_hours,
     get_generation_times,
     get_full_gridded_forecast,
     get_gridded_forecast,
     get_tropical_cyclones,
-    get_population_weighted_hdd
+    get_population_weighted_hdd,
+    get_population_weighted_cdd
 
 )
 
@@ -169,6 +171,8 @@ def main():
         'dewpoint_2m',
         'wind_u_10m',
         'wind_v_10m',
+        'wind_u_100m',
+        'wind_v_100m',
         'pressure_msl',
         '500/temperature',
         '500/wind_u',
@@ -235,11 +239,24 @@ def main():
     hdd_parser.add_argument('-m', '--external-model', help='External model (eg gfs, ifs, hrrr, aifs)')
     hdd_parser.add_argument('-o', '--output', help='Output file (supports .csv and .json formats)')
 
+    # Population Weighted CDD Command
+    cdd_parser = subparsers.add_parser('cdd', help='Get population weighted cooling degree days (CDD) forecasts')
+    cdd_parser.add_argument('initialization_time', help='Initialization time (YYYYMMDDHH, YYYY-MM-DDTHH, or YYYY-MM-DDTHH:mm:ss)')
+    cdd_parser.add_argument('-i', '--intracycle', action='store_true', help='Use the intracycle forecast')
+    cdd_parser.add_argument('-e', '--ens-member', help='Ensemble member (eg 1 or mean)')
+    cdd_parser.add_argument('-m', '--external-model', help='External model (eg gfs, ifs, hrrr, aifs)')
+    cdd_parser.add_argument('-o', '--output', help='Output file (supports .csv and .json formats)')
+
     # Initialization Times Command
     initialization_times_parser = subparsers.add_parser('init_times', help='Get available initialization times for point forecasts')
     initialization_times_parser.add_argument('-i', '--intracycle', action='store_true', help='Use the intracycle forecast')
     initialization_times_parser.add_argument('-e', '--ens-member', help='Ensemble member (eg 1 or mean)')
 
+    # Historical Initialization Times Command
+    hist_initialization_times_parser = subparsers.add_parser('hist_init_times', help='Get available historical initialization times')
+    hist_initialization_times_parser.add_argument('-i', '--intracycle', action='store_true', help='Use the intracycle forecast')
+    hist_initialization_times_parser.add_argument('-e', '--ens-member', help='Ensemble member (eg 1 or mean)')
+
     # Forecast Hours Command
     forecast_hours_parser = subparsers.add_parser('forecast_hours', help='Get available forecast hours for WeatherMesh')
     forecast_hours_parser.add_argument('-i', '--intracycle', action='store_true', help='Use the intracycle forecast')
@@ -449,6 +466,9 @@ def main():
     elif args.command == 'init_times':
         get_initialization_times(print_response=True, ensemble_member=args.ens_member, intracycle=args.intracycle)
 
+    elif args.command == 'hist_init_times':
+        get_historical_initialization_times(print_response=True, ensemble_member=args.ens_member, intracycle=args.intracycle)
+
     elif args.command == 'forecast_hours':
         get_forecast_hours(print_response=True, ensemble_member=args.ens_member, intracycle=args.intracycle)
 
@@ -547,6 +567,17 @@ def main():
             print_response=(not args.output)
         )
 
+    elif args.command == 'cdd':
+        # Handle population weighted CDD
+        get_population_weighted_cdd(
+            initialization_time=args.initialization_time, 
+            intracycle=args.intracycle,
+            ens_member=args.ens_member,
+            external_model=args.external_model,
+            output_file=args.output,
+            print_response=(not args.output)
+        )
+
     else:
         parser.print_help()
 

{windborne-1.2.5 → windborne-1.2.7}/windborne/forecasts_api.py

@@ -301,6 +301,23 @@ def get_initialization_times(print_response=False, ensemble_member=None, intracy
     return response
 
 
+def get_historical_initialization_times(print_response=False, ensemble_member=None, intracycle=False):
+    """
+    Get historical initialization times for forecasts from our archive
+    These may be higher latency to fetch and cannot be used for custom point forecasting
+    """
+    params = {
+        'ens_member': ensemble_member,
+        'intracycle': intracycle
+    }
+    response = make_api_request(f"{FORECASTS_API_BASE_URL}/historical_initialization_times.json", params=params)
+
+    if print_response:
+        print("Available historical initialization times:")
+        for time in response:
+            print(f" - {time}")
+
+
 def get_forecast_hours(print_response=False, ensemble_member=None, intracycle=False):
     """
     Get available forecast hours for WeatherMesh
@@ -427,3 +444,41 @@ def get_population_weighted_hdd(initialization_time, intracycle=False, ens_membe
                 print(f"  {dates[i]}: {response['hdd'][region][dates[i]]}")
     
     return response
+
+def get_population_weighted_cdd(initialization_time, intracycle=False, ens_member=None, external_model=None, output_file=None, print_response=False):
+    """
+    Get population weighted CDD data from the API.
+    """
+    params = {
+        "initialization_time": initialization_time,
+        "intracycle": intracycle,
+        "ens_member": ens_member,
+        "external_model": external_model
+    }
+    response = make_api_request(f"{FORECASTS_API_BASE_URL}/cdd", params=params, as_json=True)
+    
+    if output_file:
+        if output_file.endswith('.csv'):
+            import csv
+
+            # save as csv, with a row for each region, and a column for each date, sorted alphabetically by region
+            regions = sorted(response['cdd'].keys())
+            dates = response['dates']
+            data = [[response['cdd'][region][dates[i]] for region in regions] for i in range(len(dates))]  
+
+            with open(output_file, 'w') as f:
+                writer = csv.writer(f)
+                writer.writerow(['Region'] + dates)
+
+                for region in regions:
+                    writer.writerow([region] + [response['cdd'][region][date] for date in dates])
+    
+    if print_response:
+        dates = response['dates']
+        print(response['cdd']['Alabama'])
+        for region in sorted(response['cdd'].keys()):
+            print(f"{region}:")
+            for i in range(len(dates)):
+                print(f"  {dates[i]}: {response['cdd'][region][dates[i]]}")
+    
+    return response

{windborne-1.2.5 → windborne-1.2.7}/windborne.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: windborne
-Version: 1.2.5
+Version: 1.2.7
 Summary: A Python library for interacting with WindBorne Data and Forecasts API
 Author-email: WindBorne Systems <data@windbornesystems.com>
 Classifier: Programming Language :: Python :: 3