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

Remove temp variable from set_mode()

Browse source
This commit is contained in:
2Shirt 2023-08-13 20:47:55 -07:00
parent 42720d322b
commit 7603b93338
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
1 changed files with 6 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
scripts/wk/clone/state.py
View file

@ -1047,24 +1047,15 @@ def select_disk_obj(label:str, disk_menu: cli.Menu, disk_path: str) -> hw_disk.D
def set_mode(docopt_args) -> str: def set_mode(docopt_args) -> str:
"""Set mode from docopt_args or user selection, returns str.""" """Set mode from docopt_args or user selection, returns str."""
mode = '?'
# Check docopt_args
if docopt_args['clone']: if docopt_args['clone']:
mode = 'Clone' return 'Clone'
elif docopt_args['image']:
mode = 'Image' if docopt_args['image']:
return 'Image'
# Ask user if necessary # Ask user if necessary
if not mode: answer = cli.choice('Are we cloning or imaging?', ['C', 'I'])
answer = cli.choice('Are we cloning or imaging?', ['C', 'I']) return 'Clone' if answer == 'C' else 'Image'
if answer == 'C':
mode = 'Clone'
else:
mode = 'Image'
# Done
return mode
if __name__ == '__main__': if __name__ == '__main__':