Update EGI Code

GPS is now non-blocking in a thread
Current GPS state is now reported (fix vs no fix) in json ('fix')
Exception logging for bad environment variables and MQTT bus
30 second heartbeat of state array

egi/egi_mqtt.py

@@ -9,7 +9,9 @@
 import argparse
 import logging
 import coloredlogs
+import threading
 
+gpsd = None #seting the global variable
 Active = True
 styles = {'critical': {'bold': True, 'color': 'red'}, 'debug': {'color': 'green'}, 'error': {'color': 'red'}, 'info': {'color': 'white'}, 'notice': {'color': 'magenta'}, 'spam': {'color': 'green', 'faint': True}, 'success': {'bold': True, 'color': 'green'}, 'verbose': {'color': 'blue'}, 'warning': {'color': 'yellow'}}
 level = logging.INFO 
@@ -26,9 +28,24 @@
 config = {}
 config['Local'] = ["127.0.0.1", "skyscan/egi", "Local MQTT Bus"] # updated based on naming convention here: https://www.hivemq.com/blog/mqtt-essentials-part-5-mqtt-topics-best-practices/
 timeTrigger = time.mktime(time.gmtime()) + 10
-timeHeartbeat = 0
+timeHeartbeat = time.mktime(time.gmtime()) + 10
 ID = str(random.randint(1,100001))
-gpsd = gps(mode=WATCH_ENABLE|WATCH_NEWSTYLE) 
+
+
+class GpsPoller(threading.Thread):
+ def __init__(self):
+ threading.Thread.__init__(self)
+ global gpsd #bring it in scope
+ gpsd = gps(mode=WATCH_ENABLE|WATCH_NEWSTYLE)
+ self.current_value = None
+ self.running = True #setting the thread running to true
+
+ def run(self):
+ global gpsd
+ while self.running:
+ gpsd.next() #this will continue to loop and grab EACH set of gpsd info to clear the buffer
+
+
 LLA = [38.9510808,-77.3841834,130.1337]
 RPY = [0,0,0]
 
@@ -40,7 +57,12 @@
 parser.add_argument('-r', '--roll', help="Roll Angle of Camera (degrees)", default=RPY[0])
 parser.add_argument('-p', '--pitch', help="Pitch Angle of Camera (degrees)", default=RPY[1])
 parser.add_argument('-y', '--yaw', help="Yaw Angle of Camera (degrees from True North)", default=RPY[2])
-args = parser.parse_args()
+try:
+ args = parser.parse_args()
+except:
+ logging.critical("Error in Command Line Argument Parsing. Are all environment variables set?", exc_info=True)
+ raise
+
 
 state = {}
 state['time'] = time.strftime("%Y-%m-%dT%H:%M:%SZ",time.gmtime())
@@ -50,6 +72,7 @@
 state['roll'] = float(args.roll)
 state['pitch'] = float(args.pitch)
 state['yaw'] = float(args.yaw)
+state['fix'] = 0
 logging.info("Initial State Array: " + str(state))
 
 
@@ -75,22 +98,40 @@ def on_disconnect(client, userdata, rc):
 clientLocal = mqtt.Client("EGI-"+ID) #create new instance
 clientLocal.on_message = on_message_local #attach function to callback
 clientLocal.on_disconnect = on_disconnect
-clientLocal.connect(broker_address) #connect to broker
+try:
+ clientLocal.connect(broker_address) #connect to broker
+except:
+ logging.critical("Could not connect to MQTT Broker.", exc_info=True)
+ raise
 clientLocal.loop_start() #start the loop
 clientLocal.publish("skyscan/registration","EGI-"+ID+" Registration")
 
-#############################################
-## Main Loop ##
-#############################################
-while Active:
- if timeTrigger < time.mktime(time.gmtime()):
- timeTrigger = time.mktime(time.gmtime()) + 10
- clientLocal.publish(local_topic,json.dumps(state))
-
- report = gpsd.next() #
- if report['class'] == 'TPV':
- state['time'] = getattr(report,'time',time.strftime("%Y-%m-%dT%H:%M:%SZ",time.gmtime()))
- state['lat'] = getattr(report,'lat',LLA[0])
- state['long'] = getattr(report,'lon',LLA[1])
- state['alt'] = getattr(report,'alt',LLA[2])
- time.sleep(0.01)
+gpsp = GpsPoller() # create the thread
+try:
+ gpsp.start() # start it up
+ #############################################
+ ## Main Loop ##
+ #############################################
+ while Active:
+ state['fix'] = gpsd.fix.status
+ if gpsd.fix.status:
+ state['time'] = gpsd.fix.time
+ state['lat'] = gpsd.fix.latitude
+ state['long'] = gpsd.fix.longitude
+ state['alt'] = gpsd.fix.altitude
+ if timeTrigger < time.mktime(time.gmtime()):
+ timeTrigger = time.mktime(time.gmtime()) + 10
+ clientLocal.publish(local_topic,json.dumps(state))
+ if timeHeartbeat < time.mktime(time.gmtime()):
+ timeHeartbeat = time.mktime(time.gmtime()) + 30
+ logging.info("Current EGI State: " + json.dumps(state))
+ time.sleep(0.01)
+except (KeyboardInterrupt, SystemExit): #when you press ctrl+c
+ logging.info("Killing GPS Thread...")
+ gpsp.running = False
+ gpsp.join(2)
+except:
+ logging.critical("Error starting GPS.", exc_info=True)
+ gpsp.running = False
+ gpsp.join(2)
+ raise