Merge "Make the atom class an abstract class"

1 files changed, 11 insertions, 0 deletions

diff --git a/taskflow/atom.py b/taskflow/atom.py
index 3ece83f..1c5e61e 100644
--- a/taskflow/atom.py
+++ b/taskflow/atom.py
@@ -15,6 +15,8 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
+import abc
+
 from oslo_utils import reflection
 import six
 
@@ -128,6 +130,7 @@ def _build_arg_mapping(atom_name, reqs, rebind_args, function, do_infer,
     return required, optional
 
 
+@six.add_metaclass(abc.ABCMeta)
 class Atom(object):
     """An abstract flow atom that causes a flow to progress (in some manner).
 
@@ -205,6 +208,14 @@ class Atom(object):
                 "by this atom"
                 % dict(item=self.name, oo=sorted(out_of_order)))
 
+    @abc.abstractmethod
+    def execute(self, *args, **kwargs):
+        """Executes this atom."""
+
+    @abc.abstractmethod
+    def revert(self, *args, **kwargs):
+        """Reverts this atom (undoing any :meth:`execute` side-effects)."""
+
     @property
     def name(self):
         """A non-unique name for this atom (human readable)."""