From 70b501516258ae6b76d6a8b6cd86368ed4b63f11 Mon Sep 17 00:00:00 2001 From: Ross Baquir Date: Mon, 30 Mar 2020 17:41:58 -0700 Subject: [PATCH] Use version_arg in ESSENTIAL_BINARIES dict --- lib/python/qmk/cli/doctor.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/lib/python/qmk/cli/doctor.py b/lib/python/qmk/cli/doctor.py index 002301455..7f0b52ffb 100755 --- a/lib/python/qmk/cli/doctor.py +++ b/lib/python/qmk/cli/doctor.py @@ -16,8 +16,8 @@ ESSENTIAL_BINARIES = { 'dfu-programmer': {}, 'avrdude': {}, 'dfu-util': {}, - 'avr-gcc': {}, - 'arm-none-eabi-gcc': {}, + 'avr-gcc': {'version_arg': '-dumpversion'}, + 'arm-none-eabi-gcc': {'version_arg': '-dumpversion'}, 'bin/qmk': {}, } ESSENTIAL_SUBMODULES = ['lib/chibios', 'lib/lufa'] @@ -151,10 +151,8 @@ def is_executable(command): return False # Make sure the command can be executed - check = subprocess.run([command, '-dumpversion'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, timeout=5, universal_newlines=True) - - if check.returncode > 1: # if -dumpversion returns error check with --version instead - check = subprocess.run([command, '--version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, timeout=5, universal_newlines=True) + version_arg = ESSENTIAL_BINARIES[command].get('version_arg', '--version') + check = subprocess.run([command, version_arg], stdout=subprocess.PIPE, stderr=subprocess.PIPE, timeout=5, universal_newlines=True) ESSENTIAL_BINARIES[command]['output'] = check.stdout