2Shirt/WizardKit
2Shirt/WizardKit
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases 26 Packages Wiki Activity Actions

Handle non-ISO source items

Browse source
This commit is contained in:
2Shirt 2019-04-19 18:14:10 -07:00
parent 731a281805
commit a83dcdd06a
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
.bin/Scripts/functions/ufd.py
View file

@ -59,17 +59,29 @@ def confirm_selections(args):
def copy_source(source, items, overwrite=False): def copy_source(source, items, overwrite=False):
"""Mount source and copy items to /mnt/UFD.""" """Copy source items to /mnt/UFD."""
mount(source, '/mnt/Source') is_iso = source.name.lower().endswith('.iso')
# Mount source if necessary
if is_iso:
mount(source, '/mnt/Source')
# Copy items
for i_source, i_dest in items: for i_source, i_dest in items:
i_source = '/mnt/Source{}'.format(i_source) i_source = '{}{}'.format(
'/mnt/Source' if is_iso else source,
i_source,
)
i_dest = '/mnt/UFD{}'.format(i_dest) i_dest = '/mnt/UFD{}'.format(i_dest)
try: try:
recursive_copy(i_source, i_dest, overwrite=overwrite) recursive_copy(i_source, i_dest, overwrite=overwrite)
except FileNotFoundError: except FileNotFoundError:
# Going to assume (hope) that this is fine # Going to assume (hope) that this is fine
pass pass
unmount('/mnt/Source')
# Unmount source if necessary
if is_iso:
unmount('/mnt/Source')
def find_first_partition(dev_path): def find_first_partition(dev_path):