You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
142 lines
4.4 KiB
142 lines
4.4 KiB
5 years ago
|
# Copyright (c) 2014-present PlatformIO <contact@platformio.org>
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from traceback import format_exc
|
||
|
|
|
||
|
|
import click
|
||
|
|
|
||
|
|
from platformio import __version__, exception, maintenance, util
|
||
|
|
from platformio.commands import PlatformioCLI
|
||
|
|
from platformio.compat import CYGWIN
|
||
|
|
|
||
|
|
|
||
|
|
@click.command(
|
||
|
|
cls=PlatformioCLI, context_settings=dict(help_option_names=["-h", "--help"])
|
||
|
|
)
|
||
|
|
@click.version_option(__version__, prog_name="PlatformIO")
|
||
|
|
@click.option("--force", "-f", is_flag=True, help="DEPRECATE")
|
||
|
|
@click.option("--caller", "-c", help="Caller ID (service)")
|
||
|
|
@click.option("--no-ansi", is_flag=True, help="Do not print ANSI control characters")
|
||
|
|
@click.pass_context
|
||
|
|
def cli(ctx, force, caller, no_ansi):
|
||
|
|
try:
|
||
|
|
if (
|
||
|
|
no_ansi
|
||
|
|
or str(
|
||
|
|
os.getenv("PLATFORMIO_NO_ANSI", os.getenv("PLATFORMIO_DISABLE_COLOR"))
|
||
|
|
).lower()
|
||
|
|
== "true"
|
||
|
|
):
|
||
|
|
# pylint: disable=protected-access
|
||
|
|
click._compat.isatty = lambda stream: False
|
||
|
|
elif (
|
||
|
|
str(
|
||
|
|
os.getenv("PLATFORMIO_FORCE_ANSI", os.getenv("PLATFORMIO_FORCE_COLOR"))
|
||
|
|
).lower()
|
||
|
|
== "true"
|
||
|
|
):
|
||
|
|
# pylint: disable=protected-access
|
||
|
|
click._compat.isatty = lambda stream: True
|
||
|
|
except: # pylint: disable=bare-except
|
||
|
|
pass
|
||
|
|
|
||
|
|
maintenance.on_platformio_start(ctx, force, caller)
|
||
|
|
|
||
|
|
|
||
|
|
@cli.resultcallback()
|
||
|
|
@click.pass_context
|
||
|
|
def process_result(ctx, result, *_, **__):
|
||
|
|
maintenance.on_platformio_end(ctx, result)
|
||
|
|
|
||
|
|
|
||
|
|
@util.memoized()
|
||
|
|
def configure():
|
||
|
|
if CYGWIN:
|
||
|
|
raise exception.CygwinEnvDetected()
|
||
|
|
|
||
|
|
# https://urllib3.readthedocs.org
|
||
|
|
# /en/latest/security.html#insecureplatformwarning
|
||
|
|
try:
|
||
|
|
import urllib3 # pylint: disable=import-outside-toplevel
|
||
|
|
|
||
|
|
urllib3.disable_warnings()
|
||
|
|
except (AttributeError, ImportError):
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Handle IOError issue with VSCode's Terminal (Windows)
|
||
|
|
click_echo_origin = [click.echo, click.secho]
|
||
|
|
|
||
|
|
def _safe_echo(origin, *args, **kwargs):
|
||
|
|
try:
|
||
|
|
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 = lambda *args, **kwargs: _safe_echo(0, *args, **kwargs)
|
||
|
|
click.secho = lambda *args, **kwargs: _safe_echo(1, *args, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
def main(argv=None):
|
||
|
|
exit_code = 0
|
||
|
|
prev_sys_argv = sys.argv[:]
|
||
|
|
if argv:
|
||
|
|
assert isinstance(argv, list)
|
||
|
|
sys.argv = argv
|
||
|
|
try:
|
||
|
|
configure()
|
||
|
|
cli() # pylint: disable=no-value-for-parameter
|
||
|
|
except SystemExit as e:
|
||
|
|
if e.code and str(e.code).isdigit():
|
||
|
|
exit_code = int(e.code)
|
||
|
|
except Exception as e: # pylint: disable=broad-except
|
||
|
|
if not isinstance(e, exception.ReturnErrorCode):
|
||
|
|
maintenance.on_platformio_exception(e)
|
||
|
|
error_str = "Error: "
|
||
|
|
if isinstance(e, exception.PlatformioException):
|
||
|
|
error_str += str(e)
|
||
|
|
else:
|
||
|
|
error_str += format_exc()
|
||
|
|
error_str += """
|
||
|
|
============================================================
|
||
|
|
|
||
|
|
An unexpected error occurred. Further steps:
|
||
|
|
|
||
|
|
* Verify that you have the latest version of PlatformIO using
|
||
|
|
`pip install -U platformio` command
|
||
|
|
|
||
|
|
* Try to find answer in FAQ Troubleshooting section
|
||
|
|
https://docs.platformio.org/page/faq.html
|
||
|
|
|
||
|
|
* Report this problem to the developers
|
||
|
|
https://github.com/platformio/platformio-core/issues
|
||
|
|
|
||
|
|
============================================================
|
||
|
|
"""
|
||
|
|
click.secho(error_str, fg="red", err=True)
|
||
|
|
exit_code = int(str(e)) if str(e).isdigit() else 1
|
||
|
|
sys.argv = prev_sys_argv
|
||
|
|
return exit_code
|
||
|
|
|
||
|
|
|
||
|
|
def debug_gdb_main():
|
||
|
|
return main([sys.argv[0], "debug", "--interface", "gdb"] + sys.argv[1:])
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
sys.exit(main())
|