Fix rare crashes caused by damaged disks

This commit is contained in:
Alan Mason 2017-12-03 09:10:09 -08:00
parent 7e4a0259c7
commit 1848c7dc66
2 changed files with 18 additions and 3 deletions

View file

@ -335,7 +335,12 @@ def select_disk(title='Which disk?', disks=[]):
# Build menu
disk_options = []
for disk in disks:
display_name = '{Size}\t[{Table}] ({Type}) {Name}'.format(**disk)
display_name = '{}\t[{}] ({}) {}'.format(
disk.get('Size', ''),
disk.get('Table', ''),
disk.get('Type', ''),
disk.get('Name', 'Unknown'),
)
pwidth=len(str(len(disk['Partitions'])))
for partition in disk['Partitions']:
# Main text

View file

@ -114,7 +114,12 @@ def menu_backup():
show_data(message='Ticket:', data=ticket_number)
show_data(
message = 'Source:',
data = '[{Table}] ({Type}) {Name} {Size}'.format(**disk),
data = '[{}] ({}) {} {}'.format(
disk.get('Table', ''),
disk.get('Type', ''),
disk.get('Name', 'Unknown'),
disk.get('Size', ''),
),
)
show_data(
message = 'Destination:',
@ -293,7 +298,12 @@ def menu_setup():
show_data(message='Using Image:', data=windows_image['Path'])
show_data(
message = 'ERASING:',
data = '[{Table}] ({Type}) {Name} {Size}\n'.format(**dest_disk),
data = '[{}] ({}) {} {}\n'.format(
dest_disk.get('Table', ''),
dest_disk.get('Type', ''),
dest_disk.get('Name', 'Unknown'),
dest_disk.get('Size', ''),
),
warning = True)
for par in dest_disk['Partitions']:
show_data(