Stop with the fsset usage.

1 files changed, 33 insertions, 0 deletions

diff --git a/booty/util.py b/booty/util.py
new file mode 100644
index 000000000..f9a1b3e84
--- /dev/null
+++ b/booty/util.py
@@ -0,0 +1,33 @@
+import string
+
+def getDiskPart(dev):
+    cut = len(dev)
+    if (dev.startswith('rd/') or dev.startswith('ida/') or
+            dev.startswith('cciss/') or dev.startswith('sx8/') or
+            dev.startswith('mapper/') or dev.startswith('mmcblk')):
+        if dev[-2] == 'p':
+            cut = -1
+        elif dev[-3] == 'p':
+            cut = -2
+    else:
+        if dev[-2] in string.digits:
+            cut = -2
+        elif dev[-1] in string.digits:
+            cut = -1
+
+    name = dev[:cut]
+
+    # hack off the trailing 'p' from /dev/cciss/*, for example
+    if name[-1] == 'p':
+        for letter in name:
+            if letter not in string.letters and letter != "/":
+                name = name[:-1]
+                break
+
+    if cut < 0:
+        partNum = int(dev[cut:]) - 1
+    else:
+        partNum = None
+
+    return (name, partNum)
+