Merge pull request #17 from Torxed/experimental
Enhanced the multi-drive prompt a little bit
This commit is contained in:
commit
fae0045c74
|
|
@ -481,10 +481,15 @@ if __name__ == '__main__':
|
||||||
if not 'ignore-rerun' in args: args['ignore-rerun'] = False
|
if not 'ignore-rerun' in args: args['ignore-rerun'] = False
|
||||||
if not 'localtime' in args: args['localtime'] = 'Europe/Stockholm' if args['country'] == 'SE' else 'GMT+0' # TODO: Arbitrary for now
|
if not 'localtime' in args: args['localtime'] = 'Europe/Stockholm' if args['country'] == 'SE' else 'GMT+0' # TODO: Arbitrary for now
|
||||||
if not 'drive' in args:
|
if not 'drive' in args:
|
||||||
drives = list(harddrives.keys())
|
drives = sorted(list(harddrives.keys()))
|
||||||
if len(drives) > 1 and 'force' not in args and ('default' in args and 'first-drive' not in args):
|
if len(drives) > 1 and 'force' not in args and ('default' in args and 'first-drive' not in args):
|
||||||
raise KeyError("Multiple disks found, --drive=/dev/X not specified (or --force/--first-drive)")
|
for index, drive in enumerate(drives):
|
||||||
args['drive'] = sorted(drives)[0] # First drive found
|
print(f'{index}: {drive} ({harddrives[drive]})')
|
||||||
|
drive = input('Select one of the above disks (by number): ')
|
||||||
|
if not drive.isdigit():
|
||||||
|
raise KeyError("Multiple disks found, --drive=/dev/X not specified (or --force/--first-drive)")
|
||||||
|
drives = [drives[int(drive)]] # Make sure only the selected drive is in the list of options
|
||||||
|
args['drive'] = drives[0] # First drive found
|
||||||
rerun = args['ignore-rerun']
|
rerun = args['ignore-rerun']
|
||||||
|
|
||||||
if args['drive'][0] != '/':
|
if args['drive'][0] != '/':
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue