from taskflow import task

from taskflow.patterns import linear_flow

import logging

logging.basicConfig(level=5)

from taskflow import engines
from taskflow.persistence import models

from taskflow.listeners import logging as logging_listener

from oslo_utils import uuidutils


class ProvidesThings(task.Task):

    default_provides = frozenset(['a'])

    def execute(self):
        print 'Running %s' % self.name
        return {
            'a': 2,
        }


class ExecutesThings(task.Task):
    def execute(self, *args, **kwargs):
        print 'Running %s' % self.name
        print args
        print kwargs

    
class MoreExecutesThings(task.Task):
    def execute(self, *args, **kwargs):
        print 'Running %s' % self.name
        print args
        print kwargs


stuff = linear_flow.Flow("stuff")
stuff.add(ProvidesThings(), ExecutesThings(requires=['a']))

flow_detail = models.FlowDetail("all-the-things", uuidutils.generate_uuid())

print "==== E1 ==="
e = engines.load(stuff, flow_detail=flow_detail)
e.compile()
with logging_listener.DynamicLoggingListener(e):
    e.run()
print e.storage.backend.memory.pformat()

print "==== E2 ==="
stuff.add(MoreExecutesThings(requires=['a']))
e2 = engines.load(stuff, backend=e.storage.backend, flow_detail=flow_detail)
e2.compile()
with logging_listener.DynamicLoggingListener(e2):
    e2.run()
print e2.storage.backend.memory.pformat()