ae2b051da6
This adds a DataSource for DigitalOcean's metadata service. The service is documented at https://developers.digitalocean.com/metadata/ .
105 lines
3.3 KiB
Python
105 lines
3.3 KiB
Python
# 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 ec2_utils
|
|
from types import StringType
|
|
import functools
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
BUILTIN_DS_CONFIG = {
|
|
'metadata_url': 'http://169.254.169.254/metadata/v1/',
|
|
'mirrors_url': 'http://mirrors.digitalocean.com/'
|
|
}
|
|
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 self.ds_cfg['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)
|