mirror of
https://github.com/OMGeeky/flucto-heisskleber.git
synced 2025-12-31 08:40:26 +01:00
1d1e762912
* Add verbose and pretty flags to hkcli. Adds testing for console sink. * Add argparse help strings to hkcli.
94 lines
2.2 KiB
Python
94 lines
2.2 KiB
Python
import argparse
|
|
import sys
|
|
from typing import Union
|
|
|
|
from heisskleber.config import load_config
|
|
from heisskleber.console.sink import ConsoleSink
|
|
from heisskleber.core.factories import _registered_sources
|
|
|
|
TopicType = Union[str, list[str]]
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(
|
|
prog="hkcli",
|
|
description="Heisskleber command line interface",
|
|
usage="%(prog)s [options]",
|
|
)
|
|
parser.add_argument(
|
|
"-t",
|
|
"--type",
|
|
type=str,
|
|
choices=["zmq", "mqtt", "serial", "udp"],
|
|
default="zmq",
|
|
)
|
|
parser.add_argument(
|
|
"-T",
|
|
"--topic",
|
|
type=str,
|
|
default="#",
|
|
help="Topic to subscribe to, valid for zmq and mqtt only.",
|
|
)
|
|
parser.add_argument(
|
|
"-H",
|
|
"--host",
|
|
type=str,
|
|
default="localhost",
|
|
help="Host or broker for MQTT, zmq and UDP.",
|
|
)
|
|
parser.add_argument(
|
|
"-P",
|
|
"--port",
|
|
type=int,
|
|
default=1883,
|
|
help="Port or serial interface for MQTT, zmq and UDP.",
|
|
)
|
|
parser.add_argument("-v", "--verbose", action="store_true")
|
|
parser.add_argument("-p", "--pretty", action="store_true", help="Pretty print JSON data.")
|
|
|
|
return parser.parse_args()
|
|
|
|
|
|
def run() -> None:
|
|
args = parse_args()
|
|
sink = ConsoleSink(pretty=args.pretty, verbose=args.verbose)
|
|
|
|
sub_cls, conf_cls = _registered_sources[args.type]
|
|
|
|
try:
|
|
config = load_config(conf_cls(), args.type, read_commandline=False)
|
|
except FileNotFoundError:
|
|
print(f"Using default config for {args.type}.")
|
|
config = conf_cls()
|
|
|
|
if args.port:
|
|
config.port = args.port
|
|
|
|
if args.host:
|
|
if args.type == "mqtt":
|
|
config.broker = args.host
|
|
elif args.type == "zmq":
|
|
config.interface = args.host
|
|
elif args.type == "udp":
|
|
config.ip = args.host
|
|
|
|
if args.type == "zmq" and args.topic == "#":
|
|
args.topic = ""
|
|
|
|
source = sub_cls(config, args.topic)
|
|
while True:
|
|
topic, data = source.receive()
|
|
sink.send(data, topic)
|
|
|
|
|
|
def main() -> None:
|
|
try:
|
|
run()
|
|
except KeyboardInterrupt:
|
|
print("Exiting...")
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|