RaspberrySense/messenger-chatbot

View on GitHub
monitor.py

Summary

Maintainability
A
55 mins
Test Coverage
from collections import deque
import pickle
from threading import Timer, Thread

from gpiozero import MotionSensor, LED, LightSensor

import app
from replies import get_sensor_reply
from sensors import temperature, humidity, camera


MONITOR_RATE = 0.5      # seconds


def send_alert(alert_text):
    sender_id = None
    try:
        with open('data.pickle', 'rb') as file:
            sender_id = pickle.load(file)
    except:
        pass
    if sender_id:
        app.bot.send_text_message(sender_id, alert_text)

    return sender_id


# Motion sensor ---------------------------------------------------------------
def alert_motion_detected():
    send_alert(get_sensor_reply('motion', 'movement'))
    app.log('Motion detected')
    sender_id = send_alert(get_sensor_reply('motion', 'capturing'))
    if sender_id:
        t = Thread(target=app.send_video, args=(sender_id,))
        t.start()

motion_sensor = MotionSensor(17)
motion_sensor.when_motion = alert_motion_detected


# Light sensor ----------------------------------------------------------------
def alert_light_on():
    send_alert('The lights were just switched on')
    app.log('Lights switched on')


def alert_light_off():
    send_alert('The lights were just switched off')
    app.log('Lights switched off')

light_sensor = LightSensor(4, threshold=0.9)
light_sensor.when_light = alert_light_on
light_sensor.when_dark = alert_light_off


# Monitor ---------------------------------------------------------------------
red = LED(10)
cooldown_time = 0
monitor_values = {
    'temperature': deque(maxlen=20),
    'humidity': deque(maxlen=20),
}


def get_rate(values):
    return (values[-1] - values[0]) / (len(values) * MONITOR_RATE)


def monitor():
    global cooldown_time
    cooldown_time -= 1 / MONITOR_RATE       # 1 second
    red.toggle()

    monitor_values['temperature'].append(temperature.get_temperature())
    monitor_values['humidity'].append(humidity.get_humidity())

    for sensor_type in monitor_values:
        rate = get_rate(monitor_values[sensor_type])
        if cooldown_time <= 0:
            trend = None
            if rate >= 2:
                trend = 'rising'
            elif rate <= -2:
                trend = 'dropping'
        if trend:
            send_alert(get_sensor_reply(sensor_type, 'trend', trend,
                                        value=rate))
            app.log(sensor_type + ' ' + trend)
            cooldown_time = 30 / MONITOR_RATE      # 30 seconds


if __name__ == '__main__':
    while True:
        t = Timer(MONITOR_RATE, monitor)
        t.start()
        t.join()