diff --git a/.bin/Scripts/build-ufd b/.bin/Scripts/build-ufd index 3c0abbe6..7571eae3 100755 --- a/.bin/Scripts/build-ufd +++ b/.bin/Scripts/build-ufd @@ -56,9 +56,13 @@ if __name__ == '__main__': if not args['--update']: prep_device(ufd_dev, UFD_LABEL, use_mbr=args['--use-mbr']) - # TODO: DELETEME - print_success("It's go-time!") - exit_script() + # Mount UFD + try_and_print( + message='Mounting UFD...', + function=mount, + mount_source=ufd_dev, + mount_point='/mnt/UFD', + ) # Copy sources @@ -68,8 +72,6 @@ if __name__ == '__main__': # Hide items - # Unmount sources - # Done if not args['--force']: print_standard('\nDone.') diff --git a/.bin/Scripts/functions/ufd.py b/.bin/Scripts/functions/ufd.py index 06bb3e45..96ce2d0a 100644 --- a/.bin/Scripts/functions/ufd.py +++ b/.bin/Scripts/functions/ufd.py @@ -125,6 +125,19 @@ def is_valid_path(path_obj, path_type): return valid_path +def mount(mount_source, mount_point): + """Mount mount_source on mount_point.""" + os.makedirs(mount_point, exist_ok=True) + cmd = ['mount', mount_source, mount_point] + run_program(cmd) + + +def unmount(mount_point): + """Unmount mount_point.""" + cmd = ['umount', mount_point] + run_program(cmd) + + def prep_device(dev_path, label, use_mbr=False): """Format device in preparation for applying the WizardKit components