113 linhas
4.2 KiB
Python
113 linhas
4.2 KiB
Python
import config
|
|
import logging
|
|
import mqtt
|
|
import report
|
|
import socket
|
|
import subprocess
|
|
import time
|
|
|
|
try:
|
|
from config import APP_NAME as ROOT_LOGGER_NAME
|
|
except ImportError:
|
|
ROOT_LOGGER_NAME = 'root'
|
|
logger = logging.getLogger(ROOT_LOGGER_NAME).getChild('main')
|
|
|
|
|
|
class sispmctl(object):
|
|
AMP_CHANNEL = 0
|
|
|
|
def __init__(self, mqtt_client):
|
|
logger.info("Starting sispmctl module...")
|
|
self.__mqtt_client__ = mqtt_client
|
|
self.__state_requests__ = [{}, {}, {}, {}]
|
|
self.publish_states()
|
|
|
|
def __filter_output_parameter__(self, output):
|
|
try:
|
|
return int(output)
|
|
except ValueError:
|
|
return output
|
|
|
|
def set_out_state(self, output, state):
|
|
if output == "all":
|
|
if self.__state__ != [state, state, state, state]:
|
|
self.__set_out_state__(output, state)
|
|
else:
|
|
if self.__state__[int(output) - 1] != state:
|
|
self.__set_out_state__(output, state)
|
|
|
|
def __set_out_state__(self, output, state):
|
|
try:
|
|
out_txt = subprocess.check_output(["sudo", "sispmctl", "-o" if state == True else "-f", output]).decode('UTF-8')
|
|
except subprocess.CalledProcessError as grepexc:
|
|
logger.error("sispm error code %d while changing state of channel %s", grepexc.returncode, output)
|
|
else:
|
|
logger.info('sispmctl channel %s changed to %s', output, state)
|
|
self.publish_states()
|
|
|
|
def toggle_out_state(self, output):
|
|
if output == "all":
|
|
for i in range(1,5):
|
|
self.toggle_out_state(str(i))
|
|
else:
|
|
self.set_out_state(output, not self.__state__[int(output) - 1])
|
|
|
|
def get_out_states(self):
|
|
i = 0
|
|
while i < 10: # number of tries
|
|
try:
|
|
out_txt = subprocess.check_output(["sudo", "sispmctl", "-g", "all"])
|
|
except subprocess.CalledProcessError as grepexc:
|
|
logger.error("sispm error code %d while getting states", grepexc.returncode)
|
|
else:
|
|
rv = []
|
|
for i in range(1, 5):
|
|
rv.append(out_txt.decode("UTF-8").split("\n")[i].split("\t")[1] == "on")
|
|
return rv
|
|
logger.warning("Could not get state of powerplugs")
|
|
return self.__state__
|
|
|
|
def publish_states(self):
|
|
self.__state__ = self.get_out_states()
|
|
for i in range(1, 5):
|
|
self.send_state(i)
|
|
|
|
def send_state(self, output):
|
|
topic = config.MQTT_TOPIC + "/output/" + str(output)
|
|
logger.debug("Sending Powerplug status information of plug %s to mqtt %s = %s", str(output), topic, str(self.__state__[output - 1]))
|
|
try:
|
|
self.__mqtt_client__.send(topic, "true" if self.__state__[output - 1] else "false")
|
|
except (socket.timeout, OSError) as e:
|
|
logger.warning("Error while sending state information via mqtt")
|
|
|
|
|
|
class mqtt_powerplug(mqtt.mqtt_client):
|
|
def __init__(self):
|
|
mqtt.mqtt_client.__init__(self, config.APP_NAME, config.MQTT_SERVER, 1883, config.MQTT_USER, config.MQTT_PASS)
|
|
for target in ["1", "2", "3", "4", "all"]:
|
|
self.add_callback(config.MQTT_TOPIC + "/output/%s/set" % target, self.__set__)
|
|
#
|
|
self.__sc__ = sispmctl(self)
|
|
|
|
def __set__(self, client, userdata, message):
|
|
output = message.topic.split("/")[-2]
|
|
if message.payload == b"true" or message.payload == b"false":
|
|
self.__sc__.set_out_state(output, message.payload == b"true")
|
|
elif message.payload.lower() == b'toggle':
|
|
self.__sc__.toggle_out_state(output)
|
|
else:
|
|
logger.warning("Ignoring set command due to payload: %s", message.payload)
|
|
|
|
def __del__(self):
|
|
self.__client__.loop_stop() # stop the loop
|
|
|
|
|
|
if __name__ == '__main__':
|
|
report.appLoggingConfigure(config.__BASEPATH__, config.LOGTARGET, ((config.APP_NAME, config.LOGLVL), ), fmt=config.formatter, host=config.LOGHOST, port=config.LOGPORT)
|
|
#
|
|
mp = mqtt_powerplug()
|
|
|
|
while True:
|
|
time.sleep(30)
|
|
mp.__sc__.publish_states()
|