mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-11-10 01:17:42 +01:00
80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
import logging
|
|
import asyncio
|
|
from cbpi.api import *
|
|
from cbpi.api.config import ConfigType
|
|
from cbpi.api.base import CBPiBase
|
|
from cbpi.controller.fermentation_controller import FermentationController
|
|
from cbpi.controller.kettle_controller import KettleController
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class MQTTUtil(CBPiExtension):
|
|
|
|
def __init__(self,cbpi):
|
|
self.cbpi = cbpi
|
|
self.kettlecontroller = cbpi.kettle
|
|
self.fermentationcontroller = cbpi.fermenter
|
|
# sensor and actor update is done anyhow during startup
|
|
# self.sensorcontroller = cbpi.sensor
|
|
# self.actorcontroller = cbpi.actor
|
|
|
|
self.mqttupdate = int(self.cbpi.config.get("MQTTUpdate", 0))
|
|
if self.mqttupdate != 0:
|
|
self._task = asyncio.create_task(self.run())
|
|
logger.info("INIT MQTTUtil")
|
|
else:
|
|
self._task = asyncio.create_task(self.push_once())
|
|
|
|
async def push_once(self):
|
|
# wait some time to ensure that kettlecontroller is started
|
|
await asyncio.sleep(5)
|
|
self.push_update()
|
|
|
|
|
|
async def run(self):
|
|
|
|
while True:
|
|
self.push_update()
|
|
await asyncio.sleep(self.mqttupdate)
|
|
|
|
def remove_key(self,d, key):
|
|
r = dict(d)
|
|
del r[key]
|
|
return r
|
|
|
|
def push_update(self):
|
|
# try:
|
|
# self.actor=self.actorcontroller.get_state()
|
|
# for item in self.actor['data']:
|
|
# self.cbpi.push_update("cbpi/{}/{}".format("actorupdate",item['id']), item)
|
|
# except Exception as e:
|
|
# logging.error(e)
|
|
# pass
|
|
# try:
|
|
# self.sensor=self.sensorcontroller.get_state()
|
|
# for item in self.sensor['data']:
|
|
# self.cbpi.push_update("cbpi/{}/{}".format("sensorupdate",item['id']), item)
|
|
# except Exception as e:
|
|
# logging.error(e)
|
|
# pass
|
|
try:
|
|
self.kettle=self.kettlecontroller.get_state()
|
|
for item in self.kettle['data']:
|
|
self.cbpi.push_update("cbpi/{}/{}".format("kettleupdate",item['id']), item)
|
|
except Exception as e:
|
|
logging.error(e)
|
|
pass
|
|
try:
|
|
self.fermenter=self.fermentationcontroller.get_state()
|
|
for item in self.fermenter['data']:
|
|
item_new=self.remove_key(item,"steps")
|
|
self.cbpi.push_update("cbpi/{}/{}".format("fermenterupdate",item['id']), item_new)
|
|
except Exception as e:
|
|
logging.error(e)
|
|
pass
|
|
|
|
|
|
def setup(cbpi):
|
|
if str(cbpi.static_config.get("mqtt", False)).lower() == "true":
|
|
cbpi.plugin.register("MQTTUtil", MQTTUtil)
|
|
pass
|