Author: tomaz
Date: Sat Mar 9 04:24:36 2013
New Revision: 1454654
URL: http://svn.apache.org/r1454654
Log:
Start working on Digital Ocean driver.
Added:
libcloud/trunk/libcloud/compute/drivers/digitalocean.py
Modified:
libcloud/trunk/libcloud/compute/providers.py
libcloud/trunk/libcloud/compute/types.py
Added: libcloud/trunk/libcloud/compute/drivers/digitalocean.py
URL:
http://svn.apache.org/viewvc/libcloud/trunk/libcloud/compute/drivers/digitalocean.py?rev=1454654&view=auto
==============================================================================
--- libcloud/trunk/libcloud/compute/drivers/digitalocean.py (added)
+++ libcloud/trunk/libcloud/compute/drivers/digitalocean.py Sat Mar 9 04:24:36
2013
@@ -0,0 +1,171 @@
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.
+# The ASF licenses this file to You 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.
+"""
+Digital Ocean Driver
+"""
+
+from libcloud.utils.py3 import httplib
+
+from libcloud.common.base import ConnectionUserAndKey, JsonResponse
+from libcloud.compute.types import Provider, NodeState
+from libcloud.compute.base import NodeDriver
+from libcloud.compute.base import Node, NodeImage, NodeSize, NodeLocation
+
+
+class DigitalOceanResponse(JsonResponse):
+ pass
+
+
+class SSHKey(object):
+ def __init__(self, id, name, pub_key):
+ self.id = id
+ self.name = name
+ self.pub_key = pub_key
+
+ def __repr__(self):
+ return (('<SSHKey: id=%s, name=%s, pub_key=%s>') %
+ (self.id, self.name, self.pub_key))
+
+
+class DigitalOceanConnection(ConnectionUserAndKey):
+ """
+ Connection class for the DigitalOcean driver.
+ """
+
+ host = 'api.digitalocean.com'
+ responseCls = DigitalOceanResponse
+
+ def add_default_params(self, params):
+ """
+ Add parameters that are necessary for every request
+
+ This method adds C{api_key} and C{api_responseFormat} to
+ the request.
+ """
+ params['client_id'] = self.user_id
+ params['api_key'] = self.key
+ return params
+
+
+class DigitalOceanNodeDriver(NodeDriver):
+ """
+ DigitalOceanNode node driver.
+ """
+
+ connectionCls = DigitalOceanConnection
+
+ type = Provider.DIGITAL_OCEAN
+ name = 'Digital Ocean'
+ website = 'https://www.digitalocean.com'
+ features = {'create_node': ['ssh_key']}
+
+ NODE_STATE_MAP = {'new': NodeState.PENDING,
+ 'off': NodeState.REBOOTING,
+ 'active': NodeState.RUNNING}
+
+ def list_nodes(self):
+ data = self.connection.request('/droplets').object['droplets']
+ return list(map(self._to_node, data))
+
+ def list_locations(self):
+ data = self.connection.request('/regions').object['regions']
+ return list(map(self._to_location, data))
+
+ def list_images(self):
+ data = self.connection.request('/images').object['images']
+ return list(map(self._to_image, data))
+
+ def list_sizes(self):
+ data = self.connection.request('/sizes').object['sizes']
+ return list(map(self._to_size, data))
+
+ def create_node(self, name, size, image, location, ex_ssh_key_ids=None):
+ params = {'name': name, 'size_id': size.id, 'image_id': image.id,
+ 'region_id': location.id}
+
+ if ex_ssh_key_ids:
+ params['ssh_key_ids'] = ','.join(ex_ssh_key_ids)
+
+ data = self.connection.request('/droplets/new', params=params).object
+ return self._to_node(data=data['droplet'])
+
+ def reboot_node(self, node):
+ res = self.connection.request('/droplets/%s/reboot/' % (node.id))
+ return res.status == httplib.OK
+
+ def destroy_node(self, node):
+ res = self.connection.request('/droplets/%s/destroy/' % (node.id))
+ return res.status == httplib.OK
+
+ def ex_list_ssh_keys(self):
+ data = self.connection.request('/ssh_keys').object['ssh_keys']
+ return list(map(self._to_ssh_key, data))
+
+ def ex_create_ssh_key(self, name, ssh_key_pub):
+ params = {'name': name, 'ssh_pub_key': ssh_key_pub}
+ data = self.connection.request('/ssh_keys/new/', method='GET',
+ params=params).object
+ assert 'ssh_key' in data
+ return self._to_ssh_key(data=data['ssh_key'])
+
+ def ex_destroy_ssh_key(self, key_id):
+ res = self.connection.request('/ssh_keys/%s/destroy/' % (key_id))
+ return res.status == httplib.OK
+
+ def _to_node(self, data):
+ extra_keys = ['backups_active', 'region_id']
+ if 'staus' in data:
+ state = self.NODE_STATE_MAP.get(data['status'], NodeState.UNKNOWN)
+ else:
+ state = NodeState.UNKNOWN
+
+ if 'ip_address' in data:
+ public_ips = [data['ip_address']]
+ else:
+ public_ips = []
+
+ extra = {}
+ for key in extra_keys:
+ if key in data:
+ extra[key] = data[key]
+
+ node = Node(id=data['id'], name=data['name'], state=state,
+ public_ips=public_ips, private_ips=None, extra=extra,
+ driver=self)
+ return node
+
+ def _to_image(self, data):
+ extra = {'distribution': data['distribution']}
+ return NodeImage(id=data['id'], name=data['name'], extra=extra,
+ driver=self)
+
+ def _to_location(self, data):
+ return NodeLocation(id=data['id'], name=data['name'], country=None,
+ driver=self)
+
+ def _to_size(self, data):
+ ram = data['name'].lower()
+
+ if 'mb' in ram:
+ ram = int(ram.replace('mb', ''))
+ elif 'gb' in ram:
+ ram = int(ram.replace('gb', '')) * 1024
+
+ return NodeSize(id=data['id'], name=data['name'], ram=ram, disk=0,
+ bandwidth=0, price=0, driver=self)
+
+ def _to_ssh_key(self, data):
+ return SSHKey(id=data['id'], name=data['name'],
+ pub_key=data.get('ssh_pub_key', None))
Modified: libcloud/trunk/libcloud/compute/providers.py
URL:
http://svn.apache.org/viewvc/libcloud/trunk/libcloud/compute/providers.py?rev=1454654&r1=1454653&r2=1454654&view=diff
==============================================================================
--- libcloud/trunk/libcloud/compute/providers.py (original)
+++ libcloud/trunk/libcloud/compute/providers.py Sat Mar 9 04:24:36 2013
@@ -122,7 +122,9 @@ DRIVERS = {
Provider.HOSTVIRTUAL:
('libcloud.compute.drivers.hostvirtual', 'HostVirtualNodeDriver'),
Provider.ABIQUO:
- ('libcloud.compute.drivers.abiquo', 'AbiquoNodeDriver')
+ ('libcloud.compute.drivers.abiquo', 'AbiquoNodeDriver'),
+ Provider.DIGITAL_OCEAN:
+ ('libcloud.compute.drivers.digitalocean', 'DigitalOceanNodeDriver')
}
Modified: libcloud/trunk/libcloud/compute/types.py
URL:
http://svn.apache.org/viewvc/libcloud/trunk/libcloud/compute/types.py?rev=1454654&r1=1454653&r2=1454654&view=diff
==============================================================================
--- libcloud/trunk/libcloud/compute/types.py (original)
+++ libcloud/trunk/libcloud/compute/types.py Sat Mar 9 04:24:36 2013
@@ -114,6 +114,7 @@ class Provider(object):
RACKSPACE_FIRST_GEN = 'rackspace_first_gen'
HOSTVIRTUAL = 'hostvirtual'
ABIQUO = 'abiquo'
+ DIGITAL_OCEAN = 'digitalocean'
# Deprecated constants which are still supported
EC2_US_EAST = 'ec2_us_east'
