this fixes #527

allow enums and states to be passed to `get_triggers` as well

Author: aleneum

Changelog.md

@@ -6,6 +6,7 @@
 - Feature #517: Introduce `try/except` for finalize callbacks in `Machine` and `HierachicalMachine`. Thus, errors occurring in finalize callbacks will be suppressed and only the original error will be raised.
 - Feature #520: Show references in graphs and markup. Introduce `MarkupMachine.format_references` to tweak reference formatting (thanks @StephenCarboni)
 - Feature #485: Introduce `Machine.on_exception` to handle raised exceptions in callbacks (thanks @thedrow)
+- Feature #527: `Machine.get_triggers` now supports `State` and `Enum` as arguments (thanks @luup2k)
 
 ## 0.8.7 (February 2021)
 

tests/test_core.py

@@ -871,7 +871,11 @@ def test_get_triggers(self):
         self.assertEqual(len(machine.get_triggers('B')), 0)
         self.assertEqual(len(machine.get_triggers('C')), 1)
         # self stuff machine should have to-transitions to every state
-        self.assertEqual(len(self.stuff.machine.get_triggers('B')), len(self.stuff.machine.states))
+        m = self.stuff.machine
+        self.assertEqual(len(m.get_triggers('B')), len(m.states))
+        trigger_name = m.get_triggers('B')
+        trigger_state = m.get_triggers(m.states['B'])
+        self.assertEqual(trigger_name, trigger_state)
 
     def test_skip_override(self):
         local_mock = MagicMock()

tests/test_enum.py

@@ -148,6 +148,12 @@ def test_get_transitions(self):
         # we expect two return values. 'switch_to_yellow' and 'to_YELLOW'
         self.assertEqual(2, len(m.get_transitions(source=self.States.RED, dest=self.States.YELLOW)))
 
+    def test_get_triggers(self):
+        m = self.machine_cls(states=self.States, initial=self.States.RED)
+        trigger_name = m.get_triggers(m.state.name)
+        trigger_enum = m.get_triggers(m.state)
+        self.assertEqual(trigger_enum, trigger_name)
+
 
 @skipIf(enum is None, "enum is not available")
 class TestNestedStateEnums(TestEnumsAsStates):

transitions/core.py

@@ -890,8 +890,8 @@ def get_triggers(self, *args):
         Returns:
             list of transition/trigger names.
         """
-        states = set(args)
-        return [t for (t, ev) in self.events.items() if any(state in ev.transitions for state in states)]
+        names = set([state.name if hasattr(state, 'name') else state for state in args])
+        return [t for (t, ev) in self.events.items() if any(name in ev.transitions for name in names)]
 
     def add_transition(self, trigger, source, dest, conditions=None,
                        unless=None, before=None, after=None, prepare=None, **kwargs):

transitions/extensions/nesting.py

@@ -710,7 +710,8 @@ def get_triggers(self, *args):
         """ Extends transitions.core.Machine.get_triggers to also include parent state triggers. """
         triggers = []
         with self():
-            for state_name in args:
+            for state in args:
+                state_name = state.name if hasattr(state, 'name') else state
                 state_path = state_name.split(self.state_cls.separator)
                 if len(state_path) > 1:  # we only need to check substates when 'state_name' refers to a substate
                     with self(state_path[0]):