import ST7735
import time
from bme280 import BME280
-from pms5003 import PMS5003, ReadTimeoutError
+from pms5003 import PMS5003, ReadTimeoutError, SerialTimeoutError
from enviroplus import gas
try:
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:
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
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)
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("'")])
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)
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)
# 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)
projects / EVA-2020-02-2.git / blobdiff