From 32e2d929136550962f3bc108f51c7161f40d96b2 Mon Sep 17 00:00:00 2001 From: Ed Bartosh Date: Fri, 10 Feb 2017 16:23:59 +0200 Subject: wic: direct: add PartitionedImage.prepare method Moved code that calls prepare method of Partition objects from DirectPlugin to PartitionedImage.prepare. The idea is to have all code that works with partitions in PartitionedImage class. (From OE-Core rev: 700aa424f0aa239cf4149eed4bfb1dc7d9677431) Signed-off-by: Ed Bartosh Signed-off-by: Richard Purdie --- scripts/lib/wic/plugins/imager/direct.py | 24 +++++++++++------------- 1 file changed, 11 insertions(+), 13 deletions(-) diff --git a/scripts/lib/wic/plugins/imager/direct.py b/scripts/lib/wic/plugins/imager/direct.py index 64a3368526..9c4109e6b8 100644 --- a/scripts/lib/wic/plugins/imager/direct.py +++ b/scripts/lib/wic/plugins/imager/direct.py @@ -182,12 +182,8 @@ class DirectPlugin(ImagerPlugin): rsize_bb = get_bitbake_var('ROOTFS_SIZE', image_name) if rsize_bb: part.size = int(round(float(rsize_bb))) - # need to create the filesystems in order to get their - # sizes before we can add them and do the layout. - part.prepare(self, self.workdir, self.oe_builddir, self.rootfs_dir, - self.bootimg_dir, self.kernel_dir, self.native_sysroot) - self._image.add_partition(part) + self._image.prepare(self) if fstab_path: shutil.move(fstab_path + ".orig", fstab_path) @@ -329,15 +325,17 @@ class PartitionedImage(): else: # msdos partition table part.uuid = '%0x-%02d' % (self.identifier, part.realnum) - def add_partition(self, part): - """ - Add the next partition. Partitions have to be added in the - first-to-last order. - """ - part.ks_pnum = len(self.partitions) + def prepare(self, imager): + """Prepare an image. Call prepare method of all image partitions.""" + for part in self.partitions: + # need to create the filesystems in order to get their + # sizes before we can add them and do the layout. + part.prepare(imager, imager.workdir, imager.oe_builddir, + imager.rootfs_dir, imager.bootimg_dir, + imager.kernel_dir, imager.native_sysroot) - # Converting kB to sectors for parted - part.size_sec = part.disk_size * 1024 // self.sector_size + # Converting kB to sectors for parted + part.size_sec = part.disk_size * 1024 // self.sector_size def layout_partitions(self): """ Layout the partitions, meaning calculate the position of every -- cgit 1.2.3-korg