From 49ec67f07fe2d9247471c928302db4c1b133f34d Mon Sep 17 00:00:00 2001
From: aaronayres35 <36972686+aaronayres35@users.noreply.github.com>
Date: Wed, 21 Oct 2020 08:20:35 -0500
Subject: [PATCH] Add relevant classes to envisage.ui.tasks.api (#322)

* adding classes to envisage.ui.tasks.api and importing them from api in tests/examples

* Add TasksApplicationState to the api

Co-authored-by: Poruri Sai Rahul <rporuri@enthought.com>

Co-authored-by: Poruri Sai Rahul <rporuri@enthought.com>
---
 envisage/tests/test_ids.py                       | 2 +-
 envisage/ui/tasks/api.py                         | 5 ++++-
 examples/plugins/tasks/attractors/run.py         | 2 +-
 examples/plugins/tasks/ipython_kernel/example.py | 3 +--
 4 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/envisage/tests/test_ids.py b/envisage/tests/test_ids.py
index d94432a7e..c805ba60a 100644
--- a/envisage/tests/test_ids.py
+++ b/envisage/tests/test_ids.py
@@ -14,7 +14,7 @@
 from envisage.core_plugin import CorePlugin
 from envisage.plugins.ipython_kernel.api import IPythonKernelPlugin
 from envisage.plugins.python_shell.python_shell_plugin import PythonShellPlugin
-from envisage.ui.tasks.tasks_plugin import TasksPlugin
+from envisage.ui.tasks.api import TasksPlugin
 
 
 class TestIds(unittest.TestCase):
diff --git a/envisage/ui/tasks/api.py b/envisage/ui/tasks/api.py
index 43e23410b..a14f93e35 100644
--- a/envisage/ui/tasks/api.py
+++ b/envisage/ui/tasks/api.py
@@ -8,8 +8,11 @@
 #
 # Thanks for using Enthought open source!
 from .preferences_category import PreferencesCategory
+from .preferences_dialog import PreferencesDialog, PreferencesTab
 from .preferences_pane import PreferencesPane
 from .task_extension import TaskExtension
 from .task_factory import TaskFactory
 from .task_window import TaskWindow
-from .tasks_application import TasksApplication
+from .task_window_event import TaskWindowEvent, VetoableTaskWindowEvent
+from .tasks_application import TasksApplication, TasksApplicationState
+from .tasks_plugin import TasksPlugin
diff --git a/examples/plugins/tasks/attractors/run.py b/examples/plugins/tasks/attractors/run.py
index ff6f429ea..9eed20800 100644
--- a/examples/plugins/tasks/attractors/run.py
+++ b/examples/plugins/tasks/attractors/run.py
@@ -3,7 +3,7 @@
 
 # Plugin imports.
 from envisage.core_plugin import CorePlugin
-from envisage.ui.tasks.tasks_plugin import TasksPlugin
+from envisage.ui.tasks.api import TasksPlugin
 from attractors.attractors_plugin import AttractorsPlugin
 
 # Local imports.
diff --git a/examples/plugins/tasks/ipython_kernel/example.py b/examples/plugins/tasks/ipython_kernel/example.py
index 89acd56f1..ce63e7a57 100644
--- a/examples/plugins/tasks/ipython_kernel/example.py
+++ b/examples/plugins/tasks/ipython_kernel/example.py
@@ -9,8 +9,7 @@
     IPYTHON_KERNEL_PROTOCOL,
     IPythonKernelUIPlugin,
 )
-from envisage.ui.tasks.api import TasksApplication, TaskFactory
-from envisage.ui.tasks.tasks_plugin import TasksPlugin
+from envisage.ui.tasks.api import TasksApplication, TaskFactory, TasksPlugin
 from pyface.qt import QtCore
 from pyface.tasks.api import TaskWindowLayout
 from pyface.util.guisupport import get_app_qt4