mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-11-26 17:05:31 +01:00
83 lines
2 KiB
Python
83 lines
2 KiB
Python
import asyncio
|
|
|
|
from core.api import Property, on_event
|
|
from core.api.kettle_logic import CBPiKettleLogic
|
|
|
|
|
|
class CustomLogic(CBPiKettleLogic):
|
|
|
|
test = Property.Number(label="Test")
|
|
|
|
running = True
|
|
|
|
|
|
async def wait_for_event(self, topic, callback=None, timeout=None):
|
|
|
|
|
|
future_obj = self.cbpi.app.loop.create_future()
|
|
|
|
async def default_callback(id, **kwargs):
|
|
future_obj.set_result("HELLO")
|
|
|
|
|
|
if callback is None:
|
|
self.cbpi.bus.register(topic=topic, method=default_callback)
|
|
else:
|
|
callback.future = future_obj
|
|
self.cbpi.bus.register(topic=topic, method=callback)
|
|
|
|
if timeout is not None:
|
|
|
|
try:
|
|
print("----> WAIT FOR FUTURE")
|
|
await asyncio.wait_for(future_obj, timeout=timeout)
|
|
print("------> TIMEOUT")
|
|
return future_obj.result()
|
|
except asyncio.TimeoutError:
|
|
print('timeout!')
|
|
else:
|
|
print("----> WAIT FOR FUTURE")
|
|
await future_obj
|
|
return future_obj.result()
|
|
|
|
|
|
|
|
async def run(self):
|
|
|
|
|
|
async def my_callback(value, **kwargs):
|
|
|
|
if value == 5:
|
|
self.cbpi.bus.unregister(my_callback)
|
|
kwargs["future"].set_result("AMAZING")
|
|
else:
|
|
print("OTHER VALUE", value)
|
|
|
|
result = await self.wait_for_event("sensor/1", callback=my_callback)
|
|
print("THE RESULT", result)
|
|
|
|
|
|
'''
|
|
while self.running:
|
|
|
|
|
|
print("RUN", self.test)
|
|
value = await self.cbpi.sensor.get_value(1)
|
|
print(value)
|
|
if value >= 10:
|
|
break
|
|
await asyncio.sleep(1)
|
|
'''
|
|
print("YES IM FINISHED STOP LOGIC")
|
|
|
|
def setup(cbpi):
|
|
|
|
'''
|
|
This method is called by the server during startup
|
|
Here you need to register your plugins at the server
|
|
|
|
:param cbpi: the cbpi core
|
|
:return:
|
|
'''
|
|
|
|
cbpi.plugin.register("CustomKettleLogic", CustomLogic)
|