update codegen to use literal unions for enums

rerun-io · May 22, 2024 · 970bcab · 970bcab
1 parent 25977ee
commit 970bcab
Show file tree

Hide file tree

Showing 8 changed files with 41 additions and 13 deletions.
diff --git a/crates/re_types_builder/src/codegen/python/mod.rs b/crates/re_types_builder/src/codegen/python/mod.rs
@@ -847,7 +847,12 @@ fn code_for_enum(
         }
     }
 
-    code.push_unindented(format!("{name}Like = Union[{name}, str]"), 1);
+    let variants = obj
+        .fields
+        .iter()
+        .map(|v| format!("Literal[{:?}]", v.snake_case_name()))
+        .join(" | ");
+    code.push_unindented(format!("{name}Like = Union[{name}, {variants}]"), 1);
     code.push_unindented(
         format!(
             r#"

diff --git a/rerun_py/rerun_sdk/rerun/blueprint/__init__.py b/rerun_py/rerun_sdk/rerun/blueprint/__init__.py
diff --git a/rerun_py/rerun_sdk/rerun/blueprint/components/background_kind.py b/rerun_py/rerun_sdk/rerun/blueprint/components/background_kind.py
diff --git a/rerun_py/rerun_sdk/rerun/blueprint/components/container_kind.py b/rerun_py/rerun_sdk/rerun/blueprint/components/container_kind.py
diff --git a/rerun_py/rerun_sdk/rerun/blueprint/components/corner2d.py b/rerun_py/rerun_sdk/rerun/blueprint/components/corner2d.py
diff --git a/rerun_py/rerun_sdk/rerun/blueprint/components/panel_state.py b/rerun_py/rerun_sdk/rerun/blueprint/components/panel_state.py
diff --git a/rerun_py/rerun_sdk/rerun/components/marker_shape.py b/rerun_py/rerun_sdk/rerun/components/marker_shape.py
diff --git a/rerun_py/tests/test_types/components/enum_test.py b/rerun_py/tests/test_types/components/enum_test.py