|
|
@ -8,6 +8,21 @@ target_filename = "FIRMWARE.CUR" |
|
|
|
target_drive = "REARM" |
|
|
|
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 |
|
|
|
current_OS = platform.system() |
|
|
|
|
|
|
@ -37,6 +52,14 @@ if current_OS == 'Windows': |
|
|
|
upload_disk = 'Disk not found' |
|
|
|
target_file_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: |
|
|
|
final_drive_name = drive.strip().rstrip('\\') # typical result (string): 'C:' |
|
|
|
# modified version of walklevel() |
|
|
@ -49,10 +72,6 @@ if current_OS == 'Windows': |
|
|
|
num_sep_this = root.count(os.path.sep) |
|
|
|
if num_sep + level <= num_sep_this: |
|
|
|
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_file_found == False: |
|
|
|
upload_disk = root |
|
|
|