From d2f772764dc4e54481ce68df9d4b2952011da0a9 Mon Sep 17 00:00:00 2001
From: Richard Meadows <richardeoin@gmail.com>
Date: Thu, 11 Feb 2016 18:14:09 +0000
Subject: [PATCH] [aprs-daemon] Update for write-to-file, additional callsign,
 colour printing etc

---
 firmware/tools/aprs_daemon.py | 68 ++++++++++++++++++++++++++---------
 1 file changed, 51 insertions(+), 17 deletions(-)

diff --git a/firmware/tools/aprs_daemon.py b/firmware/tools/aprs_daemon.py
index 90d621f..e268a34 100644
--- a/firmware/tools/aprs_daemon.py
+++ b/firmware/tools/aprs_daemon.py
@@ -1,3 +1,6 @@
+#!/usr/bin/env python
+# coding=utf-8
+
 """
 This script opens a tcp connection with and aprs-is server.
 
@@ -10,46 +13,77 @@ import aprslib
 from ukhas_format import *
 from habitat_upload import *
 from extract_backlog import *
+
+from colorama import *
 from datetime import datetime
 from math import log, exp
 
 # Regex for balloon callsign
-callsign_re = re.compile(r'M0SBU-(\d{1,2})')
+callsign_re = re.compile(r'(M0SBU|AD6AM)-(\d{1,2})')
 
 """
 Returns callsign for given SSID
 """
 def callsign_from_ssid(ssid):
-    if ssid == "11":
-        return "UBSEDS13"
+    if ssid == '11':
+        return 'UBSEDS14'
     else:
         return None
 
 """
 Attempts to extract a backlog frame and upload it
 """
-def extract_and_upload(packet, ssid):
+def extract_and_upload(packet, aprs_call, ssid):
+
+    # Callsign Lookup
+    callsign = callsign_from_ssid(ssid)
+    if callsign is None:
+        print Fore.RED + "No callsign match for {}-{}".format(aprs_call, ssid) + Fore.RESET
+        print
+        return
+    else:
+        print Fore.GREEN + "Packet from {} ({}-{}) ✓".format(callsign, aprs_call, ssid) + Fore.RESET
+        print
+
+
+    # Attempt to parse with aprslib
+    try:
+        pkt = aprslib.parse(packet)
+        print Fore.GREEN + "Parsed with aprslib ✓" + Fore.RESET
+
+        # Identify duplicate packets from telemetry sequence
+
+        # Save to raw data file
+        rawdata_filename = "rawdata-{}.txt".format(callsign.lower())
+        with open(rawdata_filename, 'a') as rawdata_file:
+            rawdata_file.write(packet)
+
+            print Fore.GREEN + "(wrote to {})".format(rawdata_filename) + Fore.RESET
+
+        print
+    except (aprslib.ParseError, aprslib.UnknownFormat) as exp:
+        print Fore.RED + "Error parsing with aprslib ✗" + Fore.RESET
+        print
+
+
+
+    # Backlog
     datum = extract_backlog_datum(packet)
 
     if datum: # valid backlog
         print
-        print "Extracted valid backlog from M0SBU-{}:".format(ssid)
+        print Fore.CYAN +  "Extracted valid backlog from {}-{}:".format(aprs_call, ssid)
         print_datum(datum)
-        print
-
-        callsign = "UBSEDS13"#callsign_from_ssid(ssid)
-        if callsign is None:
-            print "No callsign match for M0SBU-{}".format(ssid)
-            print
-            return
+        print Fore.RESET
 
+        # Habitat upload
         ukhas_str = ukhas_format(datum, callsign)
         print ukhas_str
         try:
-            print habitat_upload(datum['time'], ukhas_str)
+            print Fore.CYAN + habitat_upload(datum['time'], ukhas_str) + " ✓" + Fore.RESET
             print
         except:
-            print "Not accepted by habitat (duplicate?)"
+            print Fore.YELLOW + "Not accepted by habitat (duplicate?)" + Fore.RESET
             print
 
 
@@ -63,7 +97,7 @@ def callback(packet):
     match = callsign_re.match(packet)
 
     if match is not None:
-        extract_and_upload(packet, match.group(1))
+        extract_and_upload(packet, match.group(1), match.group(2))
 
 
 
@@ -77,9 +111,9 @@ print "debug level = 10"
 print "immortal = true"
 print
 
-# Get packets for all mike-zero users. Reasonably frequent stream but not enough to overload
+# Get packets for all mike-zero / alpha-delta-six users. Reasonably frequent stream but not enough to overload
 AIS = aprslib.IS("M0SBU-1", port=14580)
-AIS.set_filter("b/M0*")
+AIS.set_filter("b/M0*/AD6*")
 
 AIS.connect()
 # by default `raw` is False, then each line is ran through aprslib.parse()