blob: dab0e5f58a218cc1ba36b24311523198db02d2a3 (
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
|
import os
from cloudinit import util
from cloudinit.settings import (PER_ALWAYS, PER_INSTANCE)
from cloudinit import log as logging
from cloudinit import parts
LOG = logging.getLogger(__name__)
class CloudConfigPartHandler(parts.PartHandler):
def __init__(self, cloud_fn):
parts.PartHandler.__init__(self, PER_ALWAYS)
self.cloud_buf = []
self.cloud_fn = cloud_fn
def list_types(self):
return ['text/cloud-config']
def _handle_part(self, _data, ctype, filename, payload, _frequency):
if ctype == CONTENT_START:
self.cloud_buf = []
return
if ctype == CONTENT_END:
payload = "\n".join(self.cloud_buf)
util.write_file(self.cloud_fn, payload, 0600)
self.cloud_buf = []
return
filename = util.clean_filename(filename)
entry = "\n".join(["#%s" % (filename), str(payload)])
self.config_buf.append(entry)
|
