From c05e4e07ee9bb8f264a257476423eaf191779758 Mon Sep 17 00:00:00 2001
From: Jeff Bradberry <jeff.bradberry@gmail.com>
Date: Mon, 6 Jul 2020 15:44:06 -0400
Subject: [PATCH] Expose execution environments in awxkit and awx-cli

---
 awxkit/awxkit/api/pages/__init__.py           |  1 +
 awxkit/awxkit/api/pages/api.py                |  2 ++
 .../api/pages/execution_environments.py       | 33 +++++++++++++++++++
 awxkit/awxkit/api/resources.py                |  3 ++
 4 files changed, 39 insertions(+)
 create mode 100644 awxkit/awxkit/api/pages/execution_environments.py

diff --git a/awxkit/awxkit/api/pages/__init__.py b/awxkit/awxkit/api/pages/__init__.py
index fafe5dc08f..1d78d4ba5e 100644
--- a/awxkit/awxkit/api/pages/__init__.py
+++ b/awxkit/awxkit/api/pages/__init__.py
@@ -14,6 +14,7 @@ from .teams import *  # NOQA
 from .credentials import *  # NOQA
 from .unified_jobs import *  # NOQA
 from .unified_job_templates import *  # NOQA
+from .execution_environments import *  # NOQA
 from .projects import *  # NOQA
 from .inventory import *  # NOQA
 from .system_job_templates import *  # NOQA
diff --git a/awxkit/awxkit/api/pages/api.py b/awxkit/awxkit/api/pages/api.py
index 3209232352..4edc07857f 100644
--- a/awxkit/awxkit/api/pages/api.py
+++ b/awxkit/awxkit/api/pages/api.py
@@ -23,6 +23,7 @@ EXPORTABLE_RESOURCES = [
     'inventory_sources',
     'job_templates',
     'workflow_job_templates',
+    'execution_environments',
 ]
 
 
@@ -33,6 +34,7 @@ EXPORTABLE_RELATIONS = [
     'Credentials',
     'Hosts',
     'Groups',
+    'ExecutionEnvironments',
 ]
 
 
diff --git a/awxkit/awxkit/api/pages/execution_environments.py b/awxkit/awxkit/api/pages/execution_environments.py
new file mode 100644
index 0000000000..e48ef324bd
--- /dev/null
+++ b/awxkit/awxkit/api/pages/execution_environments.py
@@ -0,0 +1,33 @@
+import logging
+
+from awxkit.api.mixins import HasCreate
+from awxkit.api.pages import (
+    Credential,
+    Organization,
+)
+from awxkit.api.resources import resources
+
+from . import base
+from . import page
+
+
+log = logging.getLogger(__name__)
+
+
+class ExecutionEnvironment(HasCreate, base.Base):
+
+    dependencies = [Organization, Credential]
+    NATURAL_KEY = ('organization', 'image')
+
+
+page.register_page([resources.execution_environment,
+                    (resources.execution_environments, 'post'),
+                    (resources.organization_execution_environments, 'post')], ExecutionEnvironment)
+
+
+class ExecutionEnvironments(page.PageList, ExecutionEnvironment):
+    pass
+
+
+page.register_page([resources.execution_environments,
+                    resources.organization_execution_environments], ExecutionEnvironments)
diff --git a/awxkit/awxkit/api/resources.py b/awxkit/awxkit/api/resources.py
index d6340cd2d7..997ada1e70 100644
--- a/awxkit/awxkit/api/resources.py
+++ b/awxkit/awxkit/api/resources.py
@@ -28,6 +28,8 @@ class Resources(object):
     _credential_types = 'credential_types/'
     _credentials = 'credentials/'
     _dashboard = 'dashboard/'
+    _execution_environment = r'execution_environments/\d+/'
+    _execution_environments = 'execution_environments/'
     _fact_view = r'hosts/\d+/fact_view/'
     _group = r'groups/\d+/'
     _group_access_list = r'groups/\d+/access_list/'
@@ -141,6 +143,7 @@ class Resources(object):
     _organization_access_list = r'organizations/\d+/access_list/'
     _organization_admins = r'organizations/\d+/admins/'
     _organization_applications = r'organizations/\d+/applications/'
+    _organization_execution_environments = r'organizations/\d+/execution_environments/'
     _organization_inventories = r'organizations/\d+/inventories/'
     _organization_users = r'organizations/\d+/users/'
     _organizations = 'organizations/'