Fixed several bugs in lmi partition metacommand.

2 files changed, 5 insertions, 1 deletions

diff --git a/commands/storage/lmi/scripts/storage/partition.py b/commands/storage/lmi/scripts/storage/partition.py
index 03d33c1..af8bf1c 100644
--- a/commands/storage/lmi/scripts/storage/partition.py
+++ b/commands/storage/lmi/scripts/storage/partition.py
@@ -242,6 +242,7 @@ def get_partition_tables(ns, devices=None):
             yield table.Antecedent.to_instance(), table.Dependent.to_instance()
     else:
         for device in devices:
+            device = common.str2device(ns, device)
             table = get_disk_partition_table(ns, device)
             if table:
                 yield device, table
diff --git a/commands/storage/lmi/scripts/storage/partition_cmd.py b/commands/storage/lmi/scripts/storage/partition_cmd.py
index d054871..7a4879a 100644
--- a/commands/storage/lmi/scripts/storage/partition_cmd.py
+++ b/commands/storage/lmi/scripts/storage/partition_cmd.py
@@ -79,6 +79,7 @@ from lmi.scripts.common import command
 from lmi.scripts.storage import partition, show
 from lmi.scripts.storage.common import str2size, str2device, size2str
 from lmi.scripts.common import formatter
+from lmi.scripts.common.formatter import command as fcmd
 
 class Lister(command.LmiLister):
     COLUMNS = ('DeviceID', "Name", "ElementName", "Type", "Size")
@@ -130,13 +131,15 @@ class Create(command.LmiCheckResult):
         Implementation of 'partition create' command.
         """
         device = str2device(ns, device)
-        size = str2size(size)
+        if size:
+            size = str2size(size)
         ptype = None
         if _extended:
             ptype = partition.PARTITION_TYPE_EXTENDED
         elif _logical:
             ptype = partition.PARTITION_TYPE_LOGICAL
         p = partition.create_partition(ns, device, size, ptype)
+        p = p.to_instance()
         print "Partition %s, with DeviceID %s created." % (p.Name, p.DeviceID)