Merge pull request #20 from EnergieID/feat/allow-multiple-forcast

Support multiple different forecasts

Author: jorg-vr

.env.example

@@ -4,6 +4,9 @@ ENERGY_ID_COFY_API_TOKEN=foo
 # --- Database --------------------------------------------------------------
 DB_URL=postgresql+asyncpg://cofy:cofy@localhost:5432/epv
 
+# --- Forecasts -------------------------------------------------------------
+FORECASTS={"ACC_1": 42923}
+
 # dev only, not needed for production, but useful for local development with Docker
 POSTGRES_DB=epv
 POSTGRES_USER=cofy

main.py

@@ -10,6 +10,7 @@
   3. `poe dev` to start the dev server (auto-reloads, reads .env)
 """
 
+import json
 from os import environ
 
 from cofy import CofyAPI
@@ -32,14 +33,18 @@
 # Each module exposes its own set of API routes under the name you choose.
 # Browse the available modules:  https://github.com/EnergieID/cofy-api
 
-cofy.register_module(
-    DirectiveModule(
-        source=DirectiveSource(
-            source=DBSource(
-                db_url=environ.get("DB_URL", "postgresql+asyncpg://cofy:cofy@localhost:5432/epv"),
-                itemid=42923,
+forecasts = json.loads(environ.get("FORECASTS", "{}"))
+for name, id in forecasts.items():
+    cofy.register_module(
+        DirectiveModule(
+            source=DirectiveSource(
+                source=DBSource(
+                    db_url=environ.get("DB_URL", "postgresql+asyncpg://cofy:cofy@localhost:5432/epv"),
+                    itemid=id,
+                ),
+                boundaries=(-100000, 0, 100000, 500000),
             ),
-            boundaries=(-100000, 0, 100000, 500000),
-        ),
+            name=name,
+            description=f"Directive for community {name}",
+        )
     )
-)