Reordered functions
This commit is contained in:
parent
e56296d8b0
commit
b5d8a55031
GPG key ID:
152FAC923B0E132C
1 changed files with 80 additions and 80 deletions
|
|
@ -14,26 +14,6 @@ USAGE=""" {script_name} clone [source [destination]]
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# Functions
|
# Functions
|
||||||
def show_device_details(dev_path):
|
|
||||||
"""Display device details on screen."""
|
|
||||||
cmd = (
|
|
||||||
'lsblk', '--nodeps',
|
|
||||||
'--output', 'NAME,TRAN,TYPE,SIZE,VENDOR,MODEL,SERIAL',
|
|
||||||
dev_path)
|
|
||||||
result = run_program(cmd)
|
|
||||||
output = result.stdout.decode().splitlines()
|
|
||||||
print_info(output.pop(0))
|
|
||||||
for line in output:
|
|
||||||
print_standard(line)
|
|
||||||
|
|
||||||
# Children info
|
|
||||||
cmd = ('lsblk', '--output', 'NAME,SIZE,FSTYPE,LABEL,MOUNTPOINT', dev_path)
|
|
||||||
result = run_program(cmd)
|
|
||||||
output = result.stdout.decode().splitlines()
|
|
||||||
print_info(output.pop(0))
|
|
||||||
for line in output:
|
|
||||||
print_standard(line)
|
|
||||||
|
|
||||||
def get_device_details(dev_path):
|
def get_device_details(dev_path):
|
||||||
"""Get device details via lsblk, returns JSON dict."""
|
"""Get device details via lsblk, returns JSON dict."""
|
||||||
try:
|
try:
|
||||||
|
|
@ -52,24 +32,6 @@ def get_device_details(dev_path):
|
||||||
# Just return the first device (there should only be one)
|
# Just return the first device (there should only be one)
|
||||||
return json_data['blockdevices'][0]
|
return json_data['blockdevices'][0]
|
||||||
|
|
||||||
def setup_loopback_device(source_path):
|
|
||||||
"""Setup a loopback device for source_path, returns dev_path as str."""
|
|
||||||
cmd = (
|
|
||||||
'losetup',
|
|
||||||
'--find',
|
|
||||||
'--partscan',
|
|
||||||
'--show',
|
|
||||||
source_path)
|
|
||||||
try:
|
|
||||||
out = run_program(cmd, check=True)
|
|
||||||
dev_path = out.stdout.decode().strip()
|
|
||||||
sleep(1)
|
|
||||||
except CalledProcessError:
|
|
||||||
print_error('Failed to setup loopback device for source.')
|
|
||||||
abort()
|
|
||||||
else:
|
|
||||||
return dev_path
|
|
||||||
|
|
||||||
def menu_clone(source_path, dest_path):
|
def menu_clone(source_path, dest_path):
|
||||||
"""ddrescue cloning menu."""
|
"""ddrescue cloning menu."""
|
||||||
source_is_image = False
|
source_is_image = False
|
||||||
|
|
@ -137,48 +99,6 @@ def menu_image(source_path, dest_path):
|
||||||
print_success('GNU ddrescue: Imaging Menu')
|
print_success('GNU ddrescue: Imaging Menu')
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def select_device(description='device', provided_path=None,
|
|
||||||
skip_device={}, allow_image_file=True):
|
|
||||||
"""Select device via provided path or menu, return dev as dict."""
|
|
||||||
dev = {'Is Image': False}
|
|
||||||
|
|
||||||
# Set path
|
|
||||||
if provided_path:
|
|
||||||
dev['Path'] = provided_path
|
|
||||||
else:
|
|
||||||
dev['Path'] = menu_select_device(
|
|
||||||
title='Please select a {}'.format(description),
|
|
||||||
skip_device=skip_device)
|
|
||||||
dev['Path'] = os.path.realpath(dev['Path'])
|
|
||||||
|
|
||||||
# Check path
|
|
||||||
if pathlib.Path(dev['Path']).is_block_device():
|
|
||||||
dev['Dev Path'] = dev['Path']
|
|
||||||
elif allow_image_file and pathlib.Path(dev['Path']).is_file():
|
|
||||||
dev['Dev Path'] = setup_loopback_device(dev['Path'])
|
|
||||||
dev['Is Image'] = True
|
|
||||||
else:
|
|
||||||
print_error('Invalid {} "{}".'.format(description, dev['Path']))
|
|
||||||
abort()
|
|
||||||
|
|
||||||
# Get device details
|
|
||||||
dev['Details'] = get_device_details(dev['Dev Path'])
|
|
||||||
|
|
||||||
# Check for parent device(s)
|
|
||||||
while dev['Details']['pkname']:
|
|
||||||
print_warning('{} "{}" is a child device.'.format(
|
|
||||||
description.title(), dev['Dev Path']))
|
|
||||||
if ask('Use parent device "{}" instead?'.format(
|
|
||||||
dev['Details']['pkname'])):
|
|
||||||
# Update dev with parent info
|
|
||||||
dev['Dev Path'] = dev['Details']['pkname']
|
|
||||||
dev['Details'] = get_device_details(dev['Dev Path'])
|
|
||||||
else:
|
|
||||||
# Leave alone
|
|
||||||
break
|
|
||||||
|
|
||||||
return dev
|
|
||||||
|
|
||||||
def menu_select_device(title='Which device?', skip_device={}):
|
def menu_select_device(title='Which device?', skip_device={}):
|
||||||
"""Select block device via a menu, returns dev_path as str."""
|
"""Select block device via a menu, returns dev_path as str."""
|
||||||
skip_names = [
|
skip_names = [
|
||||||
|
|
@ -230,6 +150,86 @@ def menu_select_device(title='Which device?', skip_device={}):
|
||||||
elif selection == 'Q':
|
elif selection == 'Q':
|
||||||
abort()
|
abort()
|
||||||
|
|
||||||
|
def select_device(description='device', provided_path=None,
|
||||||
|
skip_device={}, allow_image_file=True):
|
||||||
|
"""Select device via provided path or menu, return dev as dict."""
|
||||||
|
dev = {'Is Image': False}
|
||||||
|
|
||||||
|
# Set path
|
||||||
|
if provided_path:
|
||||||
|
dev['Path'] = provided_path
|
||||||
|
else:
|
||||||
|
dev['Path'] = menu_select_device(
|
||||||
|
title='Please select a {}'.format(description),
|
||||||
|
skip_device=skip_device)
|
||||||
|
dev['Path'] = os.path.realpath(dev['Path'])
|
||||||
|
|
||||||
|
# Check path
|
||||||
|
if pathlib.Path(dev['Path']).is_block_device():
|
||||||
|
dev['Dev Path'] = dev['Path']
|
||||||
|
elif allow_image_file and pathlib.Path(dev['Path']).is_file():
|
||||||
|
dev['Dev Path'] = setup_loopback_device(dev['Path'])
|
||||||
|
dev['Is Image'] = True
|
||||||
|
else:
|
||||||
|
print_error('Invalid {} "{}".'.format(description, dev['Path']))
|
||||||
|
abort()
|
||||||
|
|
||||||
|
# Get device details
|
||||||
|
dev['Details'] = get_device_details(dev['Dev Path'])
|
||||||
|
|
||||||
|
# Check for parent device(s)
|
||||||
|
while dev['Details']['pkname']:
|
||||||
|
print_warning('{} "{}" is a child device.'.format(
|
||||||
|
description.title(), dev['Dev Path']))
|
||||||
|
if ask('Use parent device "{}" instead?'.format(
|
||||||
|
dev['Details']['pkname'])):
|
||||||
|
# Update dev with parent info
|
||||||
|
dev['Dev Path'] = dev['Details']['pkname']
|
||||||
|
dev['Details'] = get_device_details(dev['Dev Path'])
|
||||||
|
else:
|
||||||
|
# Leave alone
|
||||||
|
break
|
||||||
|
|
||||||
|
return dev
|
||||||
|
|
||||||
|
def setup_loopback_device(source_path):
|
||||||
|
"""Setup a loopback device for source_path, returns dev_path as str."""
|
||||||
|
cmd = (
|
||||||
|
'losetup',
|
||||||
|
'--find',
|
||||||
|
'--partscan',
|
||||||
|
'--show',
|
||||||
|
source_path)
|
||||||
|
try:
|
||||||
|
out = run_program(cmd, check=True)
|
||||||
|
dev_path = out.stdout.decode().strip()
|
||||||
|
sleep(1)
|
||||||
|
except CalledProcessError:
|
||||||
|
print_error('Failed to setup loopback device for source.')
|
||||||
|
abort()
|
||||||
|
else:
|
||||||
|
return dev_path
|
||||||
|
|
||||||
|
def show_device_details(dev_path):
|
||||||
|
"""Display device details on screen."""
|
||||||
|
cmd = (
|
||||||
|
'lsblk', '--nodeps',
|
||||||
|
'--output', 'NAME,TRAN,TYPE,SIZE,VENDOR,MODEL,SERIAL',
|
||||||
|
dev_path)
|
||||||
|
result = run_program(cmd)
|
||||||
|
output = result.stdout.decode().splitlines()
|
||||||
|
print_info(output.pop(0))
|
||||||
|
for line in output:
|
||||||
|
print_standard(line)
|
||||||
|
|
||||||
|
# Children info
|
||||||
|
cmd = ('lsblk', '--output', 'NAME,SIZE,FSTYPE,LABEL,MOUNTPOINT', dev_path)
|
||||||
|
result = run_program(cmd)
|
||||||
|
output = result.stdout.decode().splitlines()
|
||||||
|
print_info(output.pop(0))
|
||||||
|
for line in output:
|
||||||
|
print_standard(line)
|
||||||
|
|
||||||
def show_usage(script_name):
|
def show_usage(script_name):
|
||||||
print_info('Usage:')
|
print_info('Usage:')
|
||||||
print_standard(USAGE.format(script_name=script_name))
|
print_standard(USAGE.format(script_name=script_name))
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue