00da1982ce
This moves to a more generic config parser that doesn't have plugins parsing part of the tree. I understand why it ended up that way; we have "partitions" key which has special semantics compared to others keys and there was a desire to keep it isolated from core tree->graph code. But this isn't really isolated; you have to reverse-engineer several module-crossing boundaries, extras classes and repetitive recursive functions. Ultimately, plugins should have access to the node graph, but not participate in configuration parsing. This way we ensure that plugins can't invent new methods of configuration parsing. Note: unit tests produce the same tree -> graph conversion as the old method. i.e. this is not intended to have a functional change. Change-Id: I8a5d62a076a5a50597f2f1df3a8615afba6dadb2
96 lines
2.9 KiB
Python
96 lines
2.9 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.graph.digraph import Digraph
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Partition(Digraph.Node):
|
|
|
|
type_string = "partitions"
|
|
|
|
flag_boot = 1
|
|
flag_primary = 2
|
|
|
|
def __init__(self, config, parent, prev_partition):
|
|
if 'name' not in config:
|
|
raise BlockDeviceSetupException(
|
|
"Missing 'name' in partition config: %s" % config)
|
|
self.name = config['name']
|
|
|
|
Digraph.Node.__init__(self, self.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 __repr__(self):
|
|
return "<Partition [%s] on [%s] size [%s] prev [%s]>" \
|
|
% (self.name, self.base, self.size,
|
|
self.prev_partition.name if self.prev_partition else "UNSET")
|
|
|
|
def get_flags(self):
|
|
return self.flags
|
|
|
|
def get_size(self):
|
|
return self.size
|
|
|
|
def get_type(self):
|
|
return self.ptype
|
|
|
|
def get_name(self):
|
|
return self.name
|
|
|
|
def insert_edges(self, dg):
|
|
bnode = dg.find(self.base)
|
|
assert bnode is not None
|
|
dg.create_edge(bnode, self)
|
|
if self.prev_partition is not None:
|
|
logger.debug("Insert edge [%s]" % self)
|
|
dg.create_edge(self.prev_partition, self)
|
|
|
|
def create(self, result, rollback):
|
|
self.partitioning.create(result, rollback)
|
|
|
|
def umount(self, state):
|
|
"""Partition does not need any umount task."""
|
|
pass
|
|
|
|
def cleanup(self, state):
|
|
"""Partition does not need any cleanup."""
|
|
pass
|
|
|
|
def delete(self, state):
|
|
"""Partition does not need any cleanup."""
|
|
pass
|