diff options
Diffstat (limited to 'cloudinit')
| -rw-r--r-- | cloudinit/sources/__init__.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/cloudinit/sources/__init__.py b/cloudinit/sources/__init__.py index 46350255..5842d41b 100644 --- a/cloudinit/sources/__init__.py +++ b/cloudinit/sources/__init__.py @@ -42,7 +42,7 @@ class DataSource(object): __metaclass__ = abc.ABCMeta - def __init__(self, sys_cfg, distro, paths): + def __init__(self, sys_cfg, distro, paths, ud_proc=None): self.sys_cfg = sys_cfg self.distro = distro self.paths = paths @@ -54,11 +54,15 @@ class DataSource(object): name = name[DS_PREFIX:] self.ds_cfg = util.get_cfg_by_path(self.sys_cfg, ("datasource", name), {}) + if not ud_proc: + self.ud_proc = ud.UserDataProcessor(self.paths) + else: + self.ud_proc = ud_proc def get_userdata(self): if self.userdata is None: raw_data = self.get_userdata_raw() - self.userdata = ud.UserDataProcessor(self.paths).process(raw_data) + self.userdata = self.ud_proc.process(raw_data) return self.userdata def get_userdata_raw(self): |