feat: add strum EnumVariantNames macro to CoreNode enum

pywr · Dec 11, 2023 · 97e69e9 · 97e69e9
1 parent 767cfe8
commit 97e69e9
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/pywr_schema/Cargo.toml b/pywr_schema/Cargo.toml
@@ -15,3 +15,5 @@ serde_json = "1.0"
 time = { version = "0.3.14", features = ["serde", "serde-well-known", "serde-human-readable"] }
 thiserror = "1.0.44"
 pywr-schema-macros = { path = "../pywr-schema-macros" }
+strum = "0.25"
+strum_macros = "0.25"
diff --git a/pywr_schema/src/nodes/mod.rs b/pywr_schema/src/nodes/mod.rs
@@ -29,6 +29,7 @@ pub use virtual_storage::{
     AnnualVirtualStorageNode, MonthlyVirtualStorageNode, RollingVirtualStorageNode,
     SeasonalVirtualStorageNode, VirtualStorageNode,
 };
+use strum_macros::EnumVariantNames;
 
 #[derive(serde::Deserialize, serde::Serialize, Debug, Clone)]
 pub struct NodePosition {
@@ -57,7 +58,7 @@ pub struct CustomNode {
     pub attributes: HashMap<String, Value>,
 }
 
-#[derive(serde::Deserialize, serde::Serialize, Clone)]
+#[derive(serde::Deserialize, serde::Serialize, EnumVariantNames, Clone)]
 #[serde(tag = "type")]
 pub enum CoreNode {
     #[serde(alias = "input")]