Skip to content

Commit

Permalink
pw_hdlc: Remove rpc_console.py
Browse files Browse the repository at this point in the history
Remove rpc_console.py and point users to pw_system.console.

Remove pw_console and ipython deps from pw_hdlc. They were only used
for rpc_console.py which is deprecated in favor of
https://cs.opensource.google/pigweed/pigweed/+/main:pw_system/py/pw_system/console.py

Create an additional Python distribution target containing only pw_hdlc,
pw_protobuf_compiler and pw_rpc.

Test: gn gen out
Test: ninja -C out \
Test: pw_env_setup:generate_pigweed_python_package_with_only_hdlc_proto_rpc_tokenizer._build_wheel
Test: cd out/obj/pw_env_setup
Test: python3 -m venv venv
Test: . ./venv/bin/activate
Test: cd generate_pigweed_python_package_with_only_hdlc_proto_rpc_tokenizer
Test: pip install .
Change-Id: I08497f9a54239b74096990607247c505f0038560
Reviewed-on: https://pigweed-review.googlesource.com/c/pigweed/pigweed/+/102280
Commit-Queue: Auto-Submit <auto-submit@pigweed.google.com.iam.gserviceaccount.com>
Reviewed-by: Wyatt Hepler <hepler@google.com>
Pigweed-Auto-Submit: Anthony DiGirolamo <tonymd@google.com>
  • Loading branch information
AnthonyDiGirolamo authored and CQ Bot Account committed Jul 29, 2022
1 parent 71b85f8 commit 4441396
Show file tree
Hide file tree
Showing 7 changed files with 60 additions and 291 deletions.
25 changes: 25 additions & 0 deletions pw_env_setup/BUILD.gn
Original file line number Diff line number Diff line change
Expand Up @@ -183,6 +183,31 @@ if (pw_build_USE_NEW_PYTHON_BUILD) {
]
}

_hdlc_proto_rpc_tokenizer = [
"$dir_pw_hdlc/py",
"$dir_pw_protobuf_compiler/py",
"$dir_pw_rpc/py",
"$dir_pw_tokenizer/py",
]

# Create a Python distributeable with just pw_hdlc, pw_protobuf_compiler,
# pw_rpc, pw_tokenizer and their dependencies.
pw_create_python_source_tree(
"generate_pigweed_python_package_with_only_hdlc_proto_rpc_tokenizer") {
packages = _hdlc_proto_rpc_tokenizer
public_deps = _hdlc_proto_rpc_tokenizer

generate_setup_cfg = {
common_config_file = "pypi_common_setup.cfg"
append_date_to_version = true
}
extra_files = [
"$dir_pigweed/LICENSE > LICENSE",
"$dir_pigweed/README.md > README.md",
"pypi_pyproject.toml > pyproject.toml",
]
}

