bmaptool: add image_size parameter to BmapBdevCopy constructor

Align BmapBdevCopy class contructor parameters with BmapCopy
class consturctor parameters.

Change-Id: Idc941e0f4baaeee5c8f52772aa536e56848129fa
Signed-off-by: Mikko Ylinen <mikko.ylinen@intel.com>
Signed-off-by: Artem Bityutskiy <artem.bityutskiy@intel.com>

diff --git a/bmaptools/BmapCopy.py b/bmaptools/BmapCopy.py
index f5dca16..92532dc 100644 (file)
--- a/bmaptools/BmapCopy.py
+++ b/bmaptools/BmapCopy.py
@@ -580,12 +580,12 @@ class BmapBdevCopy(BmapCopy):
             self._restore_bdev_settings()
             raise
 
-    def __init__(self, image, dest, bmap = None):
+    def __init__(self, image, dest, bmap = None, image_size = None):
         """ The same as the constructor of the 'BmapCopy' base class, but adds
         useful guard-checks specific to block devices. """
 
         # Call the base class constructor first
-        BmapCopy.__init__(self, image, dest, bmap)
+        BmapCopy.__init__(self, image, dest, bmap, image_size)
 
         self._batch_bytes = 1024 * 1024
         self._batch_blocks = self._batch_bytes / self.block_size