mirror of
https://github.com/meshtastic/python.git
synced 2026-01-17 04:08:13 -05:00
108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
#!python3
|
|
|
|
import argparse
|
|
from . import StreamInterface, test
|
|
import logging
|
|
import sys
|
|
from pubsub import pub
|
|
import google.protobuf.json_format
|
|
|
|
"""The command line arguments"""
|
|
args = None
|
|
|
|
|
|
def onReceive(packet):
|
|
"""Callback invoked when a packet arrives"""
|
|
print(f"Received: {packet}")
|
|
|
|
|
|
def onConnection(interface, topic=pub.AUTO_TOPIC):
|
|
"""Callback invoked when we connect/disconnect from a radio"""
|
|
print(f"Connection changed: {topic.getName()}")
|
|
global args
|
|
if topic.getName() == "meshtastic.connection.established":
|
|
try:
|
|
if args.setpref:
|
|
name = args.setpref[0]
|
|
val = int(args.setpref[1])
|
|
setattr(interface.radioConfig.preferences, name, val)
|
|
print("Writing modified preferences to device...")
|
|
interface.writeConfig()
|
|
|
|
if args.info:
|
|
print(interface.myInfo)
|
|
print(interface.radioConfig)
|
|
print("Nodes in mesh:")
|
|
for n in interface.nodes.values():
|
|
asDict = google.protobuf.json_format.MessageToJson(n)
|
|
print(asDict)
|
|
except Exception as ex:
|
|
print(ex)
|
|
|
|
if args.info or args.setpref:
|
|
interface.close() # after running command then exit
|
|
|
|
|
|
def onNode(node):
|
|
"""Callback invoked when the node DB changes"""
|
|
print(f"Node changed: {node}")
|
|
|
|
|
|
def subscribe():
|
|
"""Subscribe to the topics the user probably wants to see, prints output to stdout"""
|
|
pub.subscribe(onReceive, "meshtastic.receive")
|
|
pub.subscribe(onConnection, "meshtastic.connection")
|
|
pub.subscribe(onNode, "meshtastic.node")
|
|
|
|
|
|
def main():
|
|
"""Perform command line meshtastic operations"""
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument(
|
|
"--device",
|
|
help="The port the Meshtastic device is connected to, i.e. /dev/ttyUSB0. If unspecified, we'll try to find it.",
|
|
default=None)
|
|
|
|
parser.add_argument(
|
|
"--seriallog",
|
|
help="Log device serial output to either 'stdout', 'none' or a filename to append to. Defaults to stdout.",
|
|
default="stdout")
|
|
|
|
parser.add_argument("--info", help="Read and display the radio config information",
|
|
action="store_true")
|
|
|
|
parser.add_argument("--setpref", help="Set a preferences field", nargs=2)
|
|
|
|
parser.add_argument("--debug", help="Show API library debug log messages",
|
|
action="store_true")
|
|
|
|
parser.add_argument("--test", help="Run stress test against all connected Meshtastic devices",
|
|
action="store_true")
|
|
|
|
global args
|
|
args = parser.parse_args()
|
|
logging.basicConfig(level=logging.DEBUG if args.debug else logging.INFO)
|
|
|
|
if args.info or args.setpref:
|
|
args.seriallog = "none" # assume no debug output in this case
|
|
|
|
if args.test:
|
|
test.testAll()
|
|
else:
|
|
if args.seriallog == "stdout":
|
|
logfile = sys.stdout
|
|
elif args.seriallog == "none":
|
|
logging.debug("Not logging serial output")
|
|
logfile = None
|
|
else:
|
|
logging.info(f"Logging serial output to {args.seriallog}")
|
|
logfile = open(args.seriallog, 'w+', buffering=1) # line buffering
|
|
|
|
subscribe()
|
|
client = StreamInterface(args.device, debugOut=logfile)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|