mirror of
https://github.com/supleed2/ELEC60013-ES-CW1.git
synced 2024-11-10 01:35:50 +00:00
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
import os, sys
|
|
from time import sleep
|
|
import yaml
|
|
import paho.mqtt.client as mqtt
|
|
import json, smbus2, si7201, tmp006, lis3dh, hci, gpiozero
|
|
|
|
# Global Sensor Data Variables
|
|
dailysteps = 0
|
|
fallen = False
|
|
|
|
def incrementStepCount(interrupt, sensor) -> None:
|
|
global dailysteps
|
|
dailysteps += 1
|
|
sensor.resetint2()
|
|
|
|
def setFallen(interrupt, sensor) -> None:
|
|
global fallen
|
|
fallen = True
|
|
sensor.resetint1()
|
|
|
|
# Setup
|
|
bus = smbus2.SMBus(1) # set up I2C bus 1
|
|
|
|
temphum = si7201.Si7201(bus, 0x40) # set up Si7201 sensor
|
|
temphum.reset() # reset Si7201
|
|
|
|
irtemp = tmp006.TMP006(bus, 0x41, tmp006.SAMPLERATE_4HZ) # set up TMP006 sensor
|
|
irtemp.active = 1 # turn on TMP006
|
|
|
|
accel = lis3dh.LIS3DH(bus, 10, 0x18) # set up LIS3DH sensor
|
|
fall = gpiozero.Button(18, pull_up = False) # GPIO17: Freefall Interrupt (INT1)
|
|
fall.when_activated = lambda: setFallen(fall, accel) # set fallen to True when Freefall Interrupt (INT1) is triggered
|
|
step = gpiozero.Button(17, pull_up = False) # GPIO18: Step Counter Interrupt (INT2)
|
|
step.when_activated = lambda: incrementStepCount(step, accel) # increment step count when Step Counter Interrupt (INT2) is triggered
|
|
|
|
with open(".secrets.yml", "r") as secrets:
|
|
try:
|
|
secrets = yaml.load(secrets, Loader = yaml.SafeLoader)
|
|
key = secrets["key"] # Get Base64 encoded device public key from secrets file
|
|
except ImportError as exc:
|
|
print(exc)
|
|
sleep(60) # 60s delay before restarting
|
|
os.execl(sys.executable, os.path.abspath(__file__), *sys.argv) # Restart propgram
|
|
btcast = hci.HCIBroadcaster(key) # set up HCI Broadcaster
|
|
|
|
client = mqtt.Client("RaspberryPi") # set up MQTT client
|
|
client.connect("add8.duckdns.org", 8883, 60) # connect to MQTT broker
|
|
client.loop_start() # Start a new thread to handle sending MQTT messages
|
|
|
|
# Main Loop
|
|
while True:
|
|
data = {
|
|
"devID": "testdoggo",
|
|
"air_temp": temphum.temperature,
|
|
"day_steps": dailysteps,
|
|
"hum_perc": temphum.humidity,
|
|
"pet_temp": irtemp.temperature
|
|
}
|
|
mqtt_data = json.dumps(data)
|
|
client.publish("/data", mqtt_data)
|
|
btcast.start_advertising() # Send out BT advertisement
|
|
sleep(5) # Sleep for 5 seconds to lower power consumption
|