✨Add daily forecast data API endpoint

Author: Ilya Lityuga

app.py

@@ -6,7 +6,12 @@
 import starlette.status as status
 from immanuel import charts
 from immanuel.const import chart, names
-from utils import retrograde_periods, weekly_forecast_data, yearly_forecast_data
+from utils import (
+    retrograde_periods,
+    daily_forecast_data,
+    weekly_forecast_data,
+    yearly_forecast_data,
+)
 
 tags_metadata = [
     {
@@ -17,6 +22,10 @@
         "name": "retrograde_calendar",
         "description": "Retrograde calendar for a given year",
     },
+    {
+        "name": "get_daily_forecast_data",
+        "description": "Get daily forecast data - planet positions, major aspects, moon phase etc - for a given date",
+    },
     {
         "name": "get_weekly_forecast_data",
         "description": "Get weekly forecast data - planet positions, major aspects, retrograde etc - for a given date",
@@ -427,6 +436,22 @@ def solar_returns(year: int, month: int, day: int, hour: int, lat: float, lon: f
     return charts.SolarReturns(native)
 
 
+@app.get("/get_daily_forecast_data", tags=["get_daily_forecast_data"])
+def get_daily_forecast_data(
+    start_date: Annotated[
+        date,
+        Query(
+            title="date",
+            description="Input date for the forecast",
+            examples=[date.today()],
+        ),
+    ],
+):
+    datetime_obj = datetime.combine(start_date, time.min)
+    dfd = daily_forecast_data(datetime_obj)
+    return {"success": 1, "data": dfd}
+
+
 @app.get("/get_weekly_forecast_data", tags=["get_weekly_forecast_data"])
 def get_weekly_forecast_data(
     start_date: Annotated[

test_app.py

@@ -24,7 +24,25 @@ def test_retrograde_calendar():
     assert response.status_code == 200
     assert response_json["success"] == 1
     assert response_json["data"][0]["planet"] == "Mercury"
-    assert len(response_json["data"][0]["periods"]) == 6
+    assert len(response_json["data"][0]["periods"]) == 7
+
+
+def test_get_daily_forecast_data():
+    response = client.get(
+        "/get_daily_forecast_data?start_date=2024-03-19",
+        headers={"X-Token": "coneofsilence"},
+    )
+    response_json = response.json()
+    assert response.status_code == 200
+    assert response_json["success"] == 1
+    assert len(response_json["data"]) == 3
+    assert len(response_json["data"]["planets"]) == 8
+    assert "moon_phase" in response_json["data"]
+    assert "planets" in response_json["data"]
+    assert "aspects" in response_json["data"]
+    assert "sign" in response_json["data"]["planets"]["Mercury"]
+    assert "house" in response_json["data"]["planets"]["Mercury"]
+    assert "movement" in response_json["data"]["planets"]["Mercury"]
 
 
 def test_get_weekly_forecast_data():
@@ -48,6 +66,6 @@ def test_get_yearly_forecast_data():
     assert response.status_code == 200
     assert response_json["success"] == 1
     assert len(response_json["data"]) == 8
-    assert len(response_json["data"]["Mercury"]) == 2
+    assert len(response_json["data"]["Mercury"]) == 3
     assert "sign" in response_json["data"]["Mercury"]
     assert "movement" in response_json["data"]["Mercury"]

utils.py

@@ -88,6 +88,11 @@ def retrograde_periods(n, lat, lon):
     return retro_table
 
 
+def daily_forecast_data(start_date):
+    settings.set({"objects": planets})
+    return day_forecast(start_date)
+
+
 def weekly_forecast_data(start_date):
     settings.set({"objects": planets})
 
@@ -96,41 +101,44 @@ def weekly_forecast_data(start_date):
 
     for i in range(7):  # for each day of week
         date = start_date + timedelta(days=i)
+        weekly_data[date.strftime("%Y-%m-%d")] = day_forecast(date)
 
-        native = charts.Subject(date_time=date, latitude=0.0, longitude=0.0)
-        natal = charts.Natal(native)
+    return weekly_data
 
-        planet_positions = {}
-        for object in natal.objects.values():
-            planet_positions[object.name] = {
-                "sign": object.sign.name,
-                "house": object.house.number,
-                "movement": object.movement.formatted,
-            }
-
-        planet_aspects = []
-        aspects_set = set()
-        for index, aspects in natal.aspects.items():
-            for aspect in aspects.values():
-                aspect_key = aspect._active_name + " " + aspect._passive_name
-                if aspect_key not in aspects_set:
-                    planet_aspects.append(
-                        {
-                            "active": aspect._active_name,
-                            "passive": aspect._passive_name,
-                            "aspect": aspect.aspect,
-                            "type": aspect.type,
-                        }
-                    )
-                    aspects_set.add(aspect_key)
 
-        weekly_data[date.strftime("%Y-%m-%d")] = {
-            "moon_phase": natal.moon_phase.formatted,
-            "planets": planet_positions,
-            "aspects": planet_aspects,
+def day_forecast(date):
+    native = charts.Subject(date_time=date, latitude=0.0, longitude=0.0)
+    natal = charts.Natal(native)
+
+    planet_positions = {}
+    for object in natal.objects.values():
+        planet_positions[object.name] = {
+            "sign": object.sign.name,
+            "house": object.house.number,
+            "movement": object.movement.formatted,
         }
 
-    return weekly_data
+    planet_aspects = []
+    aspects_set = set()
+    for index, aspects in natal.aspects.items():
+        for aspect in aspects.values():
+            aspect_key = aspect._active_name + " " + aspect._passive_name
+            if aspect_key not in aspects_set:
+                planet_aspects.append(
+                    {
+                        "active": aspect._active_name,
+                        "passive": aspect._passive_name,
+                        "aspect": aspect.aspect,
+                        "type": aspect.type,
+                    }
+                )
+                aspects_set.add(aspect_key)
+
+    return {
+        "moon_phase": natal.moon_phase.formatted,
+        "planets": planet_positions,
+        "aspects": planet_aspects,
+    }
 
 
 def yearly_forecast_data(start_date):