Add a kind item to allow differentiating dora and external events

dora-rs · Aug 22, 2023 · bc7bfd5 · bc7bfd5
1 parent bd875f6
commit bc7bfd5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/apis/python/operator/src/lib.rs b/apis/python/operator/src/lib.rs
@@ -19,6 +19,13 @@ pub struct PyEvent {
 #[pymethods]
 impl PyEvent {
     pub fn __getitem__(&self, key: &str, py: Python<'_>) -> PyResult<Option<PyObject>> {
+        if key == "kind" {
+            let kind = match &self.event {
+                MergedEvent::Dora(_) => "dora",
+                MergedEvent::External(_) => "external",
+            };
+            return Ok(Some(kind.to_object(py)));
+        }
         match &self.event {
             MergedEvent::Dora(event) => {
                 let value = match key {