Update Executor.submit(): handle TypeError exceptions of issubclass()…

… and isinstance()
botify-labs · Jul 16, 2014 · 8faa863 · 8faa863
1 parent f9454d1
commit 8faa863
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/simpleflow/executor.py b/simpleflow/executor.py
@@ -96,11 +96,14 @@ def submit(self, func, *args, **kwargs):
         except exceptions.ExecutionBlocked:
             return futures.Future()
 
-        if isinstance(func, Activity):
-            task = self.make_activity_task(func, *args, **kwargs)
-        elif issubclass(func, Workflow):
-            task = self.make_workflow_task(func, *args, **kwargs)
-        else:
+        try:
+            if isinstance(func, Activity):
+                task = self.make_activity_task(func, *args, **kwargs)
+            elif issubclass(func, Workflow):
+                task = self.make_workflow_task(func, *args, **kwargs)
+            else:
+                raise TypeError
+        except TypeError:
             raise TypeError('invalid type {} for {}'.format(
                 type(func), func))