Implement Block Storage client

kamaki/clients/blockstorage/__init__.py

@@ -32,3 +32,108 @@
 # or implied, of GRNET S.A.
 
 from kamaki.clients.blockstorage.rest_api import BlockStorageRestClient
+
+
+class BlockStorageClient(BlockStorageRestClient):
+    """OpenStack Block Storage v2 client"""
+
+    def list_volumes(self, detail=None):
+        """:returns: (list)"""
+        r = self.volumes_get(detail=detail)
+        return r.json['volumes']
+
+    def get_volume_details(self, volume_id):
+        """:returns: (dict)"""
+        r = self.volumes_get(volume_id=volume_id)
+        return r.json['volume']
+
+    def create_volume(
+            self, size,
+            availability_zone=None,
+            source_volid=None,
+            display_name=None,
+            display_description=None,
+            snapshot_id=None,
+            imageRef=None,
+            volume_type=None,
+            bootable=None,
+            metadata=None):
+        """:returns: (dict) new volumes' details"""
+        r = self.volumes_post(
+            size,
+            availability_zone=availability_zone,
+            source_volid=source_volid,
+            display_name=display_name,
+            display_description=display_description,
+            snapshot_id=snapshot_id,
+            imageRef=imageRef,
+            volume_type=volume_type,
+            bootable=bootable,
+            metadata=metadata)
+        return r.json['volume']
+
+    def update_volume(
+            self, volume_id,
+            display_name=None,
+            display_description=None,
+            delete_on_termination=None,
+            metadata=None):
+        """:returns: (dict) volumes' new details"""
+        args = (
+            display_description, display_name, delete_on_termination, metadata)
+        if args == (None, None, None, None):
+            return self.get_volume_details(volume_id)
+        r = self.volumes_put(
+            volume_id,
+            display_name=display_name,
+            display_description=display_description,
+            delete_on_termination=delete_on_termination,
+            metadata=metadata)
+        return r.json['volume']
+
+    def delete_volume(self, volume_id):
+        r = self.volumes_delete(volume_id)
+        return r.headers
+
+    def list_snapshots(self, detail=None):
+        """:returns: (list)"""
+        r = self.snapshots_get(detail=detail)
+        return r.json['snapshots']
+
+    def get_snapshot_details(self, snapshot_id):
+        """:returns: (dict)"""
+        r = self.snapshots_get(snapshot_id=snapshot_id)
+        return r.json['snapshot']
+
+    def create_snapshot(
+            self, volume_id,
+            force=None, display_name=None, display_description=None):
+        """:returns: (dict) new snapshots' details"""
+        r = self.snapshots_post(
+            volume_id,
+            force=force,
+            display_name=display_name,
+            display_description=display_description)
+        return r.json['snapshot']
+
+    def update_snapshot(
+            self, snapshot_id, display_name=None, display_description=None):
+        """:returns: (dict) snapshots' new details"""
+        if (display_name, display_description) == (None, None):
+            return self.get_snapshot_details(snapshot_id)
+        r = self.snapshots_put(
+            snapshot_id,
+            display_name=display_name, display_description=display_description)
+        return r.json['snapshot']
+
+    def delete_snapshot(self, snapshot_id):
+        r = self.snapshots_delete(snapshot_id)
+        return r.headers
+
+    def list_volume_types(self):
+        r = self.types_get()
+        return r.json['volume_types']
+
+    def get_volume_type_details(self, type_id):
+        r = self.types_get(type_id)
+        return r.json['volume_type']