status.py

#!/usr/bin/env python3

import sys
import time
import json
import arrow
import bluetooth._bluetooth as bluez
from pprint import pprint
from logger import open_bluetooth, process_packet

def save(data):
    with open('seen.json', 'w') as outfile:
        json.dump(data, outfile)

def load():
    try:
        with open('seen.json', 'r') as infile:
            return(json.load(infile))
    except FileNotFoundError:
        return({})

def display(seen):
    # Clear terminal
    print(chr(27) + "[2J")

    for mac in seen.keys():
        print("{} {} {:04}".format(
            mac,
            seen[mac]['first_ts'],
            seen[mac]['counter'],
            seen[mac]['last_ts'] if 'last_ts' in seen[mac] else ''),
            arrow.get(seen[mac]['last_ts']) - arrow.get(seen[mac]['first_ts']) if 'last_ts' in seen[mac] else ''
        )
    print()

seen = load()

sock = open_bluetooth()

try:
    while True:
        try:
            packet = process_packet(sock)
            # print(packet)
            mac = packet['mac']
            ts = packet['ts']

            if not mac in seen:
                seen[mac] = {}
                seen[mac]['first_ts'] = ts
                seen[mac]['counter'] = 1
            else:
                seen[mac]['last_ts'] = ts
                seen[mac]['counter'] += 1

        except bluez.error as e:
            save(seen)
            time.sleep(10)
            print("Bluetooth failed with error {}, reconnecting".format(e), file=sys.stderr)
            sock = open_bluetooth()
        
        display(seen)

except KeyboardInterrupt:
    save(seen)
    pass