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

Allow using the live CD/DVD/etc as a source

Browse source
This commit is contained in:
2Shirt 2024-03-30 23:04:49 -07:00
parent 3aff533c4d
commit ee7c7c2448
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/wk/kit/ufd.py
View file

@ -5,6 +5,7 @@ import logging
import math
import os
import pathlib
import re
import shutil
from subprocess import CalledProcessError
@ -287,7 +288,7 @@ def confirm_selections(update=False) -> None:
def copy_source(source, items, from_live=False, overwrite=False) -> None:
"""Copy source items to /mnt/UFD."""
is_image = not from_live and source.is_file()
is_image = not from_live and (source.is_file() or source.is_block_device())
items_not_found = False
# Mount source if necessary
@ -472,6 +473,7 @@ def is_valid_path(path_obj, path_type) -> bool:
valid_path = path_obj.is_file() and path_obj.suffix.lower() == '.img'
elif path_type == 'ISO':
valid_path = path_obj.is_file() and path_obj.suffix.lower() == '.iso'
valid_path = valid_path or re.match(r'^/dev/sr\d+$', str(path_obj))
elif path_type == 'UFD':
valid_path = path_obj.is_block_device()