56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
|
from time import gmtime, strftime
|
||
|
|
import paho.mqtt.client as mqtt
|
||
|
|
import sqlite3
|
||
|
|
|
||
|
|
temperature_topic = "temperature"
|
||
|
|
humidity_topic = "humidity"
|
||
|
|
dbFile = "esp8266.db"
|
||
|
|
|
||
|
|
dataTuple = [-1,-1]
|
||
|
|
|
||
|
|
# The callback for when the client receives a CONNACK response from the server.
|
||
|
|
def on_connect(client, userdata, flags, rc):
|
||
|
|
print("Connected with result code "+str(rc))
|
||
|
|
|
||
|
|
# Subscribing in on_connect() means that if we lose the connection and
|
||
|
|
# reconnect then subscriptions will be renewed.
|
||
|
|
client.subscribe(temperature_topic)
|
||
|
|
client.subscribe(humidity_topic)
|
||
|
|
|
||
|
|
# The callback for when a PUBLISH message is received from the server.
|
||
|
|
def on_message(client, userdata, msg):
|
||
|
|
theTime = strftime("%Y-%m-%d %H:%M:%S", gmtime())
|
||
|
|
|
||
|
|
result = (theTime + "\t" + str(msg.payload))
|
||
|
|
print(msg.topic + ":\t" + result)
|
||
|
|
if (msg.topic == temperature_topic):
|
||
|
|
dataTuple[0] = str(msg.payload)
|
||
|
|
if (msg.topic == humidity_topic):
|
||
|
|
dataTuple[1] = str(msg.payload)
|
||
|
|
#return
|
||
|
|
if (dataTuple[0] != -1 and dataTuple[1] != -1):
|
||
|
|
writeToDb(theTime, dataTuple[0], dataTuple[1])
|
||
|
|
return
|
||
|
|
|
||
|
|
def writeToDb(theTime, temperature, humidity):
|
||
|
|
conn = sqlite3.connect(dbFile)
|
||
|
|
c = conn.cursor()
|
||
|
|
print "Writing to db..."
|
||
|
|
c.execute("INSERT INTO climate VALUES (?,?,?)", (theTime, temperature, humidity))
|
||
|
|
conn.commit()
|
||
|
|
|
||
|
|
global dataTuple
|
||
|
|
dataTuple = [-1, -1]
|
||
|
|
|
||
|
|
client = mqtt.Client()
|
||
|
|
client.on_connect = on_connect
|
||
|
|
client.on_message = on_message
|
||
|
|
|
||
|
|
client.connect("raspberrypi", 1883, 60)
|
||
|
|
|
||
|
|
# Blocking call that processes network traffic, dispatches callbacks and
|
||
|
|
# handles reconnecting.
|
||
|
|
# Other loop*() functions are available that give a threaded interface and a
|
||
|
|
# manual interface.
|
||
|
|
client.loop_forever()
|