mirror of
https://github.com/warkanum/warkanums-pi-device-snippets.git
synced 2025-05-19 02:57:28 +00:00
221 lines
8.7 KiB
Python
Executable File
221 lines
8.7 KiB
Python
Executable File
#!/usr/bin/python3
|
|
from threading import Thread
|
|
|
|
import RPi.GPIO as GPIO
|
|
import time
|
|
import datetime
|
|
import os
|
|
import sys
|
|
import dht11_sensor
|
|
import psycopg2
|
|
import copy
|
|
|
|
###-----------------Hardware Settings-----------------------
|
|
|
|
PIN_LC=25 #Light Sensor (GPIO.IN, pull_up_down=GPIO.PUD_UP)
|
|
PIN_MC=17 #Motion Sensor (GPIO.IN, pull_up_down=GPIO.PUD_UP)
|
|
PIN_TC=4 #Temp Sensor (GPIO.IN, pull_up_down=GPIO.PUD_UP)
|
|
PIN_TC_WP=7 #Temp Sensor #Wirepi pin 7
|
|
PIN_LED1=23 #LED Blue 1
|
|
PIN_LED2=24 #LED Blue 2
|
|
|
|
|
|
###------------------SQL Settings-----------------------------
|
|
SQL_SRV='127.0.0.1'
|
|
SQL_USER='pistats'
|
|
SQL_PASSWD='pistats'
|
|
SQL_DB='pistats'
|
|
|
|
|
|
#setup pins. Some are setup by functions below.
|
|
GPIO.setmode(GPIO.BCM)
|
|
GPIO.setup(PIN_LED1,GPIO.OUT)
|
|
GPIO.setup(PIN_LED2,GPIO.OUT)
|
|
|
|
GPIO.setup(PIN_MC, GPIO.IN, pull_up_down=GPIO.PUD_UP)
|
|
|
|
|
|
#dim leds
|
|
GPIO.output(PIN_LED1,GPIO.LOW)
|
|
GPIO.output(PIN_LED2,GPIO.LOW)
|
|
|
|
|
|
def UnixLocalEpoch():
|
|
dt = datetime.datetime.now()
|
|
return int((dt - datetime.datetime(1970,1,1)).total_seconds())
|
|
|
|
def PhotoSensor(RCpin):
|
|
reading = 0
|
|
GPIO.setup(RCpin, GPIO.OUT)
|
|
GPIO.output(RCpin, GPIO.LOW)
|
|
time.sleep(0.1)
|
|
|
|
GPIO.setup(RCpin, GPIO.IN)
|
|
# This takes about 1 millisecond per loop cycle
|
|
while (GPIO.input(RCpin) == GPIO.LOW):
|
|
reading += 1
|
|
|
|
return reading
|
|
|
|
def TempsensorRead():
|
|
### Loop through the temp sensor library until we get a valid reading ###
|
|
|
|
for i in range(1,100):
|
|
data = dht11_sensor.read(PIN_TC_WP)
|
|
#print('Temp={0}*C Humidity={1}% Status={2} Error={3}'.format(data['temperature'], data['humidity'], data['valid'], data['err']))
|
|
if data['valid'] == 1:
|
|
return data
|
|
|
|
return None
|
|
|
|
def save_data(p_SensorValues):
|
|
try:
|
|
sql_con = psycopg2.connect(host=SQL_SRV, user=SQL_USER,password=SQL_PASSWD,database=SQL_DB)
|
|
sql_cur = sql_con.cursor()
|
|
|
|
print('2temp->' + str(p_SensorValues['temperature']['save']))
|
|
print('2light->' + str(p_SensorValues['light']['save']))
|
|
print('2motion->' + str(p_SensorValues['motion']['save']))
|
|
|
|
if p_SensorValues.get('motion', None):
|
|
sql_cur.execute("""select id, data_read from sensordata
|
|
where sensor_type = 'motion' order by id desc limit 1""")
|
|
data = sql_cur.fetchone()
|
|
if not data or p_SensorValues['motion']['save']: #(data and str(data[1]) != str(p_SensorValues['motion']['data'])):
|
|
sql_cur.execute("""INSERT INTO sensordata (sensor_type, data_read, date_added,time_added)
|
|
VALUES (%s, %s, TIMESTAMP 'epoch' + %s * INTERVAL '1 second',TIMESTAMP 'epoch' + %s * INTERVAL '1 second')""",
|
|
('motion', p_SensorValues['motion']['data'], p_SensorValues['motion']['read'],p_SensorValues['motion']['read'] ))
|
|
|
|
|
|
if p_SensorValues.get('light', None):
|
|
sql_cur.execute("select id, data_read from sensordata where sensor_type = 'light' order by id desc limit 1")
|
|
data = sql_cur.fetchone()
|
|
#we have a +- 10 variance on light.
|
|
if not data or p_SensorValues['light']['save']: #(data and (int(p_SensorValues['light']['data']) > int(data[1])+10 or int(p_SensorValues['light']['data']) < int(data[1]) - 10) ):
|
|
sql_cur.execute("""INSERT INTO sensordata (sensor_type, data_read, date_added,time_added)
|
|
VALUES(%s, %s, TIMESTAMP 'epoch' + %s * INTERVAL '1 second',TIMESTAMP 'epoch' + %s * INTERVAL '1 second')""",
|
|
('light', p_SensorValues['light']['data'], p_SensorValues['light']['read'],p_SensorValues['light']['read'] ))
|
|
|
|
if p_SensorValues.get('temperature', None):
|
|
sql_cur.execute("select id, data_read from sensordata where sensor_type = 'temperature' order by id desc limit 1")
|
|
data = sql_cur.fetchone()
|
|
if not data or p_SensorValues['temperature']['save']: #(data and str(data[1]) != str(p_SensorValues['temperature']['temperature'])):
|
|
sql_cur.execute("""INSERT INTO sensordata (sensor_type, data_read, date_added,time_added)
|
|
VALUES(%s, %s, TIMESTAMP 'epoch' + %s * INTERVAL '1 second',TIMESTAMP 'epoch' + %s * INTERVAL '1 second')""",
|
|
('temperature', p_SensorValues['temperature']['temperature'], p_SensorValues['temperature']['read'], p_SensorValues['temperature']['read'] ))
|
|
|
|
if p_SensorValues.get('temperature', None):
|
|
sql_cur.execute("select id, data_read from sensordata where sensor_type = 'humidity' order by id desc limit 1")
|
|
data = sql_cur.fetchone()
|
|
if not data or p_SensorValues['temperature']['save']:#(data and str(data[1]) != str(p_SensorValues['temperature']['humidity'])):
|
|
sql_cur.execute("""INSERT INTO sensordata (sensor_type, data_read, date_added,time_added)
|
|
VALUES(%s, %s, TIMESTAMP 'epoch' + %s * INTERVAL '1 second',TIMESTAMP 'epoch' + %s * INTERVAL '1 second')""",
|
|
('humidity', p_SensorValues['temperature']['humidity'], p_SensorValues['temperature']['read'], p_SensorValues['temperature']['read'] ))
|
|
|
|
sql_con.commit()
|
|
sql_cur.close()
|
|
sql_con.close()
|
|
|
|
except psycopg2.Error as e:
|
|
print("SQL error in save_data: " + str(e))
|
|
|
|
except Exception as e:
|
|
print("Unknown error in save_data: " + str(e))
|
|
|
|
|
|
def main():
|
|
SensorValue = {}
|
|
TICK_LT = 0 #light detect ticker
|
|
TICK_LTI = 0 #light insert ticker
|
|
TICK_TMP = 0 #temp ticker
|
|
|
|
BlueLed = False
|
|
|
|
while True:
|
|
changed = False
|
|
|
|
|
|
motionData = GPIO.input(PIN_MC)
|
|
if not SensorValue.get('motion', None):
|
|
SensorValue['motion'] = {'data': motionData , 'read': UnixLocalEpoch(), 'changed': UnixLocalEpoch(), 'save': False}
|
|
|
|
SensorValue['motion']['save'] = False
|
|
|
|
if int(SensorValue['motion'].get('data', 0)) != int(motionData) :
|
|
changed = True
|
|
SensorValue['motion']['changed'] = UnixLocalEpoch()
|
|
SensorValue['motion']['save'] = True
|
|
SensorValue['motion']['data'] = int(motionData)
|
|
SensorValue['motion']['read'] = UnixLocalEpoch()
|
|
|
|
|
|
if (SensorValue['motion']['data'] > 0):
|
|
#GPIO.output(PIN_LED1,GPIO.HIGH) #flash led
|
|
SensorValue['motion']['lastmotion'] = UnixLocalEpoch()
|
|
BlueLed = True
|
|
else:
|
|
#GPIO.output(PIN_LED1,GPIO.LOW) #flash led stop
|
|
BlueLed = False
|
|
|
|
|
|
#Measure Light
|
|
if not SensorValue.get('light', None):
|
|
SensorValue['light'] = {'data': PhotoSensor(PIN_LC) , 'read': UnixLocalEpoch(), 'save': False }
|
|
|
|
SensorValue['light']['save'] = False
|
|
|
|
lightChanges = 0
|
|
if (TICK_LT < time.perf_counter()):
|
|
TICK_LT = time.perf_counter()+1
|
|
lightData = PhotoSensor(PIN_LC)
|
|
lightChanges = abs(SensorValue['light'].get('data', 0) - lightData)
|
|
#print("LC->" + str(lightData ) + "DF:" + str(lightChanges))
|
|
|
|
if (TICK_LTI < time.perf_counter() or (lightData > 600 and lightChanges > 200) or (lightData < 600 and lightChanges > 30)):
|
|
TICK_LTI = time.perf_counter()+30
|
|
|
|
if SensorValue['light'].get('data', 0) != lightData :
|
|
changed = True
|
|
SensorValue['light']['changed'] = UnixLocalEpoch()
|
|
SensorValue['light']['save'] = True
|
|
|
|
SensorValue['light']['data'] = lightData
|
|
SensorValue['light']['read'] = UnixLocalEpoch()
|
|
|
|
#Measure Temprature, this might hold the thread for a few seconds at most.
|
|
if not SensorValue.get('temperature', None):
|
|
SensorValue['temperature'] = {'temperature': 0, 'humidity': 0, 'changed': 0, 'save': False}
|
|
|
|
SensorValue['temperature']['save'] = False
|
|
|
|
if (TICK_TMP < time.perf_counter()):
|
|
TICK_TMP = time.perf_counter()+10
|
|
tempData = TempsensorRead()
|
|
if tempData:
|
|
print('temperature reading...')
|
|
if (SensorValue['temperature'].get('temperature', 0) != tempData['temperature']
|
|
or SensorValue['temperature'].get('humidity', 0) != tempData['humidity']):
|
|
SensorValue['temperature']['changed'] = UnixLocalEpoch()
|
|
SensorValue['temperature']['temperature'] = tempData['temperature']
|
|
SensorValue['temperature']['humidity'] = tempData['humidity']
|
|
SensorValue['temperature']['save'] = True
|
|
changed = True
|
|
SensorValue['temperature']['read'] = UnixLocalEpoch()
|
|
|
|
if changed:
|
|
print('---------------change-------------')
|
|
print('temp->' + str(SensorValue['temperature']['save']))
|
|
print('light->' + str(SensorValue['light']['save']))
|
|
print('motion->' + str(SensorValue['motion']['save']))
|
|
#Gosh we need a copy cause sql can be to slow sometimes
|
|
|
|
ThreadsData = copy.deepcopy(SensorValue)
|
|
t = Thread(target=save_data, args=(ThreadsData,))
|
|
t.start()
|
|
|
|
|
|
|
|
time.sleep(0.01)
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main()) |