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

More letter updates

Browse source
This commit is contained in:
Alan Mason 2017-11-30 21:52:51 -08:00
parent deb1e8f4fd
commit e9ff02375f
1 changed files with 15 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
Scripts/functions/disk.py
View file

@ -4,13 +4,16 @@ from functions.common import *
import partition_uids import partition_uids
def assign_volume_letters(): def assign_volume_letters():
with open(DISKPART_SCRIPT, 'w') as script:
for vol in get_volumes():
script.write('select volume {Number}\n'.format(**vol))
script.write('assign\n')
# Remove current letters
remove_volume_letters() remove_volume_letters()
# Run script
try: try:
# Run script
with open(DISKPART_SCRIPT, 'w') as script:
for vol in get_volumes():
script.write('select volume {Number}\n'.format(**vol))
script.write('assign\n')
run_program('diskpart /s {script}'.format(script=DISKPART_SCRIPT)) run_program('diskpart /s {script}'.format(script=DISKPART_SCRIPT))
except subprocess.CalledProcessError: except subprocess.CalledProcessError:
pass pass
@ -303,13 +306,14 @@ def reassign_volume_letter(letter, new_letter='I'):
def remove_volume_letters(keep=None): def remove_volume_letters(keep=None):
if not keep: if not keep:
keep = '' keep = ''
with open(DISKPART_SCRIPT, 'w') as script:
for vol in get_volumes():
if vol['Letter'].upper() != keep.upper():
script.write('select volume {Number}\n'.format(**vol))
script.write('remove noerr\n')
# Run script
try: try:
# Run script
with open(DISKPART_SCRIPT, 'w') as script:
for vol in get_volumes():
if vol['Letter'].upper() != keep.upper():
script.write('select volume {Number}\n'.format(**vol))
script.write('remove noerr\n')
run_program('diskpart /s {script}'.format(script=DISKPART_SCRIPT)) run_program('diskpart /s {script}'.format(script=DISKPART_SCRIPT))
except subprocess.CalledProcessError: except subprocess.CalledProcessError:
pass pass