forked from platformio/platformio-core
106 lines
3.7 KiB
Python
106 lines
3.7 KiB
Python
# Copyright 2014-present Ivan Kravets <me@ikravets.com>
|
|
#
|
|
# 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 json
|
|
|
|
import click
|
|
|
|
from platformio.exception import APIRequestError
|
|
from platformio.managers.platform import PlatformManager
|
|
|
|
|
|
@click.command("boards", short_help="Pre-configured Embedded Boards")
|
|
@click.argument("query", required=False)
|
|
@click.option("--installed", is_flag=True)
|
|
@click.option("--json-output", is_flag=True)
|
|
def cli(query, installed, json_output): # pylint: disable=R0912
|
|
if json_output:
|
|
return _ouput_boards_json(query, installed)
|
|
|
|
BOARDLIST_TPL = ("{type:<30} {mcu:<14} {frequency:<8} "
|
|
" {flash:<7} {ram:<6} {name}")
|
|
terminal_width, _ = click.get_terminal_size()
|
|
|
|
grpboards = {}
|
|
for board in _get_boards(installed):
|
|
if board['platform'] not in grpboards:
|
|
grpboards[board['platform']] = []
|
|
grpboards[board['platform']].append(board)
|
|
|
|
for (platform, pboards) in sorted(grpboards.items()):
|
|
if query:
|
|
search_data = json.dumps(pboards).lower()
|
|
if query.lower() not in search_data.lower():
|
|
continue
|
|
|
|
click.echo("")
|
|
click.echo("Platform: ", nl=False)
|
|
click.secho(platform, bold=True)
|
|
click.echo("-" * terminal_width)
|
|
click.echo(BOARDLIST_TPL.format(
|
|
type=click.style("ID", fg="cyan"), mcu="MCU",
|
|
frequency="Frequency", flash="Flash", ram="RAM", name="Name"))
|
|
click.echo("-" * terminal_width)
|
|
|
|
for board in sorted(pboards, key=lambda b: b['id']):
|
|
if query:
|
|
search_data = "%s %s" % (
|
|
board['id'], json.dumps(board).lower())
|
|
if query.lower() not in search_data.lower():
|
|
continue
|
|
|
|
flash_size = "%dkB" % (board['rom'] / 1024)
|
|
|
|
ram_size = board['ram']
|
|
if ram_size >= 1024:
|
|
if ram_size % 1024:
|
|
ram_size = "%.1fkB" % (ram_size / 1024.0)
|
|
else:
|
|
ram_size = "%dkB" % (ram_size / 1024)
|
|
else:
|
|
ram_size = "%dB" % ram_size
|
|
|
|
click.echo(BOARDLIST_TPL.format(
|
|
type=click.style(board['id'], fg="cyan"),
|
|
mcu=board['mcu'],
|
|
frequency="%dMhz" % (board['fcpu'] / 1000000),
|
|
flash=flash_size, ram=ram_size, name=board['name']))
|
|
|
|
|
|
def _get_boards(installed=False):
|
|
boards = PlatformManager().get_installed_boards()
|
|
if not installed:
|
|
know_boards = ["%s:%s" % (b['platform'], b['id']) for b in boards]
|
|
for board in PlatformManager().get_registered_boards():
|
|
key = "%s:%s" % (board['platform'], board['id'])
|
|
if key not in know_boards:
|
|
boards.append(board)
|
|
return boards
|
|
|
|
|
|
def _ouput_boards_json(query, installed=False):
|
|
result = []
|
|
try:
|
|
boards = _get_boards(installed)
|
|
except APIRequestError:
|
|
if not installed:
|
|
boards = _get_boards(True)
|
|
for board in boards:
|
|
if query:
|
|
search_data = "%s %s" % (board['id'], json.dumps(board).lower())
|
|
if query.lower() not in search_data.lower():
|
|
continue
|
|
result.append(board)
|
|
click.echo(json.dumps(result))
|