wic: get rid of inheritance Disk->DiskImage

There is no need in this inheritance as DiskImage class
is used only in one module and no other classes are inherited.

(From OE-Core rev: 5af1d9bedc2c961eb91faf80251f24c3df754d76)

Signed-off-by: Ed Bartosh <ed.bartosh@linux.intel.com>
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
This commit is contained in:
Ed Bartosh
2016-04-28 13:58:05 +03:00
committed by Richard Purdie
parent d03cbac796
commit 4daf903fb0

View File

@@ -32,47 +32,22 @@ def makedirs(dirname):
if err.errno != errno.EEXIST: if err.errno != errno.EEXIST:
raise raise
class Disk: class DiskImage():
"""
Generic base object for a disk.
"""
def __init__(self, size, device=None):
self._device = device
self._size = size
def create(self):
pass
def cleanup(self):
pass
def get_device(self):
return self._device
def set_device(self, path):
self._device = path
device = property(get_device, set_device)
def get_size(self):
return self._size
size = property(get_size)
class DiskImage(Disk):
""" """
A Disk backed by a file. A Disk backed by a file.
""" """
def __init__(self, image_file, size): def __init__(self, device, size):
Disk.__init__(self, size) self.size = size
self.image_file = image_file self.device = device
self.created = False
def exists(self): def exists(self):
return os.path.exists(self.image_file) return os.path.exists(self.device)
def create(self): def create(self):
if self.device is not None: if self.created:
return return
# create sparse disk image # create sparse disk image
cmd = "truncate %s -s %s" % (self.image_file, self.size) cmd = "truncate %s -s %s" % (self.device, self.size)
exec_cmd(cmd) exec_cmd(cmd)
self.created = True
self.device = self.image_file