b85de3cd9e
A couple of things going on, but I think it makes sense to do them atomically. The NodeBase.create() argument "results" is the global state dictionary that will be saved to "state.json", and re-loaded in later phases and passed to them as the argument "state". So for consistency, call this argument "state" (this fits with the change out to start building the state dictionary earlier in the PluginBase.__init__() calls). Since the "state" is a pretty important part of how everything works, move it into a separate object. This is treated as essentially a singleton. It bundles it nicely together for some added documentation [1]. We move instantiation of this object out of the generic BlockDevice.__init__() call and into the actual cmd_* drivers. This is because there's two distinct instantiation operations -- creating a new state (during cmd_create) and loading an existing state (other cmd_*). This is also safer -- since we know the cmd_* arguments are looking for an existing state.json, we will fail if it somehow goes missing. To more fully unit test this, some testing plugins and new entry-points are added. These add known state values which we check for. These should be a good basis for further tests. [1] as noted, we could probably do some fun things in the future like make this implement a dictionary and have some saftey features like r/o keys. Change-Id: I90eb711b3e9b1ce139eb34bdf3cde641fd06828f
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import logging
|
|
|
|
from diskimage_builder.block_device.exception import \
|
|
BlockDeviceSetupException
|
|
from diskimage_builder.block_device.plugin import NodeBase
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class PartitionNode(NodeBase):
|
|
|
|
flag_boot = 1
|
|
flag_primary = 2
|
|
|
|
def __init__(self, config, parent, prev_partition):
|
|
|
|
super(PartitionNode, self).__init__(config['name'])
|
|
|
|
self.base = config['base']
|
|
self.partitioning = parent
|
|
self.prev_partition = prev_partition
|
|
|
|
self.flags = set()
|
|
if 'flags' in config:
|
|
for f in config['flags']:
|
|
if f == 'boot':
|
|
self.flags.add(self.flag_boot)
|
|
elif f == 'primary':
|
|
self.flags.add(self.flag_primary)
|
|
else:
|
|
raise BlockDeviceSetupException("Unknown flag: %s" % f)
|
|
|
|
if 'size' not in config:
|
|
raise BlockDeviceSetupException("No size in partition" % self.name)
|
|
self.size = config['size']
|
|
|
|
self.ptype = int(config['type'], 16) if 'type' in config else 0x83
|
|
|
|
def get_flags(self):
|
|
return self.flags
|
|
|
|
def get_size(self):
|
|
return self.size
|
|
|
|
def get_type(self):
|
|
return self.ptype
|
|
|
|
def get_edges(self):
|
|
edge_from = [self.base]
|
|
edge_to = []
|
|
if self.prev_partition is not None:
|
|
edge_from.append(self.prev_partition.name)
|
|
return (edge_from, edge_to)
|
|
|
|
def create(self, state, rollback):
|
|
self.partitioning.create(state, rollback)
|