[oe-commits] Ed Bartosh : wic: Call methods better way
git at git.openembedded.org
git at git.openembedded.org
Wed Jul 8 12:09:58 UTC 2015
Module: openembedded-core.git
Branch: master-next
Commit: 066f713f71828ff6f408448e5ea4fcd56faa423e
URL: http://git.openembedded.org/?p=openembedded-core.git&a=commit;h=066f713f71828ff6f408448e5ea4fcd56faa423e
Author: Ed Bartosh <ed.bartosh at linux.intel.com>
Date: Sat Jun 27 12:56:38 2015 +0300
wic: Call methods better way
Shortened code by using getattr to obtain a method to call for
prepare_empty_partition_* and prepare_rootfs_* methods.
Signed-off-by: Ed Bartosh <ed.bartosh at linux.intel.com>
---
.../lib/wic/kickstart/custom_commands/partition.py | 47 +++++-----------------
1 file changed, 11 insertions(+), 36 deletions(-)
diff --git a/scripts/lib/wic/kickstart/custom_commands/partition.py b/scripts/lib/wic/kickstart/custom_commands/partition.py
index e3ae3ce..0741bb2 100644
--- a/scripts/lib/wic/kickstart/custom_commands/partition.py
+++ b/scripts/lib/wic/kickstart/custom_commands/partition.py
@@ -160,8 +160,12 @@ class Wic_PartData(Mic_PartData):
self.prepare_swap_partition(cr_workdir, oe_builddir,
native_sysroot)
elif self.fstype:
- self.prepare_empty_partition(cr_workdir, oe_builddir,
- native_sysroot)
+ for prefix in ("ext", "btrfs", "vfat", "squashfs"):
+ if self.fstype.startswith(prefix):
+ method = getattr(self,
+ "prepare_empty_partition_" + prefix)
+ method(cr_workdir, oe_builddir, native_sysroot)
+ break
return
plugins = pluginmgr.get_source_plugins()
@@ -223,23 +227,11 @@ class Wic_PartData(Mic_PartData):
if os.path.isfile(rootfs):
os.remove(rootfs)
- if self.fstype.startswith("ext"):
- return self.prepare_rootfs_ext(rootfs, oe_builddir,
- rootfs_dir, native_sysroot,
- pseudo)
- elif self.fstype.startswith("btrfs"):
- return self.prepare_rootfs_btrfs(rootfs, oe_builddir,
- rootfs_dir, native_sysroot,
- pseudo)
-
- elif self.fstype.startswith("vfat"):
- return self.prepare_rootfs_vfat(rootfs, oe_builddir,
- rootfs_dir, native_sysroot,
- pseudo)
- elif self.fstype.startswith("squashfs"):
- return self.prepare_rootfs_squashfs(rootfs, oe_builddir,
- rootfs_dir, native_sysroot,
- pseudo)
+ for prefix in ("ext", "btrfs", "vfat", "squashfs"):
+ if self.fstype.startswith(prefix):
+ method = getattr(self, "prepare_rootfs_" + prefix)
+ return method(rootfs, oe_builddir, rootfs_dir,
+ native_sysroot, pseudo)
def prepare_rootfs_ext(self, rootfs, oe_builddir, rootfs_dir,
native_sysroot, pseudo):
@@ -391,23 +383,6 @@ class Wic_PartData(Mic_PartData):
return 0
- def prepare_empty_partition(self, cr_workdir, oe_builddir, native_sysroot):
- """
- Prepare an empty partition.
- """
- if self.fstype.startswith("ext"):
- return self.prepare_empty_partition_ext(cr_workdir, oe_builddir,
- native_sysroot)
- elif self.fstype.startswith("btrfs"):
- return self.prepare_empty_partition_btrfs(cr_workdir, oe_builddir,
- native_sysroot)
- elif self.fstype.startswith("vfat"):
- return self.prepare_empty_partition_vfat(cr_workdir, oe_builddir,
- native_sysroot)
- elif self.fstype.startswith("squashfs"):
- return self.prepare_empty_partition_squashfs(cr_workdir, oe_builddir,
- native_sysroot)
-
def prepare_empty_partition_ext(self, cr_workdir, oe_builddir,
native_sysroot):
"""
More information about the Openembedded-commits
mailing list