Skip to content

Update mqtt-all.py to detect presence of PMS5003 #77

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Gadgetoid merged 1 commit into from
Jul 14, 2020
Merged

Update mqtt-all.py to detect presence of PMS5003 #77

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
106 changes: 66 additions & 40 deletions examples/mqtt-all.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import ST7735
import time
from bme280 import BME280
from pms5003 import PMS5003, ReadTimeoutError
from pms5003 import PMS5003, ReadTimeoutError, SerialTimeoutError
from enviroplus import gas

try:
Expand Down Expand Up @@ -37,10 +37,10 @@
DEFAULT_MQTT_BROKER_IP = "localhost"
DEFAULT_MQTT_BROKER_PORT = 1883
DEFAULT_MQTT_TOPIC = "enviroplus"
DEFAULT_READ_INTERVAL = 5

# mqtt callbacks
def on_connect(client, userdata, flags, rc):
print(f"CONNACK received with code {rc}")
if rc == 0:
print("connected OK")
else:
Expand All @@ -51,11 +51,10 @@ def on_publish(client, userdata, mid):
print("mid: " + str(mid))


# Read values from BME280 and PMS5003 and return as dict
def read_values(bme280, pms5003):
# Read values from BME280 and return as dict
def read_bme280(bme280):
# Compensation factor for temperature
comp_factor = 2.25

values = {}
cpu_temp = get_cpu_temperature()
raw_temp = bme280.get_temperature() # float
Expand All @@ -65,6 +64,17 @@ def read_values(bme280, pms5003):
int(bme280.get_pressure() * 100), -1
) # round to nearest 10
values["humidity"] = int(bme280.get_humidity())
data = gas.read_all()
values["oxidised"] = int(data.oxidising / 1000)
values["reduced"] = int(data.reducing / 1000)
values["nh3"] = int(data.nh3 / 1000)
values["lux"] = int(ltr559.get_lux())
return values


# Read values PMS5003 and return as dict
def read_pms5003(pms5003):
values = {}
try:
pm_values = pms5003.read() # int
values["pm1"] = pm_values.pm_ug_per_m3(1)
Expand All @@ -76,17 +86,14 @@ def read_values(bme280, pms5003):
values["pm1"] = pm_values.pm_ug_per_m3(1)
values["pm25"] = pm_values.pm_ug_per_m3(2.5)
values["pm10"] = pm_values.pm_ug_per_m3(10)
data = gas.read_all()
values["oxidised"] = int(data.oxidising / 1000)
values["reduced"] = int(data.reducing / 1000)
values["nh3"] = int(data.nh3 / 1000)
values["lux"] = int(ltr559.get_lux())
return values


# Get CPU temperature to use for compensation
def get_cpu_temperature():
process = Popen(["vcgencmd", "measure_temp"], stdout=PIPE, universal_newlines=True)
process = Popen(
["vcgencmd", "measure_temp"], stdout=PIPE, universal_newlines=True
)
output, _error = process.communicate()
return float(output[output.index("=") + 1 : output.rindex("'")])

Expand All @@ -113,14 +120,16 @@ def display_status(disp, mqtt_broker):
WIDTH = disp.width
HEIGHT = disp.height
# Text settings
font_size = 16
font_size = 12
font = ImageFont.truetype(UserFont, font_size)

wifi_status = "connected" if check_wifi() else "disconnected"
text_colour = (255, 255, 255)
back_colour = (0, 170, 170) if check_wifi() else (85, 15, 15)
id = get_serial_number()
message = "{}\nWi-Fi: {}\nmqtt-broker: {}".format(id, wifi_status, mqtt_broker)
device_serial_number = get_serial_number()
message = "{}\nWi-Fi: {}\nmqtt-broker: {}".format(
device_serial_number, wifi_status, mqtt_broker
)
img = Image.new("RGB", (WIDTH, HEIGHT), color=(0, 0, 0))
draw = ImageDraw.Draw(img)
size_x, size_y = draw.textsize(message, font)
Expand All @@ -132,34 +141,50 @@ def display_status(disp, mqtt_broker):


def main():
parser = argparse.ArgumentParser(description="Publish enviroplus values over mqtt")
parser = argparse.ArgumentParser(
description="Publish enviroplus values over mqtt"
)
parser.add_argument(
"--broker", default=DEFAULT_MQTT_BROKER_IP, type=str, help="mqtt broker IP",
"--broker",
default=DEFAULT_MQTT_BROKER_IP,
type=str,
help="mqtt broker IP",
)
parser.add_argument(
"--port", default=DEFAULT_MQTT_BROKER_PORT, type=int, help="mqtt broker port",
"--port",
default=DEFAULT_MQTT_BROKER_PORT,
type=int,
help="mqtt broker port",
)
parser.add_argument(
"--topic", default=DEFAULT_MQTT_TOPIC, type=str, help="mqtt topic"
)
parser.add_argument(
"--interval",
default=DEFAULT_READ_INTERVAL,
type=int,
help="the read interval in seconds",
)
args = parser.parse_args()

# Raspberry Pi ID
device_serial_number = get_serial_number()
device_id = "raspi-" + device_serial_number

print(
"""mqtt-all.py - Reads temperature, pressure, humidity,
PM2.5, and PM10 from Enviro plus and sends data over mqtt.
f"""mqtt-all.py - Reads Enviro plus data and sends over mqtt.

broker: {}
port: {}
topic: {}
broker: {args.broker}
client_id: {device_id}
port: {args.port}
topic: {args.topic}

Press Ctrl+C to exit!

""".format(
args.broker, args.port, args.topic
)
"""
)

mqtt_client = mqtt.Client()
mqtt_client = mqtt.Client(client_id=device_id)
mqtt_client.on_connect = on_connect
mqtt_client.on_publish = on_publish
mqtt_client.connect(args.broker, port=args.port)
Expand All @@ -177,33 +202,34 @@ def main():
# Initialize display
disp.begin()

# Create PMS5003 instance
pms5003 = PMS5003()

# Raspberry Pi ID
device_serial_number = get_serial_number()
id = "raspi-" + device_serial_number
# Try to create PMS5003 instance
HAS_PMS = False
try:
pms5003 = PMS5003()
pm_values = pms5003.read()
HAS_PMS = True
print("PMS5003 sensor is connected")
except SerialTimeoutError:
print("No PMS5003 sensor connected")

# Display Raspberry Pi serial and Wi-Fi status
print("Raspberry Pi serial: {}".format(get_serial_number()))
print("RPi serial: {}".format(device_serial_number))
print("Wi-Fi: {}\n".format("connected" if check_wifi() else "disconnected"))
print("MQTT broker IP: {}".format(args.broker))

time_since_update = 0
update_time = time.time()

# Main loop to read data, display, and send over mqtt
mqtt_client.loop_start()
while True:
try:
time_since_update = time.time() - update_time
values = read_values(bme280, pms5003)
values = read_bme280(bme280)
if HAS_PMS:
pms_values = read_pms5003(pms5003)
values.update(pms_values)
values["serial"] = device_serial_number
print(values)
mqtt_client.publish(args.topic, json.dumps(values))
if time_since_update > 145:
update_time = time.time()
display_status(disp, args.broker)
time.sleep(args.interval)
except Exception as e:
print(e)

Expand Down