Allow target drive specification using lower case letter.

6 years ago · 6cbb1b791f
parent 5b1f7890a9
commit 6cbb1b791f
1 changed files with 3 additions and 1 deletions
--- a/4
+++ b/4
@ -159,7 +159,9 @@ if __name__ == '__main__':
            else:
                config.image_path = arg.split(',')
        elif opt in ('-t', '--target'):
-            config.usb_disk = arg
+            # Convert to upper if windows drive name is given
+            config.usb_disk = len(arg)==2 and arg[0].isalpha() and arg[1]==':'\
+                              and arg.upper() or arg
        elif opt in ('-c', '--command'):
            gui = False
        elif opt in ('-u', '--uninstall'):