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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
# This file is part of cloud-init. See LICENSE file for license information.
"""Base NoCloud KVM snapshot."""
import os
import shutil
import tempfile
from ..snapshots import Snapshot
class NoCloudKVMSnapshot(Snapshot):
"""NoCloud KVM image copy backed snapshot."""
platform_name = "nocloud-kvm"
def __init__(self, platform, properties, config, features, image_path):
"""Set up snapshot.
@param platform: platform object
@param properties: image properties
@param config: image config
@param features: supported feature flags
@param image_path: image file to snapshot.
"""
self._workd = tempfile.mkdtemp(prefix='NoCloudKVMSnapshot')
snapshot = os.path.join(self._workd, 'snapshot')
shutil.copyfile(image_path, snapshot)
self._image_path = snapshot
super(NoCloudKVMSnapshot, self).__init__(
platform, properties, config, features)
def launch(self, user_data, meta_data=None, block=True, start=True,
use_desc=None):
"""Launch instance.
@param user_data: user-data for the instance
@param instance_id: instance-id for the instance
@param block: wait until instance is created
@param start: start instance and wait until fully started
@param use_desc: description of snapshot instance use
@return_value: an Instance
"""
key_file = os.path.join(self.platform.config['data_dir'],
self.platform.config['public_key'])
user_data = self.inject_ssh_key(user_data, key_file)
instance = self.platform.create_instance(
self.properties, self.config, self.features,
self._image_path, image_desc=str(self), use_desc=use_desc,
user_data=user_data, meta_data=meta_data)
if start:
instance.start()
return instance
def inject_ssh_key(self, user_data, key_file):
"""Inject the authorized key into the user_data."""
with open(key_file) as f:
value = f.read()
key = 'ssh_authorized_keys:'
value = ' - %s' % value.strip()
user_data = user_data.split('\n')
if key in user_data:
user_data.insert(user_data.index(key) + 1, '%s' % value)
else:
user_data.insert(-1, '%s' % key)
user_data.insert(-1, '%s' % value)
return '\n'.join(user_data)
def destroy(self):
"""Clean up snapshot data."""
shutil.rmtree(self._workd)
super(NoCloudKVMSnapshot, self).destroy()
# vi: ts=4 expandtab
|
