Fix bug when no boot partition is specified

This commit is contained in:
Anders Ingemann 2013-10-27 08:52:26 +01:00
parent e1ab4dc1ae
commit 0fdcc2e27b
2 changed files with 4 additions and 2 deletions

View file

@ -11,7 +11,8 @@ class GPTPartitionMap(AbstractPartitionMap):
if 'boot' in data: if 'boot' in data:
self.boot = GPTPartition(data['boot']['size'], data['boot']['filesystem'], 'boot', None) self.boot = GPTPartition(data['boot']['size'], data['boot']['filesystem'], 'boot', None)
self.partitions.append(self.boot) self.partitions.append(self.boot)
self.root = GPTPartition(data['root']['size'], data['root']['filesystem'], 'root', self.boot) self.root = GPTPartition(data['root']['size'], data['root']['filesystem'], 'root',
getattr(self, 'boot', None))
self.partitions.append(self.root) self.partitions.append(self.root)
if 'swap' in data: if 'swap' in data:
self.swap = GPTSwapPartition(data['swap']['size'], self.root) self.swap = GPTSwapPartition(data['swap']['size'], self.root)

View file

@ -11,7 +11,8 @@ class MBRPartitionMap(AbstractPartitionMap):
if 'boot' in data: if 'boot' in data:
self.boot = MBRPartition(data['boot']['size'], data['boot']['filesystem'], None) self.boot = MBRPartition(data['boot']['size'], data['boot']['filesystem'], None)
self.partitions.append(self.boot) self.partitions.append(self.boot)
self.root = MBRPartition(data['root']['size'], data['root']['filesystem'], self.boot) self.root = MBRPartition(data['root']['size'], data['root']['filesystem'],
getattr(self, 'boot', None))
self.partitions.append(self.root) self.partitions.append(self.root)
if 'swap' in data: if 'swap' in data:
self.swap = MBRSwapPartition(data['swap']['size'], self.root) self.swap = MBRSwapPartition(data['swap']['size'], self.root)