meters.py

#!/usr/bin/python3
from __future__ import print_function
import argparse
import binascii
import os
import sys
import time
import json
import datetime
import threading
from tinydb import TinyDB, Query
import time
from bluepy import btle
import flask
from flask import Response, request

# Database Configuration
# Storage location of the database file
databaseWritePath = "switchbot_data.json"
# TinyDb
database = TinyDB(databaseWritePath)

# Flask configuration
app = flask.Flask(__name__)
app.config["DEBUG"] = True

# API Configuration
API_HOST="192.168.1.233"
API_PORT=5000

# SwitchBot Meter Configuration
METER_ROOMS = ['Bedroom']
METER_MACS = ['e8:fe:50:d1:75:dd']

debug_level = 1

if os.getenv('C', '1') == '0':
    ANSI_RED = ''
    ANSI_GREEN = ''
    ANSI_YELLOW = ''
    ANSI_CYAN = ''
    ANSI_WHITE = ''
    ANSI_OFF = ''
else:
    ANSI_CSI = "\033["
    ANSI_RED = ANSI_CSI + '31m'
    ANSI_GREEN = ANSI_CSI + '32m'
    ANSI_YELLOW = ANSI_CSI + '33m'
    ANSI_CYAN = ANSI_CSI + '36m'
    ANSI_WHITE = ANSI_CSI + '37m'
    ANSI_OFF = ANSI_CSI + '0m'

class MeterReading:
    def __init__(self, time, temperature, humidity, battery):
        self.time = time
        self.temperature = temperature
        self.humidity = humidity
        self.battery = battery

# Responsible for scanning Bluetooth devices for SwitchBot Meter
class ScanProcessor():
    def handleDiscovery(self, dev, isNewDev, isNewData):
        try:
            if dev.addr in METER_MACS:
                i = 0
                room = METER_ROOMS[METER_MACS.index(dev.addr)]
                if debug_level == 1:
                    print ('\nRoom: %s Device: %s (%s), %d dBm %s. ' %(ANSI_WHITE + room + ANSI_OFF,ANSI_WHITE + dev.addr + ANSI_OFF,dev.addrType,dev.rssi,('' if dev.connectable else '(not connectable)')), end='')
                for (sdid, desc, value) in dev.getScanData():
                    i=i+1
                    if debug_level == 1:
                        print( str(i) + ': ' + str(sdid) + ', '+ desc + ', ' + value)
                    
                    #Model T (WOSensorTH) example Service Data: 000d54006400962c
                    if desc == '16b Service Data':
                        if value.startswith('000d'):
                            byte2 = int(value[8:10],16)
                            battery = (byte2 & 127)
                            byte3 = int(value[10:12],16)
                            byte4 = int(value[12:14],16)
                            byte5 = int(value[14:16],16)
                            tempc = float(byte4-128)+float(byte3 / 10.0)
                            humidity = byte5
                            self._publish(room, tempc, humidity, battery)
                        else:
                            if debug_level == 1:
                                print(value.len())
    
                if not dev.scanData:
                    print ('(no data)')
                    print

        except:
            print("handleDiscovery: Oops!",sys.exc_info()[0],"occurred.")

    def _publish(self, room, tempc, humidity, battery):
        try:
            now = datetime.datetime.now()
            timeNow = now.strftime("%Y-%m-%d %H:%M:%S")

            # Get the readings table
            readingsTable = database.table('readings')

            # Add the reading to the local database
            readingsTable.insert({
                'time': timeNow,
                'room': room,
                'temperature': tempc,
                'humidity': humidity,
                'battery': battery
            })
        except:
            print("_publish: Oops!",sys.exc_info()[0],"occurred.")

# Runs in the background, getting data from SwitchBot Meters and then storing it in local memory
class ScanBackgroundWorker(object):
    def __init__(self, interval=30):
        self.interval = interval

        thread = threading.Thread(target=self.run, args=())
        thread.daemon = True                            # Daemonize thread
        thread.start()                                  # Start the execution

    def run(self):
        while True:
            # Clear previously taken readings before scan
            database.drop_table('readings')
            # Get the latest readings from the SwitchBot Meter
            scanner = btle.Scanner().withDelegate(ScanProcessor())
            scanner.scan()
            # Wait until the next interval to scan for the latest devices        
            time.sleep(self.interval)

# Shows all meter reading in local memory
@app.route('/meters', methods=['GET'])
def allMeters():
    latestReadings = database.table('readings', cache_size=0)
    latestReadingsJson = json.dumps(latestReadings.all())
    return Response(latestReadingsJson, mimetype='application/json')

@app.route('/meters/<meter_name>', methods=['GET'])
def getMeterByRoom(meter_name):
    # Get the room name
    meterRoom = meter_name
    # Get the readings table from the database
    latestReadings = database.table('readings', cache_size=0)
    # Query the table for readings matching the room name
    readings = Query()
    matchingReadingsByRoom = latestReadings.search(readings.room == meterRoom)
    # Check if the query returned a result
    if (len(matchingReadingsByRoom) > 0):
        # Select the first item on the list
        firstMatchingDevice = matchingReadingsByRoom[0]
        # Dump the results into a JSON string
        latestReadingsJson = json.dumps(firstMatchingDevice)
        # Return the result of the query in the HTTP Response
        return Response(latestReadingsJson, mimetype='application/json')
    else:
        return Response(json.dumps([]), status=204, mimetype='application/json')

@app.route('/humidity/<meter_name>', methods=['GET'])
def getHumidityByRoom(meter_name):
    # Get the room name
    meterRoom = meter_name
    # Get the readings table from the database
    latestReadings = database.table('readings', cache_size=0)
    # Query the table for readings matching the room name
    readings = Query()
    matchingReadingsByRoom = latestReadings.search(readings.room == meterRoom)
    # Check if the query returned a result
    if (len(matchingReadingsByRoom) > 0):
        # Select the first item on the list
        firstMatchingDevice = matchingReadingsByRoom[0]
        # Get the humidity value from the document
        humidity = firstMatchingDevice["humidity"]
        print('Humidity in ' + str(meterRoom) + " " + str(humidity))
        # Return the result of the query in the HTTP Response
        return Response(str(humidity))
    else:
        return Response(status=204)

@app.route('/temperature/<meter_name>', methods=['GET'])
def getTempByRoom(meter_name):
    # Get the room name
    meterRoom = meter_name
    # Get the readings table from the database
    latestReadings = database.table('readings', cache_size=0)
    # Query the table for readings matching the room name
    readings = Query()
    matchingReadingsByRoom = latestReadings.search(readings.room == meterRoom)
    # Check if the query returned a result
    if (len(matchingReadingsByRoom) > 0):
        # Select the first item on the list
        firstMatchingDevice = matchingReadingsByRoom[0]
        # Get the temperature value from the document
        temperature = firstMatchingDevice["temperature"]
        print('temperature in ' + str(meterRoom) + " " + str(temperature))
        # Return the result of the query in the HTTP Response
        return Response(str(temperature))
    else:
        return Response(status=204)

def main():
    ScanBackgroundWorker()
    app.run(host=API_HOST, port=API_PORT, debug=True)

if __name__ == "__main__":
        main()