Reordered functions
This commit is contained in:
parent
cd4ea75862
commit
427243c933
1 changed files with 39 additions and 39 deletions
|
|
@ -68,45 +68,6 @@ def copy_source(source, items, overwrite=False):
|
||||||
unmount('/mnt/Source')
|
unmount('/mnt/Source')
|
||||||
|
|
||||||
|
|
||||||
def update_boot_entries(boot_entries, boot_files, iso_label, ufd_label):
|
|
||||||
"""Update boot files for UFD usage"""
|
|
||||||
configs = []
|
|
||||||
|
|
||||||
# Find config files
|
|
||||||
for c_path, c_ext in boot_files.items():
|
|
||||||
c_path = find_path(c_path)
|
|
||||||
for item in os.scandir(c_path):
|
|
||||||
if item.name.lower().endswith(c_ext.lower()):
|
|
||||||
configs.append(item.path)
|
|
||||||
|
|
||||||
# Update Linux labels
|
|
||||||
cmd = [
|
|
||||||
'sed',
|
|
||||||
'--in-place',
|
|
||||||
'--regexp-extended',
|
|
||||||
's/{}/{}/'.format(iso_label, ufd_label),
|
|
||||||
*configs,
|
|
||||||
]
|
|
||||||
run_program(cmd)
|
|
||||||
|
|
||||||
# Uncomment extra entries if present
|
|
||||||
for b_path, b_comment in boot_entries:
|
|
||||||
try:
|
|
||||||
find_path('/mnt/UFD{}'.format(b_path))
|
|
||||||
except (FileNotFoundError, NotADirectoryError):
|
|
||||||
# Entry not found, continue to next entry
|
|
||||||
continue
|
|
||||||
|
|
||||||
# Entry found, update config files
|
|
||||||
cmd = [
|
|
||||||
'sed',
|
|
||||||
'--in-place',
|
|
||||||
'"s/#{}#//"'.format(b_comment),
|
|
||||||
*configs,
|
|
||||||
]
|
|
||||||
run_program(cmd)
|
|
||||||
|
|
||||||
|
|
||||||
def find_first_partition(dev_path):
|
def find_first_partition(dev_path):
|
||||||
"""Find path to first partition of dev, returns str."""
|
"""Find path to first partition of dev, returns str."""
|
||||||
cmd = [
|
cmd = [
|
||||||
|
|
@ -390,6 +351,45 @@ def unmount(mount_point):
|
||||||
run_program(cmd)
|
run_program(cmd)
|
||||||
|
|
||||||
|
|
||||||
|
def update_boot_entries(boot_entries, boot_files, iso_label, ufd_label):
|
||||||
|
"""Update boot files for UFD usage"""
|
||||||
|
configs = []
|
||||||
|
|
||||||
|
# Find config files
|
||||||
|
for c_path, c_ext in boot_files.items():
|
||||||
|
c_path = find_path(c_path)
|
||||||
|
for item in os.scandir(c_path):
|
||||||
|
if item.name.lower().endswith(c_ext.lower()):
|
||||||
|
configs.append(item.path)
|
||||||
|
|
||||||
|
# Update Linux labels
|
||||||
|
cmd = [
|
||||||
|
'sed',
|
||||||
|
'--in-place',
|
||||||
|
'--regexp-extended',
|
||||||
|
's/{}/{}/'.format(iso_label, ufd_label),
|
||||||
|
*configs,
|
||||||
|
]
|
||||||
|
run_program(cmd)
|
||||||
|
|
||||||
|
# Uncomment extra entries if present
|
||||||
|
for b_path, b_comment in boot_entries:
|
||||||
|
try:
|
||||||
|
find_path('/mnt/UFD{}'.format(b_path))
|
||||||
|
except (FileNotFoundError, NotADirectoryError):
|
||||||
|
# Entry not found, continue to next entry
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Entry found, update config files
|
||||||
|
cmd = [
|
||||||
|
'sed',
|
||||||
|
'--in-place',
|
||||||
|
'"s/#{}#//"'.format(b_comment),
|
||||||
|
*configs,
|
||||||
|
]
|
||||||
|
run_program(cmd)
|
||||||
|
|
||||||
|
|
||||||
def verify_sources(args, ufd_sources):
|
def verify_sources(args, ufd_sources):
|
||||||
"""Check all sources and abort if necessary, returns dict."""
|
"""Check all sources and abort if necessary, returns dict."""
|
||||||
sources = OrderedDict()
|
sources = OrderedDict()
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue