100 lines
2.5 KiB
Python
100 lines
2.5 KiB
Python
__author__ = "Aurélien Correia"
|
|
|
|
"""
|
|
components.modbus
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
Modbus component, using pymodbus (python3 branch)
|
|
|
|
typical declaration in configuration.yaml
|
|
|
|
#Modbus TCP
|
|
modbus:
|
|
type: tcp
|
|
host: 127.0.0.1
|
|
port: 2020
|
|
|
|
#Modbus RTU
|
|
modbus:
|
|
type: serial
|
|
method: rtu
|
|
port: /dev/ttyUSB0
|
|
baudrate: 9600
|
|
stopbits: 1
|
|
bytesize: 8
|
|
parity: N
|
|
|
|
"""
|
|
import time
|
|
import logging
|
|
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STOP
|
|
import homeassistant.loader as loader
|
|
from homeassistant.helpers import validate_config
|
|
import homeassistant.components as core
|
|
|
|
# The domain of your component. Should be equal to the name of your component
|
|
DOMAIN = "modbus"
|
|
|
|
# List of component names (string) your component depends upon
|
|
DEPENDENCIES = []
|
|
|
|
# Type of network
|
|
MEDIUM = "type"
|
|
|
|
## if MEDIUM == "serial"
|
|
METHOD = "method"
|
|
SERIAL_PORT = "port"
|
|
BAUDRATE = "baudrate"
|
|
STOPBITS = "stopbits"
|
|
BYTESIZE = "bytesize"
|
|
PARITY = "parity"
|
|
|
|
## if MEDIUM == "tcp" or "udp"
|
|
HOST = "host"
|
|
IP_PORT = "port"
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
NETWORK = None
|
|
TYPE = None
|
|
|
|
def setup(hass, config):
|
|
""" Setup Modbus component. """
|
|
|
|
global TYPE
|
|
TYPE = config[DOMAIN][MEDIUM]
|
|
|
|
# Connect to Modbus network
|
|
global NETWORK
|
|
|
|
if TYPE == "serial":
|
|
from pymodbus.client.sync import ModbusSerialClient as ModbusClient
|
|
NETWORK = ModbusClient( method=config[DOMAIN][METHOD],
|
|
port=config[DOMAIN][SERIAL_PORT],
|
|
baudrate=config[DOMAIN][BAUDRATE],
|
|
stopbits=config[DOMAIN][STOPBITS],
|
|
bytesize=config[DOMAIN][BYTESIZE],
|
|
parity=config[DOMAIN][PARITY])
|
|
elif TYPE == "tcp":
|
|
from pymodbus.client.sync import ModbusTcpClient as ModbusClient
|
|
NETWORK = ModbusClient( host=config[DOMAIN][HOST],
|
|
port=config[DOMAIN][IP_PORT])
|
|
elif TYPE == "udp":
|
|
from pymodbus.client.sync import ModbusUdpClient as ModbusClient
|
|
NETWORK = ModbusClient( host=config[DOMAIN][HOST],
|
|
port=config[DOMAIN][IP_PORT])
|
|
else:
|
|
return False
|
|
|
|
def stop_modbus(event):
|
|
NETWORK.close()
|
|
|
|
def start_modbus(event):
|
|
NETWORK.connect()
|
|
hass.bus.listen_once(EVENT_HOMEASSISTANT_STOP, stop_modbus)
|
|
|
|
hass.bus.listen_once(EVENT_HOMEASSISTANT_START, start_modbus)
|
|
|
|
# Tells the bootstrapper that the component was succesfully initialized
|
|
return True
|