aprs_tool/kiss_and_db.py

#!/usr/bin/env python3
import os
import sqlite3
import aprs
import json
import aprslib
import configparser
import time
import logging
from apscheduler.schedulers.asyncio import AsyncIOScheduler
import time

def read_config():
    config = configparser.ConfigParser()
    config.read('config.ini')
    return config

def get_db_connection():
    conn = sqlite3.connect('database.db')
    conn.row_factory = sqlite3.Row
    return conn

def refresh_kiss_connection(kiss_conn):
    logging.debug("Restarting KISS connection on schedule")
    logging.debug("Stopping current connection")
    kiss_conn.stop()
    logging.debug("Waiting 5 seconds")
    time.sleep(5)
    logging.debug("Starting new connection")
    kiss_conn.start()

def main():

    # Add the call and location of this station to the packet info
    config = read_config()
    # MYCALL = os.environ.get("MYCALL", "W1CDN")
    # KISS_HOST = os.environ.get("KISS_HOST", "192.168.0.30")
    # KISS_PORT = os.environ.get("KISS_PORT", "8001")

    logging.basicConfig(filename=config['Settings']['log_path'], level=logging.DEBUG, \
                        format='%(asctime)s - %(message)s')
    logging.debug('kiss_and_db.py running')

    ki = aprs.TCPKISS(host=config['Settings']['kiss_host'], port=int(config['Settings']['kiss_port']))
    ki.start()

    scheduler = AsyncIOScheduler()
    scheduler.add_job(refresh_kiss_connection, 'interval', hours = 1, args = [ki])
    scheduler.start()

    # Make a simple frame and send it
    frame = aprs.APRSFrame.ui(
        destination="APZ001",
        source=config['Settings']['mycall'],
        path=["WIDE1-1"],
        info=b">Hello World!",
    )
    #ki.write(frame)

    # Watch for new packets to come in
    while True:
        conn = get_db_connection()
        for frame in ki.read(min_frames=1):
            logging.debug("New packet, trying to parse")
            try:
                a = aprslib.parse(str(frame))
                a['station_call'] = config['Settings']['station_call']
                a['station_lat'] = config['Settings']['station_lat']
                a['station_lon'] = config['Settings']['station_lon']
                a['created_unix'] = int(time.time())
                print(a)
                # Make this a string and deal with it later (probably a mistake)
                a['path'] = str(a['path'])
                # Store true/false as 1/0
                if 'alive' in a:
                    if a['alive'] == True:
                        a['alive'] = 1
                    else:
                        a['alive'] = 0
                # Build an INSERT statement based on the fields we have from the frame
                attrib_names = ', '.join('"%s"' % w for w in a.keys())
                attrib_values = ", ".join("?" * len(a.keys()))

                logging.debug("Inserting into database")
                try:
                    # Insert data
                    sql = "INSERT INTO frames ("+attrib_names+") VALUES ("+attrib_values+")"
                    conn.execute(sql, list(a.values()))
                    logging.debug("Frames table updated")
                    # TODO update stations table here
                    # Original intent was to include the id from the frames table,
                    # but that would mean making another query.
                    # It's not immediately needed, so I'm skipping it.
                    # Build query
                    # "from" is wrappedin [] because it is a reserved word and using '' doesn't work.
                    # https://www.sqlite.org/lang_keywords.html
                    #try:
                    station_update = "'"+a['from'] +"', '"+ str(a['created_unix']) +"', '1'"
                    query3 = "INSERT INTO stations ([from], last_heard_unix, count) \
                    VALUES("+station_update+") \
                    ON CONFLICT([from]) \
                    DO UPDATE SET count = count + 1,\
                                  last_heard_unix = excluded.last_heard_unix;"
                    # Insert/update data
                    conn.execute(query3)
                    logging.debug("Station table updated")
                    conn.commit()
                    #except:
                     #   print("Stations table couldn't be updated.")

                    # TODO remove packets that are older ('created') than a limit set in config.ini
                    # "5 minutes" also works
                    #conn.execute("DELETE FROM frames WHERE created < DATETIME('now', '"+config['Settings']['keep_time']+"')")
                    #conn.commit()
                except:
                    #print("Error with SQLite!")
                    logging.error("Error with SQLite!")
            except:
                #print("Frame could not be parsed.")
                logging.error("Frame could not be parsed.")


        conn.close()



if __name__ == "__main__":
    main()