SunRed/Marlin-Artillery-M600
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

[2.0.x] Update/Fix LPC1768 extra script upload_extra_script.py (#10843)

Browse source 
* Use a different method to find the volume info in Windows
This commit is contained in:
Bob Kuhn 2018-05-25 04:31:18 -05:00 committed by Scott Lahteine
parent 7261f48872
commit e2db509d58
1 changed files with 23 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
Marlin/src/HAL/HAL_LPC1768/upload_extra_script.py
View file

@ -8,6 +8,21 @@ target_filename = "FIRMWARE.CUR"
target_drive = "REARM" target_drive = "REARM"
upload_disk = "" upload_disk = ""
import os
import subprocess
#env_vars = subprocess.check_output('platformio run -t envdump')
#env_vars = env_vars.split('\n')
#for env in env_vars:
# print env
#exit(0)
build_type = os.environ.get("BUILD_TYPE", 'Not Set')
if not(build_type == 'upload' or build_type == 'traceback' or build_type == 'Not Set') :
exit(0)
print '\nSearching for upload disk'
import platform import platform
current_OS = platform.system() current_OS = platform.system()
@ -37,6 +52,14 @@ if current_OS == 'Windows':
upload_disk = 'Disk not found' upload_disk = 'Disk not found'
target_file_found = False target_file_found = False
target_drive_found = False target_drive_found = False
volume_info = subprocess.check_output('powershell -Command volume ')
volume_info = volume_info.split('\n')
for entry in volume_info:
if target_drive in entry and target_drive_found == False: # set upload if not found target file yet
target_drive_found = True
upload_disk = entry[ : entry.find(' ')] + ':'
for drive in drives: for drive in drives:
final_drive_name = drive.strip().rstrip('\\') # typical result (string): 'C:' final_drive_name = drive.strip().rstrip('\\') # typical result (string): 'C:'
# modified version of walklevel() # modified version of walklevel()
@ -49,10 +72,6 @@ if current_OS == 'Windows':
num_sep_this = root.count(os.path.sep) num_sep_this = root.count(os.path.sep)
if num_sep + level <= num_sep_this: if num_sep + level <= num_sep_this:
del dirs[:] del dirs[:]
volume_info = subprocess.check_output('fsutil fsinfo volumeinfo ' + final_drive_name)
if target_drive in volume_info and target_file_found == False: # set upload if not found target file yet
target_drive_found = True
upload_disk = root
if target_filename in files: if target_filename in files:
if target_file_found == False: if target_file_found == False:
upload_disk = root upload_disk = root