# This pip installs the generate_pigweed_python_package
pw_internal_pip_install("pip_install_pigweed_package") {
packages = [ ":generate_pigweed_python_package" ]
Expand Down
3 changes: 0 additions & 3 deletions pw_hdlc/py/BUILD.gn
Original file line number Diff line number Diff line change
Expand Up @@ -35,11 +35,8 @@ pw_python_package("py") {
"encode_test.py",
]
python_deps = [
"$dir_pw_cli/py",
"$dir_pw_console/py",
"$dir_pw_protobuf_compiler/py",
"$dir_pw_rpc/py",
"$dir_pw_tokenizer/py",
]
python_test_deps = [
"$dir_pw_build/py",
Expand Down
285 changes: 6 additions & 279 deletions pw_hdlc/py/pw_hdlc/rpc_console.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2020 The Pigweed Authors
# Copyright 2022 The Pigweed Authors
#
# 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
Expand All @@ -13,293 +13,20 @@
# the License.
"""Console for interacting with pw_rpc over HDLC.
To start the console, provide a serial port as the --device argument and paths
or globs for .proto files that define the RPC services to support:
This command is no longer supported. Please run pw_system.console instead.
python -m pw_hdlc.rpc_console --device /dev/ttyUSB0 sample.proto
This starts an IPython console for communicating with the connected device. A
few variables are predefined in the interactive console. These include:
rpcs - used to invoke RPCs
device - the serial device used for communication
client - the pw_rpc.Client
protos - protocol buffer messages indexed by proto package
An example echo RPC command:
rpcs.pw.rpc.EchoService.Echo(msg="hello!")
python -m pw_system.console --device /dev/ttyUSB0 --proto-globs sample.proto
"""

import argparse
import glob
from inspect import cleandoc
import logging
from pathlib import Path
import sys
from types import ModuleType
from typing import (
Any,
BinaryIO,
Collection,
Iterable,
Iterator,
List,
Optional,
Union,
)
import socket

import serial # type: ignore

import pw_cli.log
import pw_console.python_logging
from pw_console import PwConsoleEmbed
from pw_console.pyserial_wrapper import SerialWithLogging
from pw_console.plugins.bandwidth_toolbar import BandwidthToolbar

from pw_log.proto import log_pb2
from pw_rpc.console_tools.console import ClientInfo, flattened_rpc_completions
from pw_rpc import callback_client
from pw_tokenizer.database import LoadTokenDatabases
from pw_tokenizer.detokenize import Detokenizer, detokenize_base64
from pw_tokenizer import tokens

from pw_hdlc.rpc import HdlcRpcClient, default_channels

_LOG = logging.getLogger(__name__)
_DEVICE_LOG = logging.getLogger('rpc_device')

# TODO(tonymd): Delete this when no longer needed.
PW_RPC_MAX_PACKET_SIZE = 256
SOCKET_SERVER = 'localhost'
SOCKET_PORT = 33000
MKFIFO_MODE = 0o666


def _parse_args():
"""Parses and returns the command line arguments."""
parser = argparse.ArgumentParser(description=__doc__)
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument('-d', '--device', help='the serial port to use')
parser.add_argument('-b',
'--baudrate',
type=int,
default=115200,
help='the baud rate to use')
parser.add_argument(
'--serial-debug',
action='store_true',
help=('Enable debug log tracing of all data passed through'
'pyserial read and write.'))
parser.add_argument(
'-o',
'--output',
type=argparse.FileType('wb'),
default=sys.stdout.buffer,
help=('The file to which to write device output (HDLC channel 1); '
'provide - or omit for stdout.'))
parser.add_argument('--logfile', help='Console debug log file.')
group.add_argument('-s',
'--socket-addr',
type=str,
help='use socket to connect to server, type default for\
localhost:33000, or manually input the server address:port')
parser.add_argument("--token-databases",
metavar='elf_or_token_database',
nargs="+",
action=LoadTokenDatabases,
help="Path to tokenizer database csv file(s).")
parser.add_argument('--config-file',
type=Path,
help='Path to a pw_console yaml config file.')
parser.add_argument('--proto-globs',
nargs='+',
help='glob pattern for .proto files')
return parser.parse_args()


def _expand_globs(globs: Iterable[str]) -> Iterator[Path]:
for pattern in globs:
for file in glob.glob(pattern, recursive=True):
yield Path(file)


def _start_ipython_terminal(client: HdlcRpcClient,
serial_debug: bool = False,
config_file_path: Optional[Path] = None) -> None:
"""Starts an interactive IPython terminal with preset variables."""
local_variables = dict(
client=client,
device=client.client.channel(1),
rpcs=client.client.channel(1).rpcs,
protos=client.protos.packages,
# Include the active pane logger for creating logs in the repl.
DEVICE_LOG=_DEVICE_LOG,
LOG=logging.getLogger(),
)

welcome_message = cleandoc("""
Welcome to the Pigweed Console!
Help: Press F1 or click the [Help] menu
To move focus: Press Shift-Tab or click on a window
Example Python commands:
device.rpcs.pw.rpc.EchoService.Echo(msg='hello!')
LOG.warning('Message appears in Host Logs window.')
DEVICE_LOG.warning('Message appears in Device Logs window.')
""")

client_info = ClientInfo('device',
client.client.channel(1).rpcs, client.client)
completions = flattened_rpc_completions([client_info])

log_windows = {
'Device Logs': [_DEVICE_LOG],
'Host Logs': [logging.getLogger()],
}
if serial_debug:
log_windows['Serial Debug'] = [
logging.getLogger('pw_console.serial_debug_logger')
]

interactive_console = PwConsoleEmbed(
global_vars=local_variables,
local_vars=None,
loggers=log_windows,
repl_startup_message=welcome_message,
help_text=__doc__,
config_file_path=config_file_path,
)
interactive_console.hide_windows('Host Logs')
interactive_console.add_sentence_completer(completions)
if serial_debug:
interactive_console.add_bottom_toolbar(BandwidthToolbar())

# Setup Python logger propagation
interactive_console.setup_python_logging()

# Don't send device logs to the root logger.
_DEVICE_LOG.propagate = False

interactive_console.embed()


class SocketClientImpl:
def __init__(self, config: str):
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
socket_server = ''
socket_port = 0

if config == 'default':
socket_server = SOCKET_SERVER
socket_port = SOCKET_PORT
else:
socket_server, socket_port_str = config.split(':')
socket_port = int(socket_port_str)
self.socket.connect((socket_server, socket_port))

def write(self, data: bytes):
self.socket.sendall(data)

def read(self, num_bytes: int = PW_RPC_MAX_PACKET_SIZE):
return self.socket.recv(num_bytes)


def console(device: str,
baudrate: int,
proto_globs: Collection[str],
token_databases: Collection[tokens.Database],
socket_addr: str,
logfile: str,
output: Any,
serial_debug: bool = False,
config_file: Optional[Path] = None) -> int:
"""Starts an interactive RPC console for HDLC."""
# argparse.FileType doesn't correctly handle '-' for binary files.
if output is sys.stdout:
output = sys.stdout.buffer

if not logfile:
# Create a temp logfile to prevent logs from appearing over stdout. This
# would corrupt the prompt toolkit UI.
logfile = pw_console.python_logging.create_temp_log_file()
pw_cli.log.install(logging.INFO, True, False, logfile)

detokenizer = None
if token_databases:
detokenizer = Detokenizer(tokens.Database.merged(*token_databases),
show_errors=False)

if not proto_globs:
proto_globs = ['**/*.proto']

protos: List[Union[ModuleType, Path]] = list(_expand_globs(proto_globs))

# Append compiled log.proto library to avoid include errors when manually
# provided, and shadowing errors due to ordering when the default global
# search path is used.
protos.append(log_pb2)

if not protos:
_LOG.critical('No .proto files were found with %s',
', '.join(proto_globs))
_LOG.critical('At least one .proto file is required')
return 1

_LOG.debug('Found %d .proto files found with %s', len(protos),
', '.join(proto_globs))

serial_impl = serial.Serial
if serial_debug:
serial_impl = SerialWithLogging

if socket_addr is None:
serial_device = serial_impl(
device,
baudrate,
timeout=0, # Non-blocking mode
)
read = lambda: serial_device.read(8192)
write = serial_device.write
else:
try:
socket_device = SocketClientImpl(socket_addr)
read = socket_device.read
write = socket_device.write
except ValueError:
_LOG.exception('Failed to initialize socket at %s', socket_addr)
return 1

callback_client_impl = callback_client.Impl(
default_unary_timeout_s=5.0,
default_stream_timeout_s=None,
)
_start_ipython_terminal(
HdlcRpcClient(read,
protos,
default_channels(write),
lambda data: detokenize_and_write_to_output(
data, output, detokenizer),
client_impl=callback_client_impl), serial_debug,
config_file)
return 0


def detokenize_and_write_to_output(data: bytes,
unused_output: BinaryIO = sys.stdout.buffer,
detokenizer=None):
log_line = data
if detokenizer:
log_line = detokenize_base64(detokenizer, data)

for line in log_line.decode(errors="surrogateescape").splitlines():
_DEVICE_LOG.info(line)


def main() -> int:
return console(**vars(_parse_args()))
print(__doc__)
return 1


if __name__ == '__main__':
Expand Down
2 changes: 0 additions & 2 deletions pw_hdlc/py/setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,6 @@ description = Tools for Encoding/Decoding data using the HDLC protocol
[options]
packages = find:
zip_safe = False
install_requires =
ipython

[options.package_data]
pw_hdlc = py.typed
6 changes: 3 additions & 3 deletions pw_hdlc/rpc_example/docs.rst
Original file line number Diff line number Diff line change
Expand Up @@ -52,14 +52,14 @@ replacing ``/dev/ttyACM0`` with the correct serial device for your board.

.. code-block:: text
$ python -m pw_hdlc.rpc_console --device /dev/ttyACM0
$ python -m pw_system.console --device /dev/ttyACM0
Console for interacting with pw_rpc over HDLC.
To start the console, provide a serial port as the --device argument and paths
or globs for .proto files that define the RPC services to support:
python -m pw_hdlc.rpc_console --device /dev/ttyUSB0 sample.proto
python -m pw_system.console --device /dev/ttyUSB0 --proto-globs pw_rpc/echo.proto
This starts an IPython console for communicating with the connected device. A
few variables are predefined in the interactive console. These include:
Expand Down Expand Up @@ -123,7 +123,7 @@ Run pw_rpc client (i.e. use echo.proto)

.. code-block:: sh
python -m pw_hdlc.rpc_console path/to/echo.proto -s localhost:33000
python -m pw_system.console path/to/echo.proto -s localhost:33000
Run pw_rpc server

Expand Down
Loading

0 comments on commit 4441396

Please sign in to comment.