Add on_error state draw

README.md

@@ -324,7 +324,7 @@ Changelog
 
 <img src="https://f.cloud.github.com/assets/41479/2227946/a9e77760-9ad0-11e3-804f-301d075470fe.png" alt="django-fsm" width="100px"/>
 
-### django-fsm GIT
+### django-fsm 2.2.0 2014-09-03
 * Support for [class substitution](http://schinckel.net/2013/06/13/django-proxy-model-state-machine/) to proxy classes depending on the state
 * Added ConcurrentTransitionMixin with optimistic locking support
 * Default db_index=True for FSMIntegerField removed

django_fsm/management/commands/graph_transitions.py

@@ -32,13 +32,17 @@ def generate_dot(fields_data):
                     target_name = node_name(field, transition.target)
                     sources.add((source_name, transition.source))
                     targets.add((target_name, transition.target))
-                    edges.add((source_name, target_name))
+                    edges.add((source_name, target_name, ()))
+            if transition.on_error:
+                on_error_name = node_name(field, transition.on_error)
+                targets.add((on_error_name, transition.on_error))
+                edges.add((source_name, on_error_name, (('style', 'dotted'),)))
 
         for target in any_targets:
             target_name = node_name(field, target)
             targets.add((target_name, target))
             for source_name, label in sources:
-                edges.add((source_name, target_name))
+                edges.add((source_name, target_name, ()))
 
         # construct subgraph
         opts = field.model._meta
@@ -48,8 +52,8 @@ def generate_dot(fields_data):
 
         for name, label in sources | targets:
             subgraph.node(name, label=label)
-        for source_name, target_name in edges:
-            subgraph.edge(source_name, target_name)
+        for source_name, target_name, attrs in edges:
+            subgraph.edge(source_name, target_name, **dict(attrs))
 
         result.subgraph(subgraph)
 

setup.py

@@ -7,7 +7,7 @@
 
 setup(
     name='django-fsm',
-    version='2.1.0',
+    version='2.2.0',
     description='Django friendly finite state machine support.',
     author='Mikhail Podgurskiy',
     author_email='kmmbvnr@gmail.com',

tests/testapp/models.py

@@ -85,20 +85,20 @@ class BlogPost(models.Model):
     state = FSMField(default='new', protected=True)
 
     @transition(field=state, source='new', target='published',
-                permission='testapp.can_publish_post')
+                on_error='failed', permission='testapp.can_publish_post')
     def publish(self):
         pass
 
     @transition(field=state, source='published')
     def notify_all(self):
         pass
 
-    @transition(field=state, source='published', target='hidden')
+    @transition(field=state, source='published', target='hidden', on_error='failed',)
     def hide(self):
         pass
 
     @transition(field=state, source='new', target='removed',
-                permission=lambda u: u.has_perm('testapp.can_remove_post'))
+                on_error='failed', permission=lambda u: u.has_perm('testapp.can_remove_post'))
     def remove(self):
         raise Exception('No rights to delete %s' % self)