forked from platformio/platformio-core
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
# Copyright (C) Ivan Kravets <me@ikravets.com>
|
|
# See LICENSE for details.
|
|
|
|
from os import listdir
|
|
from os.path import join
|
|
from sys import exit as sys_exit
|
|
from traceback import format_exc
|
|
|
|
import click
|
|
import requests
|
|
|
|
from platformio import __version__, exception, maintenance
|
|
from platformio.util import get_source_dir
|
|
|
|
|
|
class PlatformioCLI(click.MultiCommand): # pylint: disable=R0904
|
|
|
|
def list_commands(self, ctx):
|
|
cmds = []
|
|
for filename in listdir(join(get_source_dir(), "commands")):
|
|
if filename.startswith("__init__"):
|
|
continue
|
|
if filename.endswith(".py"):
|
|
cmds.append(filename[:-3])
|
|
cmds.sort()
|
|
return cmds
|
|
|
|
def get_command(self, ctx, name):
|
|
mod = None
|
|
try:
|
|
mod = __import__("platformio.commands." + name,
|
|
None, None, ["cli"])
|
|
except ImportError:
|
|
try:
|
|
return self._handle_obsolate_command(name)
|
|
except AttributeError:
|
|
raise exception.UnknownCLICommand(name)
|
|
return mod.cli
|
|
|
|
@staticmethod
|
|
def _handle_obsolate_command(name):
|
|
if name in ("install", "list", "search", "show", "uninstall"):
|
|
click.secho(
|
|
"Warning! `platformio %s` command is obsoleted and will be "
|
|
"removed in the next release! Please use "
|
|
"`platformio platforms %s` instead." % (name, name),
|
|
fg="red"
|
|
)
|
|
from platformio.commands import platforms
|
|
return getattr(platforms, "platforms_" + name)
|
|
raise AttributeError()
|
|
|
|
|
|
@click.command(cls=PlatformioCLI)
|
|
@click.version_option(__version__, prog_name="PlatformIO")
|
|
@click.option("--force", "-f", is_flag=True,
|
|
help="Force to accept any confirmation prompts")
|
|
@click.pass_context
|
|
def cli(ctx, force):
|
|
maintenance.on_platformio_start(ctx, force)
|
|
|
|
|
|
@cli.resultcallback()
|
|
@click.pass_context
|
|
def process_result(ctx, result, force): # pylint: disable=W0613
|
|
maintenance.on_platformio_end(ctx, result)
|
|
|
|
|
|
def main():
|
|
try:
|
|
# https://urllib3.readthedocs.org
|
|
# /en/latest/security.html#insecureplatformwarning
|
|
requests.packages.urllib3.disable_warnings()
|
|
|
|
cli(None, None)
|
|
except Exception as e: # pylint: disable=W0703
|
|
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()
|
|
click.secho(error_str, fg="red", err=True)
|
|
sys_exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys_exit(main())
|