LinHES-system: add_storage.py: if can't find last disk num falback to /data/storage/disk# links

1 files changed, 14 insertions, 1 deletions

diff --git a/abs/core/LinHES-system/add_storage.py b/abs/core/LinHES-system/add_storage.py
index 82de56e..c172f6f 100755
--- a/abs/core/LinHES-system/add_storage.py
+++ b/abs/core/LinHES-system/add_storage.py
@@ -564,7 +564,20 @@ def last_disk_num():
         disk_num = parser.get('storage', 'disk_num')
         num_list.append(int(disk_num))
         num_list.sort()
-    return num_list[-1]
+    try:
+        return num_list[-1]
+    except:
+        # conf file or disk_num is missing so fallback to /data/storage/disk# links
+        for disk_name in glob.glob('/data/storage/disk*'):
+            disk_num = disk_name.strip('/data/storage/disk')
+            num_list.append(int(disk_num))
+            num_list.sort()
+        try:
+            return num_list[-1]
+        except:
+            print "Couldn't find last disk number."
+            sys.exit(1)
+
 #--------------------------------------------
 
 def main(scan_only, destruction, no_mount, install_call, dir_sg):