1 files changed, 211 insertions, 0 deletions
diff --git a/openstackclient/api/compute_v2.py b/openstackclient/api/compute_v2.py
new file mode 100644
index 00000000..3bf3a0d8
--- /dev/null
+++ b/openstackclient/api/compute_v2.py
@@ -0,0 +1,211 @@
+#   Licensed under the Apache License, Version 2.0 (the "License"); you may
+#   not use this file except in compliance with the License. You may obtain
+#   a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#   WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#   License for the specific language governing permissions and limitations
+#   under the License.
+#
+
+"""Compute v2 API Library"""
+
+from keystoneauth1 import exceptions as ksa_exceptions
+from osc_lib.api import api
+from osc_lib import exceptions
+from osc_lib.i18n import _
+
+
+class APIv2(api.BaseAPI):
+    """Compute v2 API"""
+
+    def __init__(self, **kwargs):
+        super(APIv2, self).__init__(**kwargs)
+
+    # Overrides
+
+    # TODO(dtroyer): Override find() until these fixes get into an osc-lib
+    #                minimum release
+    def find(
+        self,
+        path,
+        value=None,
+        attr=None,
+    ):
+        """Find a single resource by name or ID
+
+        :param string path:
+            The API-specific portion of the URL path
+        :param string value:
+            search expression (required, really)
+        :param string attr:
+            name of attribute for secondary search
+        """
+
+        try:
+            ret = self._request('GET', "/%s/%s" % (path, value)).json()
+            if isinstance(ret, dict):
+                # strip off the enclosing dict
+                key = list(ret.keys())[0]
+                ret = ret[key]
+        except (
+            ksa_exceptions.NotFound,
+            ksa_exceptions.BadRequest,
+        ):
+            kwargs = {attr: value}
+            try:
+                ret = self.find_one(path, **kwargs)
+            except ksa_exceptions.NotFound:
+                msg = _("%s not found") % value
+                raise exceptions.NotFound(msg)
+
+        return ret
+
+    # Security Groups
+
+    def security_group_create(
+        self,
+        name=None,
+        description=None,
+    ):
+        """Create a new security group
+
+        https://developer.openstack.org/api-ref/compute/#create-security-group
+
+        :param string name:
+            Security group name
+        :param integer description:
+            Security group description
+        """
+
+        url = "/os-security-groups"
+
+        params = {
+            'name': name,
+            'description': description,
+        }
+
+        return self.create(
+            url,
+            json={'security_group': params},
+        )['security_group']
+
+    def security_group_delete(
+        self,
+        security_group=None,
+    ):
+        """Delete a security group
+
+        https://developer.openstack.org/api-ref/compute/#delete-security-group
+
+        :param string security_group:
+            Security group name or ID
+        """
+
+        url = "/os-security-groups"
+
+        security_group = self.find(
+            url,
+            attr='name',
+            value=security_group,
+        )['id']
+        if security_group is not None:
+            return self.delete('/%s/%s' % (url, security_group))
+
+        return None
+
+    def security_group_find(
+        self,
+        security_group=None,
+    ):
+        """Return a security group given name or ID
+
+        https://developer.openstack.org/api-ref/compute/#show-security-group-details
+
+        :param string security_group:
+            Security group name or ID
+        :returns: A dict of the security group attributes
+        """
+
+        url = "/os-security-groups"
+
+        return self.find(
+            url,
+            attr='name',
+            value=security_group,
+        )
+
+    def security_group_list(
+        self,
+        limit=None,
+        marker=None,
+        search_opts=None,
+    ):
+        """Get security groups
+
+        https://developer.openstack.org/api-ref/compute/#list-security-groups
+
+        :param integer limit:
+            query return count limit
+        :param string marker:
+            query marker
+        :param search_opts:
+            (undocumented) Search filter dict
+            all_tenants: True|False - return all projects
+        :returns:
+            list of security groups names
+        """
+
+        params = {}
+        if search_opts is not None:
+            params = dict((k, v) for (k, v) in search_opts.items() if v)
+        if limit:
+            params['limit'] = limit
+        if marker:
+            params['offset'] = marker
+
+        url = "/os-security-groups"
+        return self.list(url, **params)["security_groups"]
+
+    def security_group_set(
+        self,
+        security_group=None,
+        # name=None,
+        # description=None,
+        **params
+    ):
+        """Update a security group
+
+        https://developer.openstack.org/api-ref/compute/#update-security-group
+
+        :param string security_group:
+            Security group name or ID
+
+        TODO(dtroyer): Create an update method in osc-lib
+        """
+
+        # Short-circuit no-op
+        if params is None:
+            return None
+
+        url = "/os-security-groups"
+
+        security_group = self.find(
+            url,
+            attr='name',
+            value=security_group,
+        )
+        if security_group is not None:
+            for (k, v) in params.items():
+                # Only set a value if it is already present
+                if k in security_group:
+                    security_group[k] = v
+            return self._request(
+                "PUT",
+                "/%s/%s" % (url, security_group['id']),
+                json={'security_group': security_group},
+            ).json()['security_group']
+        return None