140 lines
3.6 KiB
Python
Executable file
140 lines
3.6 KiB
Python
Executable file
#!/bin/env python3
|
|
#
|
|
## Wizard Kit: UFD build tool
|
|
|
|
import os
|
|
import sys
|
|
|
|
# Init
|
|
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
|
from collections import OrderedDict
|
|
from docopt import docopt
|
|
from functions.common import *
|
|
from functions.ufd import *
|
|
from settings.ufd import *
|
|
init_global_vars(silent=True)
|
|
set_log_file('Build UFD ({Date-Time}).log'.format(**global_vars))
|
|
|
|
|
|
# Main section
|
|
if __name__ == '__main__':
|
|
try:
|
|
args = docopt(DOCSTRING)
|
|
except SystemExit as err:
|
|
# Catch docopt exits
|
|
print(err)
|
|
exit_script()
|
|
except:
|
|
major_exception()
|
|
try:
|
|
sources = OrderedDict()
|
|
|
|
# Verify selections
|
|
## UFD
|
|
try:
|
|
ufd_dev = find_path(args['--ufd-device'])
|
|
except FileNotFoundError:
|
|
print_error('ERROR: UFD device not found: {}'.format(
|
|
args['--ufd-device']))
|
|
abort()
|
|
if not is_valid_path(ufd_dev, 'UFD'):
|
|
print_error('ERROR: Invalid UFD device: {}'.format(ufd_dev))
|
|
abort()
|
|
## Sources
|
|
for label, data in UFD_SOURCES.items():
|
|
s_path = args[data['Arg']]
|
|
if s_path:
|
|
try:
|
|
s_path_obj = find_path(s_path)
|
|
except FileNotFoundError:
|
|
print_error('ERROR: {} not found: {}'.format(label, s_path))
|
|
abort()
|
|
if not is_valid_path(s_path_obj, data['Type']):
|
|
print_error('ERROR: Invalid {} source: {}'.format(label, s_path))
|
|
abort()
|
|
sources[label] = s_path_obj
|
|
|
|
# Show selections
|
|
## Header
|
|
clear_screen()
|
|
print_success(KIT_NAME_FULL)
|
|
print_standard('UFD Build Tool')
|
|
print_standard(' ')
|
|
## Sources
|
|
print_info('Sources')
|
|
for label in UFD_SOURCES.keys():
|
|
if label in sources:
|
|
print_standard(' {label:<18} {path}'.format(
|
|
label=label+':',
|
|
path=sources[label],
|
|
))
|
|
else:
|
|
print_standard(' {label:<18} {YELLOW}Not Specified{CLEAR}'.format(
|
|
label=label+':',
|
|
**COLORS,
|
|
))
|
|
## Destination
|
|
print_standard(' ')
|
|
print_info('Destination')
|
|
cmd = [
|
|
'lsblk', '--nodeps', '--noheadings',
|
|
'--output', 'NAME,TRAN,SIZE,VENDOR,MODEL,SERIAL',
|
|
]
|
|
result = run_program(cmd, check=False, encoding='utf-8', errors='ignore')
|
|
print_standard(result.stdout.strip())
|
|
cmd = [
|
|
'lsblk', '--noheadings',
|
|
'--output', 'NAME,SIZE,FSTYPE,LABEL,MOUNTPOINT',
|
|
]
|
|
result = run_program(cmd, check=False, encoding='utf-8', errors='ignore')
|
|
for line in result.stdout.splitlines()[1:]:
|
|
print_standard(line)
|
|
if args['--update']:
|
|
print_warning('Updating kit in-place')
|
|
elif args['--use-mbr']:
|
|
print_warning('Formatting using legacy MBR')
|
|
print_standard(' ')
|
|
if not ask('Is the above information correct?'):
|
|
abort()
|
|
## Safety check
|
|
if not args['--update']:
|
|
print_standard(' ')
|
|
print_warning('SAFETY CHECK')
|
|
print_standard(
|
|
'All data will be DELETED from the disk and partition(s) listed above.')
|
|
print_standard(
|
|
'This is irreversible and will lead to {RED}DATA LOSS.{CLEAR}'.format(
|
|
**COLORS))
|
|
if not ask('Asking again to confirm, is this correct?'):
|
|
abort()
|
|
|
|
print_standard(' ')
|
|
print_success("It's go-time!")
|
|
exit_script(1)
|
|
|
|
# Double-check if formating device
|
|
|
|
# Format and partition device
|
|
|
|
# Copy sources
|
|
|
|
# Update boot entries
|
|
|
|
# Install syslinux
|
|
|
|
# Hide items
|
|
|
|
# Unmount sources
|
|
|
|
# Done
|
|
if not args['--force']:
|
|
print_standard('\nDone.')
|
|
pause('Press Enter to exit...')
|
|
exit_script()
|
|
except SystemExit:
|
|
pass
|
|
except:
|
|
major_exception()
|
|
|
|
|
|
# vim: sts=2 sw=2 ts=2
|