From a95cdeb08657ebb901a7a9dbd9d7eb09a2f04640 Mon Sep 17 00:00:00 2001 From: SpudGunMan Date: Sat, 11 Oct 2025 12:35:58 -0700 Subject: [PATCH] -pickle or print either way --- script/addFav.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/script/addFav.py b/script/addFav.py index 7858246..ef7d3a5 100644 --- a/script/addFav.py +++ b/script/addFav.py @@ -4,11 +4,28 @@ import sys import os import pickle +import argparse favList = [] roofNodeList = [] roof_node = False +if __name__ == "__main__": + parser = argparse.ArgumentParser(description="Add favorite nodes or print pickle contents.") + parser.add_argument('-pickle', '-p', action='store_true', help="Print the contents of roofNodeList.pkl and exit") + args = parser.parse_args() + + if args.pickle: + try: + with open('roofNodeList.pkl', 'rb') as f: + data = pickle.load(f) + #print a simple list of nodeID:x\n + for item in data: + print(f"{item.get('nodeID', 'N/A')}") + except Exception as e: + print(f"Error reading roofNodeList.pkl: {e}") + exit(0) + # welcome header print("meshing-around: addFav - Auto-Add favorite nodes to all interfaces from config.ini data") print("---------------------------------------------------------------")