Viewing file:  DataSourceDigitalOcean.py (3.42 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# vi: ts=4 expandtab
#
# Author: Neal Shrader <neal@digitalocean.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3, as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from cloudinit import log as logging
from cloudinit import util
from cloudinit import sources
from cloudinit import url_helper
from cloudinit import ec2_utils
from types import *
import functools
LOG = logging.getLogger(__name__)
BUILTIN_DS_CONFIG = {
'metadata_url': 'http://169.254.169.254/metadata/v1/',
'primary_mirrors_url': 'http://mirrors.digitalocean.com/ubuntu',
'security_mirrors_url': 'http://security.ubuntu.com/ubuntu'
}
MD_RETRIES = 0
MD_TIMEOUT = 1
class DataSourceDigitalOcean(sources.DataSource):
def __init__(self, sys_cfg, distro, paths):
sources.DataSource.__init__(self, sys_cfg, distro, paths)
self.metadata = dict()
self.ds_cfg = util.mergemanydict([
util.get_cfg_by_path(sys_cfg, ["datasource", "DigitalOcean"], {}),
BUILTIN_DS_CONFIG])
self.metadata_address = self.ds_cfg['metadata_url']
if self.ds_cfg.get('retries'):
self.retries = self.ds_cfg['retries']
else:
self.retries = MD_RETRIES
if self.ds_cfg.get('timeout'):
self.timeout = self.ds_cfg['timeout']
else:
self.timeout = MD_TIMEOUT
def get_data(self):
caller = functools.partial(util.read_file_or_url, timeout=self.timeout,
retries=self.retries)
md = ec2_utils.MetadataMaterializer(str(caller(self.metadata_address)),
base_url=self.metadata_address,
caller=caller)
self.metadata = md.materialize()
if self.metadata.get('id'):
return True
else:
return False
def get_userdata_raw(self):
return "\n".join(self.metadata['user-data'])
def get_vendordata_raw(self):
return "\n".join(self.metadata['vendor-data'])
def get_public_ssh_keys(self):
if type(self.metadata['public-keys']) is StringType:
return [self.metadata['public-keys']]
else:
return self.metadata['public-keys']
@property
def availability_zone(self):
return self.metadata['region']
def get_instance_id(self):
return self.metadata['id']
def get_hostname(self, fqdn=False):
return self.metadata['hostname']
def get_package_mirror_info(self):
return dict(primary=self.ds_cfg['primary_mirrors_url'], security=self.ds_cfg['security_mirrors_url'])
@property
def launch_index(self):
return None
# Used to match classes to dependencies
datasources = [
(DataSourceDigitalOcean, (sources.DEP_FILESYSTEM, sources.DEP_NETWORK)),
]
# Return a list of data sources that match this set of dependencies
def get_datasource_list(depends):
return sources.list_from_depends(depends, datasources)
|