From 07854a236662de1a2b382d1b162d6f53b3c5241d Mon Sep 17 00:00:00 2001
From: Joshua Harlow <harlowja@yahoo-inc.com>
Date: Thu, 1 Aug 2013 18:44:38 -0700
Subject: [PATCH] Add a more complicated graph example.

Add an example that creates tasks that will
build a car and has the dependencies specified
to build the car in the right order and have each
task be submitted to the flow and then run and
watch the tasks occur in the right order automatically.

Change-Id: I17a2f621dfd7ceefb0a9c863e3e68b21b92672c2
---
 docs/examples/complex_graph.py | 152 +++++++++++++++++++++++++++++++++
 1 file changed, 152 insertions(+)
 create mode 100644 docs/examples/complex_graph.py

diff --git a/docs/examples/complex_graph.py b/docs/examples/complex_graph.py
new file mode 100644
index 00000000..9e66ef9f
--- /dev/null
+++ b/docs/examples/complex_graph.py
@@ -0,0 +1,152 @@
+import logging
+import os
+import sys
+
+logging.basicConfig(level=logging.ERROR)
+
+my_dir_path = os.path.dirname(os.path.abspath(__file__))
+sys.path.insert(0, os.path.join(os.path.join(my_dir_path, os.pardir),
+                                os.pardir))
+
+from taskflow import decorators
+from taskflow.patterns import graph_flow as gf
+from taskflow import utils
+
+
+def flow_notify(state, details):
+    print("'%s' entered state: %s" % (details['flow'], state))
+
+
+def task_notify(state, details):
+    print("'%s' entered state: %s" % (details['runner'], state))
+
+
+@decorators.task(provides=['spec'])
+def build_spec(context):
+    params = context['inputs']
+    verified = {}
+    for k, v in params.items():
+        verified[k] = int(v)
+    return {
+        'spec': verified,
+    }
+
+
+@decorators.task(provides=['frame'])
+def build_frame(context, spec):
+    return {
+        'frame': 'steel',
+    }
+
+
+@decorators.task(provides=['engine'])
+def build_engine(context, spec):
+    return {
+        'engine': 'honda',
+    }
+
+
+@decorators.task(provides=['doors'])
+def build_doors(context, spec):
+    return {
+        'doors': '2',
+    }
+
+
+@decorators.task(provides=['wheels'])
+def build_wheels(context, spec):
+    return {
+        'wheels': '4',
+    }
+
+
+@decorators.task(provides=['wheels'])
+def build_windows(context, spec):
+    return {
+        'windows': '4',
+    }
+
+
+@decorators.task(provides=['engine_installed'])
+def install_engine(context, frame, engine):
+    return {
+        'engine_installed': True,
+    }
+
+
+@decorators.task
+def install_doors(context, frame, windows_installed, doors):
+    pass
+
+
+@decorators.task(provides=['windows_installed'])
+def install_windows(context, frame, doors):
+    return {
+        'windows_installed': True,
+    }
+
+
+@decorators.task
+def install_wheels(context, frame, engine, engine_installed, wheels):
+    pass
+
+
+def trash(context, result, cause):
+    print("Throwing away pieces of car!")
+
+
+@decorators.task(revert_with=trash)
+def startup(context, **kwargs):
+    pass
+    # TODO(harlowja): try triggering reversion here!
+    # raise ValueError("Car not verified")
+    return {
+        'ran': True,
+    }
+
+
+flow_notifier = utils.TransitionNotifier()
+flow_notifier.register('*', flow_notify)
+
+task_notifier = utils.TransitionNotifier()
+task_notifier.register('*', task_notify)
+
+flow = gf.Flow("make-auto")
+flow.notifier = flow_notifier
+flow.task_notifier = task_notifier
+
+# Lets build a car!!
+flow.add(build_spec)
+flow.add(build_frame)
+flow.add(build_engine)
+flow.add(build_doors)
+flow.add(build_wheels)
+i_uuid1 = flow.add(install_engine)
+i_uuid2 = flow.add(install_doors)
+i_uuid3 = flow.add(install_windows)
+i_uuid4 = flow.add(install_wheels)
+install_uuids = [i_uuid1, i_uuid2, i_uuid3, i_uuid4]
+
+# Lets add a manual dependency that startup needs all the installation to
+# complete, this could be done automatically but lets now instead ;)
+startup_uuid = flow.add(startup)
+for i_uuid in install_uuids:
+    flow.add_dependency(i_uuid, startup_uuid)
+
+# Now begin the build!
+context = {
+    "inputs": {
+        'engine': 123,
+        'tire': '234',
+    }
+}
+print '-' * 7
+print 'Running'
+print '-' * 7
+flow.run(context)
+
+print '-' * 11
+print 'All results'
+print '-' * 11
+for (uuid, v) in flow.results.items():
+    print '%s => %s' % (uuid, v)