interactive part of removal of mbr code

Signed-off-by: Mattias Andrée <maandree@operamail.com>

1 files changed, 32 insertions, 0 deletions

diff --git a/bootcleanse.py b/bootcleanse.py
new file mode 100644
index 0000000..a65229f
--- /dev/null
+++ b/bootcleanse.py
@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import os
+import sys
+from subprocess import Popen, PIPE
+
+
+mbr = []
+
+for arg in sys.argv[1:]:
+    mbr.append(mbr)
+    print('Remove MBR code from ' + arg)
+    if not (('a' <= arg[-1]) and (arg[-1] <= 'z') and (arg[:-1] in ('/dev/sd', '/dev/hd'))):
+        print('\033[01;33mWarning:\033[21;39m Does not match /dev/(sd|hd)[a-z]')
+
+print()
+print('If you are not sure what you are doing, you may want')
+print('to back up the first 512 of each defected file.')
+print('To back up run `dd if=DEVICE bs=512 count=1 > DEV.backup`.')
+print('To restore run `dd of=DEVICE bs=512 count=1 is=DEV.backup`.')
+print()
+print('\033[01mProcessed? [yes/no]\033[21m')
+
+while True:
+    yn = input()
+    if yn == 'yes':
+        break
+    if yn == 'no':
+        exit(128)
+    print('Only ‘yes’ and ‘no’ is valid')
+