summaryrefslogtreecommitdiff
path: root/cloudinit/sources/DataSourceBigstep.py
blob: 2d66c6092b54c9ac975009e20214098d4c08e078 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#
#    Copyright (C) 2015-2016 Bigstep Cloud Ltd.
#
#    Author: Alexandru Sirbu <alexandru.sirbu@bigstep.com>
#

import json
import errno

from cloudinit import log as logging
from cloudinit import sources
from cloudinit import util
from cloudinit import url_helper

LOG = logging.getLogger(__name__)


class DataSourceBigstep(sources.DataSource):
    def __init__(self, sys_cfg, distro, paths):
        sources.DataSource.__init__(self, sys_cfg, distro, paths)
        self.metadata = {}
        self.vendordata_raw = ""
        self.userdata_raw = ""

    def get_data(self, apply_filter=False):
        try:
            url = get_url_from_file()
        except IOError as e:
            if e.errno == errno.ENOENT:
                return False
            else:
                raise
        response = url_helper.readurl(url)
        decoded = json.loads(response.contents)
        self.metadata = decoded["metadata"]
        self.vendordata_raw = decoded["vendordata_raw"]
        self.userdata_raw = decoded["userdata_raw"]
        return True


def get_url_from_file():
    content = util.load_file("/var/lib/cloud/data/seed/bigstep/url")
    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)