From a5953e03493184b08fe3495bb73470c8c106fafc Mon Sep 17 00:00:00 2001
From: Joshua Harlow <harlowja@yahoo-inc.com>
Date: Mon, 7 Oct 2013 17:57:49 -0700
Subject: [PATCH] Add a example which uses a sqlite persistence layer

Add an example that will use a sqlite layer for
persisting flow and task information, and will use
the layer for future resumption.

Show how to output what a logbook contains as well.

Change-Id: If5f936fe100cb43ed857b99e3fae0c6e6baba5ed
---
 taskflow/examples/persistence_example.py | 122 +++++++++++++++++++++++
 1 file changed, 122 insertions(+)
 create mode 100644 taskflow/examples/persistence_example.py

diff --git a/taskflow/examples/persistence_example.py b/taskflow/examples/persistence_example.py
new file mode 100644
index 00000000..760cf869
--- /dev/null
+++ b/taskflow/examples/persistence_example.py
@@ -0,0 +1,122 @@
+# -*- coding: utf-8 -*-
+
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+#    Copyright (C) 2012-2013 Yahoo! Inc. All Rights Reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+import contextlib
+import logging
+import os
+import sys
+import traceback
+
+logging.basicConfig(level=logging.ERROR)
+
+top_dir = os.path.abspath(os.path.join(os.path.dirname(__file__),
+                                       os.pardir,
+                                       os.pardir))
+sys.path.insert(0, top_dir)
+
+from taskflow import engines
+from taskflow.patterns import linear_flow as lf
+from taskflow.persistence import backends
+from taskflow.persistence import logbook
+from taskflow import task
+
+import tempfile
+
+
+class HiTask(task.Task):
+    def execute(self):
+        print("Hi!")
+
+    def revert(self, **kwargs):
+        print("Whooops, said hi to early, take that back!")
+
+
+class ByeTask(task.Task):
+    def __init__(self, blowup):
+        super(ByeTask, self).__init__()
+        self._blowup = blowup
+
+    def execute(self):
+        if self._blowup:
+            raise Exception("Fail!")
+        print("Bye!")
+
+
+def make_flow(blowup=False):
+    flo = lf.Flow("hello-world")
+    flo.add(HiTask(), ByeTask(blowup))
+    return flo
+
+
+def dump_book(book):
+    for fd in book:
+        print("+ Ran '%s' (%s)" % (fd.name, fd.state))
+        for td in fd:
+            print(" - name = %s" % (td.name))
+            print("   state = %s" % (td.state))
+            print("   results = %s" % (td.results))
+            print("   failure = %s" % (bool(td.failure)))
+            if td.meta and 'progress' in td.meta:
+                print("   progress = %0.2f%%" % (td.meta['progress'] * 100))
+
+
+def print_wrapped(text):
+    print("-" * (len(text)))
+    print(text)
+    print("-" * (len(text)))
+
+
+persist_filename = os.path.join(tempfile.gettempdir(), "persisting.db")
+if os.path.isfile(persist_filename):
+    blowup = False
+else:
+    blowup = True
+
+# Ensure schema upgraded before we continue working.
+backend_config = {
+    'connection': "sqlite:///%s" % (persist_filename),
+}
+with contextlib.closing(backends.fetch(backend_config)) as be:
+    with contextlib.closing(be.get_connection()) as conn:
+        conn.upgrade()
+
+# Now we can run.
+engine_config = {
+    'backend': backend_config,
+    'engine_conf': 'serial',
+    'book': logbook.LogBook("my-test"),
+}
+
+# Make a flow that will blowup if the file doesn't exist previously, if it
+# did exist, assume we won't blowup (and therefore this shows the undo
+# and redo that a flow will go through).
+flo = make_flow(blowup=blowup)
+print_wrapped("Running")
+
+try:
+    eng = engines.load(flo, **engine_config)
+    eng.run()
+    try:
+        os.unlink(persist_filename)
+    except (OSError, IOError):
+        pass
+except Exception:
+    traceback.print_exc(file=sys.stdout)
+
+print_wrapped("Book contents")
+dump_book(engine_config['book'])