CasperSecurity

Current Path : /lib/python3/dist-packages/cloudinit/sources/

Upload File :

Current File : //lib/python3/dist-packages/cloudinit/sources/DataSourceBigstep.py

# Copyright (C) 2015-2016 Bigstep Cloud Ltd.
#
# Author: Alexandru Sirbu <alexandru.sirbu@bigstep.com>
#
# This file is part of cloud-init. See LICENSE file for license information.

import errno
import json
import os

from cloudinit import sources, url_helper, util


class DataSourceBigstep(sources.DataSource):

    dsname = "Bigstep"

    def __init__(self, sys_cfg, distro, paths):
        super().__init__(sys_cfg, distro, paths)
        self.metadata = {}
        self.vendordata_raw = ""
        self.userdata_raw = ""

    def _get_data(self, apply_filter=False) -> bool:
        url = self._get_url_from_file()
        if url is None:
            return False
        response = url_helper.readurl(url)
        decoded = json.loads(response.contents.decode())
        self.metadata = decoded["metadata"]
        self.vendordata_raw = decoded["vendordata_raw"]
        self.userdata_raw = decoded["userdata_raw"]
        return True

    def _get_subplatform(self) -> str:
        """Return the subplatform metadata source details."""
        return f"metadata ({self._get_url_from_file()})"

    def _get_url_from_file(self):
        url_file = os.path.join(
            self.paths.cloud_dir, "data", "seed", "bigstep", "url"
        )
        try:
            content = util.load_text_file(url_file)
        except IOError as e:
            # If the file doesn't exist, then the server probably isn't a
            # Bigstep instance; otherwise, another problem exists which needs
            # investigation
            if e.errno == errno.ENOENT:
                return None
            else:
                raise
        return content


# Used to match classes to dependencies
datasources = [
    (DataSourceBigstep, (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)