Support assets

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

pyproject.toml

@@ -22,7 +22,6 @@ classifiers = [
 ]
 requires-python = ">= 3.11, < 4"
 dependencies = [
-  "frequenz-client-microgrid @ git+https://github.com/frequenz-floss/frequenz-client-microgrid-python@refs/heads/v0.18.x",
   "typing-extensions >= 4.14.1, < 5",
 ]
 dynamic = ["version"]
@@ -32,6 +31,12 @@ name = "Frequenz Energy-as-a-Service GmbH"
 email = "[email protected]"
 
 [project.optional-dependencies]
+microgrid = [
+  "frequenz-client-microgrid @ git+https://github.com/frequenz-floss/frequenz-client-microgrid-python@refs/heads/v0.18.x",
+]
+assets = [
+  "frequenz-client-assets == 0.1.0"
+]
 dev-flake8 = [
   "flake8 == 7.3.0",
   "flake8-docstrings == 1.7.0",

src/category.rs

@@ -18,19 +18,35 @@ struct ComponentClasses<'py> {
 
 impl<'py> ComponentClasses<'py> {
     fn try_new(py: Python<'py>) -> PyResult<Self> {
-        let path = "frequenz.client.microgrid.component";
+        let candidates = vec![
+            "frequenz.client.microgrid.component".to_string(),
+            "frequenz.client.assets.electrical_component".to_string(),
+        ];
 
-        Ok(Self {
-            grid_connection_point: py.import(path)?.getattr("GridConnectionPoint")?,
-            meter: py.import(path)?.getattr("Meter")?,
-            battery: py.import(path)?.getattr("Battery")?,
-            ev_charger: py.import(path)?.getattr("EvCharger")?,
-            chp: py.import(path)?.getattr("Chp")?,
-            battery_inverter: py.import(path)?.getattr("BatteryInverter")?,
-            solar_inverter: py.import(path)?.getattr("SolarInverter")?,
-            hybrid_inverter: py.import(path)?.getattr("HybridInverter")?,
-            unspecified_component: py.import(path)?.getattr("UnspecifiedComponent")?,
-        })
+        let mut last_err: Option<PyErr> = None;
+        for path in &candidates {
+            match py.import(path) {
+                Ok(module) => {
+                    return Ok(Self {
+                        grid_connection_point: module.getattr("GridConnectionPoint")?,
+                        meter: module.getattr("Meter")?,
+                        battery: module.getattr("Battery")?,
+                        ev_charger: module.getattr("EvCharger")?,
+                        chp: module.getattr("Chp")?,
+                        battery_inverter: module.getattr("BatteryInverter")?,
+                        solar_inverter: module.getattr("SolarInverter")?,
+                        hybrid_inverter: module.getattr("HybridInverter")?,
+                        unspecified_component: module.getattr("UnspecifiedComponent")?,
+                    });
+                }
+                Err(e) => last_err = Some(e),
+            }
+        }
+        Err(pyo3::exceptions::PyImportError::new_err(format!(
+            "Could not import a component provider. Tried: {candidates:?}. \
+            Install one: pip install frequenz-component-graph[microgrid] or [assets]. \
+            Last error: {last_err:?}"
+        )))
     }
 }