platformio/platformio-core
1
0
Fork 1
mirror of https://github.com/platformio/platformio-core.git synced 2025-07-29 17:47:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

Catch click's IOError for VSCode Terminal

Browse Source
This commit is contained in:
Ivan Kravets
2017-09-02 17:58:35 +03:00
parent b55b80ecc8
commit a127251107
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
platformio/__main__.py
View File

@ -101,17 +101,18 @@ def main():
except: # pylint: disable=bare-except
pass
# Handle IOError issue with colorama/VSCode
click_echo_origin = click.echo
# Handle IOError issue with VSCode's Terminal (Windows)
click_echo_origin = [click.echo, click.secho]
def _echo(*args, **kwargs):
def _safe_echo(origin, *args, **kwargs):
try:
click_echo_origin(*args, **kwargs)
click_echo_origin[origin](*args, **kwargs)
except IOError:
(sys.stderr.write if kwargs.get("err") else
sys.stdout.write)("%s\n" % (args[0] if args else ""))
click.echo = _echo
click.echo = lambda *args, **kwargs: _safe_echo(0, *args, **kwargs)
click.secho = lambda *args, **kwargs: _safe_echo(1, *args, **kwargs)
cli(None, None, None)
except Exception as e: # pylint: disable=W0703