First pass - override ident TG

identtg
Simon 4 years ago
parent 6136aff819
commit 8aecf33495

@ -2814,7 +2814,7 @@ if __name__ == '__main__':
if cli_args.LOG_LEVEL: if cli_args.LOG_LEVEL:
CONFIG['LOGGER']['LOG_LEVEL'] = cli_args.LOG_LEVEL CONFIG['LOGGER']['LOG_LEVEL'] = cli_args.LOG_LEVEL
logger = log.config_logging(CONFIG['LOGGER']) logger = log.config_logging(CONFIG['LOGGER'])
logger.info('\n\nCopyright (c) 2020, 2021 Simon G7RZU simon@gb7fr.org.uk') logger.info('\n\nCopyright (c) 2020, 2021, 2022 Simon G7RZU simon@gb7fr.org.uk')
logger.info('Copyright (c) 2013, 2014, 2015, 2016, 2018, 2019\n\tThe Regents of the K0USY Group. All rights reserved.\n') logger.info('Copyright (c) 2013, 2014, 2015, 2016, 2018, 2019\n\tThe Regents of the K0USY Group. All rights reserved.\n')
logger.debug('(GLOBAL) Logging system started, anything from here on gets logged') logger.debug('(GLOBAL) Logging system started, anything from here on gets logged')

@ -27,6 +27,8 @@ from twisted.protocols.basic import NetstringReceiver
from reporting_const import * from reporting_const import *
from pprint import pprint
class reportClient(NetstringReceiver): class reportClient(NetstringReceiver):
def stringReceived(self, data): def stringReceived(self, data):
@ -34,9 +36,11 @@ class reportClient(NetstringReceiver):
if data[:1] == REPORT_OPCODES['BRDG_EVENT']: if data[:1] == REPORT_OPCODES['BRDG_EVENT']:
self.bridgeEvent(data[1:].decode('UTF-8')) self.bridgeEvent(data[1:].decode('UTF-8'))
elif data[:1] == REPORT_OPCODES['CONFIG_SND']: elif data[:1] == REPORT_OPCODES['CONFIG_SND']:
self.configSend(data[1:]) if cli_args.CONFIG:
self.configSend(data[1:])
elif data[:1] == REPORT_OPCODES['BRIDGE_SND']: elif data[:1] == REPORT_OPCODES['BRIDGE_SND']:
self.bridgeSend(data[1:]) if cli_args.BRIDGES:
self.bridgeSend(data[1:])
elif data == b'bridge updated': elif data == b'bridge updated':
pass pass
else: else:
@ -60,15 +64,19 @@ class reportClient(NetstringReceiver):
if len(datalist) > 9: if len(datalist) > 9:
event['duration'] = datalist[9] event['duration'] = datalist[9]
print(event) if cli_args.EVENTS:
pprint(event, compact=True)
def bridgeSend(self,data): def bridgeSend(self,data):
self.BRIDGES = pickle.loads(data) self.BRIDGES = pickle.loads(data)
print(self.BRIDGES) if cli_args.WATCH and cli_args.WATCH in self.BRIDGES:
pprint(self.BRIDGES[cli_args.WATCH], compact=True)
else:
pprint(self.BRIDGES, compact=True, indent=4)
def configSend(self,data): def configSend(self,data):
self.CONFIG = pickle.loads(data) self.CONFIG = pickle.loads(data)
print(self.CONFIG) pprint(self.CONFIG, compact=True)
@ -100,6 +108,7 @@ if __name__ == '__main__':
import signal import signal
import sys import sys
import os import os
import argparse
#Set process title early #Set process title early
setproctitle(__file__) setproctitle(__file__)
@ -110,10 +119,22 @@ if __name__ == '__main__':
def sig_handler(_signal, _frame): def sig_handler(_signal, _frame):
print('SHUTDOWN: TERMINATING WITH SIGNAL {}'.format(str(_signal))) print('SHUTDOWN: TERMINATING WITH SIGNAL {}'.format(str(_signal)))
reactor.stop() reactor.stop()
# CLI argument parser - handles picking up the config file from the command line, and sending a "help" message
parser = argparse.ArgumentParser()
parser.add_argument('-e', '--events', action='store', dest='EVENTS', help='print events [0|1]')
parser.add_argument('-c', '--config', action='store', dest='CONFIG', help='print config [0|1]')
parser.add_argument('-b', '--bridges', action='store', dest='BRIDGES', help='print bridges [0|1]')
parser.add_argument('-w', '--watch', action='store', dest='WATCH', help='watch bridge <name>')
parser.add_argument('-o', '--host', action='store', dest='HOST', help='host to connect to <ip address>')
parser.add_argument('-p', '--port', action='store', dest='PORT', help='port to connect to <port>')
cli_args = parser.parse_args()
# Set signal handers so that we can gracefully exit if need be # Set signal handers so that we can gracefully exit if need be
for sig in [signal.SIGINT, signal.SIGTERM]: for sig in [signal.SIGINT, signal.SIGTERM]:
signal.signal(sig, sig_handler) signal.signal(sig, sig_handler)
reactor.connectTCP(sys.argv[1],int(sys.argv[2]), reportClientFactory(reportClient)) reactor.connectTCP(cli_args.HOST,int(cli_args.PORT), reportClientFactory(reportClient))
reactor.run() reactor.run()

Loading…
Cancel
Save

Powered by TurnKey Linux.