Fix generated stateMixin class to properly implement overridable transition methods instead of messing with getattr; allows use of `super().transition_name()` in classes that subclass from the Mixin

1 files changed, 2 insertions, 2 deletions

diff --git a/examples/statemachine/libraryBookDemo.py b/examples/statemachine/libraryBookDemo.py
index 95e5996..5e73f07 100644
--- a/examples/statemachine/libraryBookDemo.py
+++ b/examples/statemachine/libraryBookDemo.py
@@ -24,9 +24,9 @@ class RestrictedBook(Book):
     # specialized checkout to check permission of user first
     def checkout(self, user=None):
         if user in self._authorized_users:
-            self._state = self._state.checkout()
+            super().checkout()
         else:
-            raise Exception("{0} could not check out restricted book".format((user, "anonymous")[user is None]))
+            raise Exception("{0} could not check out restricted book".format(user if user is not None else "anonymous"))
 
 
 def run_demo():