jdz/modbus_gps001.py
2024-08-03 18:37:54 -06:00

233 lines
7.5 KiB
Python
Executable File

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
import RPi.GPIO as GPIO
import paho.mqtt.client as mqtt
import os
import minimalmodbus
import time
import serial
from datetime import datetime
from influxdb_client import InfluxDBClient, Point
from influxdb_client.client.write_api import SYNCHRONOUS
import logging
import subprocess
# Set up logging
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
# GPIO configuration
GPIO.setmode(GPIO.BCM)
GPIO.setwarnings(False)
GPIO.cleanup()
# MQTT Broker settings
mqttBroker = "65.108.199.212"
myhost = os.uname()[1]
mqtt_client = mqtt.Client(myhost)
mqtt_client.connect(mqttBroker, 1883)
# Modbus configuration
modbus_device = '/dev/modbus'
mb_address = 1
sensy_boi = minimalmodbus.Instrument(modbus_device, mb_address)
sensy_boi.serial.baudrate = 9600
sensy_boi.serial.bytesize = 8
sensy_boi.serial.parity = minimalmodbus.serial.PARITY_NONE
sensy_boi.serial.stopbits = 1
sensy_boi.serial.timeout = 0.5
sensy_boi.mode = minimalmodbus.MODE_RTU
sensy_boi.clear_buffers_before_each_transaction = True
sensy_boi.close_port_after_each_call = True
# GPS configuration
gps_device = '/dev/ttyUSB3'
# InfluxDB settings
INFLUXDB_URL = "http://100.64.024:6"
INFLUXDB_TOKEN = "IPtqPXbaXuuMHvx_tOt1cmIZfLHucd-9DcepXTVpQc-fNKBhp6pkhyTsq_XnoGXdxwILy5AFFgZ_QUZCE5Jhg=="
INFLUXDB_ORG = "juandiego"
INFLUXDB_BUCKET = "gpsdata"
# Initialize InfluxDB client
influx_client = InfluxDBClient(url=INFLUXDB_URL, token=INFLUXDB_TOKEN, org=INFLUXDB_ORG)
write_api = influx_client.write_api(write_options=SYNCHRONOUS)
# GPIO pin configuration
PIN_17 = 17
LuzPuerta = 18
GPIO_25 = 25
GPIO.setup(PIN_17, GPIO.IN)
GPIO.setup(LuzPuerta, GPIO.OUT)
GPIO.setup(GPIO_25, GPIO.OUT)
# Pending data list
pending_data = []
def control_gpio():
status_17 = GPIO.input(PIN_17)
if status_17 == GPIO.LOW:
GPIO.output(LuzPuerta, GPIO.HIGH) # Red light on
else:
GPIO.output(LuzPuerta, GPIO.LOW)
return status_17
def initialize_gps(gps_ser):
"""
Inicializa el dispositivo GPS enviando los comandos necesarios para configurarlo.
"""
try:
# Enviar comandos de configuración al GPS
gps_ser.write(b'AT+CGNSPWR=1\r') # Encender el GPS
time.sleep(2) # Aumentar el tiempo de espera
gps_ser.write(b'AT+CGNSSEQ="RMC"\r') # Configurar la secuencia de datos
time.sleep(2) # Aumentar el tiempo de espera
gps_ser.write(b'AT+CGNSTST=1\r') # Iniciar la transmisión de datos
time.sleep(2) # Aumentar el tiempo de espera
# Leer la respuesta del GPS para asegurarse de que está funcionando
response = gps_ser.read(100)
logging.info(f"GPS initialized with response: {response}")
except Exception as e:
logging.error(f"Error initializing GPS: {e}")
def read_gps_data(gps_ser):
"""
Lee los datos del GPS y los devuelve como una cadena.
"""
try:
gps_ser.write(b'AT+CGNSINF\r')
time.sleep(1)
response = gps_ser.read(200)
logging.info(f"GPS data read: {response}")
return response
except Exception as e:
logging.error(f"Error reading GPS data: {e}")
return None
def publish_to_mqtt(topic, payload):
"""
Publica un mensaje en el broker MQTT.
"""
try:
mqtt_client.publish(topic, payload)
logging.info(f"Published to MQTT topic {topic}: {payload}")
except Exception as e:
logging.error(f"Error publishing to MQTT: {e}")
def write_to_influxdb(data):
"""
Escribe los datos en InfluxDB.
"""
try:
point = Point("gps_data").tag("host", myhost).field("data", data)
write_api.write(bucket=INFLUXDB_BUCKET, org=INFLUXDB_ORG, record=point)
logging.info(f"Data written to InfluxDB: {data}")
except Exception as e:
logging.error(f"Error writing to InfluxDB: {e}")
def get_gps_position(gps_ser):
"""
Obtiene la posición GPS.
"""
gps_data = read_gps_data(gps_ser)
if gps_data:
# Procesar los datos GPS aquí si es necesario
return True
return False
def reset_usb_ports():
"""
Resetea los puertos USB.
"""
try:
subprocess.run(["usbreset", "/dev/bus/usb/001/001"], check=True)
logging.info("USB ports reset successfully")
except Exception as e:
logging.error(f"Error resetting USB ports: {e}")
def reinitialize_devices():
"""
Re-inicializa los dispositivos.
"""
try:
global sensy_boi, gps_ser
sensy_boi = minimalmodbus.Instrument(modbus_device, mb_address)
gps_ser = serial.Serial(gps_device, baudrate=9600, timeout=1)
initialize_gps(gps_ser)
logging.info("Devices reinitialized successfully")
except Exception as e:
logging.error(f"Error reinitializing devices: {e}")
def main():
try:
# Initialize GPS
gps_ser = serial.Serial(gps_device, baudrate=9600, timeout=1)
initialize_gps(gps_ser)
reset_counter = 0
while True:
try:
# Control GPIO
status_17 = control_gpio()
# Read Modbus data
try:
temp = sensy_boi.read_register(0, 1) # Example register for temperature
hum = sensy_boi.read_register(1, 1) # Example register for humidity
# Try to publish current data
mqtt_client.publish(f"iiot/{myhost}/temperature", temp)
mqtt_client.publish(f"iiot/{myhost}/humidity", hum)
mqtt_client.publish(f"iiot/{myhost}/door/pin17", str(status_17))
# Try to publish pending data
for item in pending_data:
mqtt_client.publish(f"iiot/{myhost}/temperature", item["temperature"])
mqtt_client.publish(f"iiot/{myhost}/humidity", item["humidity"])
mqtt_client.publish(f"iiot/{myhost}/door/pin17", item["door_status"])
pending_data.clear() # Clear the list if all data was published
except Exception as e:
logging.error(f"Error publishing data to MQTT: {str(e)}")
pending_data.append({
"temperature": temp,
"humidity": hum,
"door_status": str(status_17)
}) # Save data in memory
# Print processed data
logging.info("-------------------------------------")
logging.info(f"Temperature = {temp:.1f}°C")
logging.info(f"Relative humidity = {hum:.1f}%")
logging.info(f"Door status (PIN 17) = {status_17}")
logging.info("-------------------------------------")
except Exception as e:
logging.error(f"Error reading Modbus data: {str(e)}")
# Get GPS position
if not get_gps_position(gps_ser):
logging.warning("Failed to get GPS position")
reset_counter += 1
if reset_counter >= 60: # Reset every 30 minutes (60 * 30 seconds)
reset_usb_ports()
reinitialize_devices()
reset_counter = 0
time.sleep(30)
except KeyboardInterrupt:
logging.info("Program interrupted by user")
except Exception as e:
logging.error(f"Unexpected error: {str(e)}")
finally:
GPIO.cleanup()
gps_ser.close()
influx_client.close()
if __name__ == "__main__":
main()