tfbrew.py

"""
Backend for a yet another brewing application controlled by a Blynk frontend
"""

import sys, os
import importlib
import logging

import asyncio
from aiohttp import web
from ruamel.yaml import YAML

import interfaces
import controller
import event
from common import app, components

yaml = YAML(typ='safe')   # default, if not specfied, is 'rt' (round-trip)
configFile = 'config.yaml'
if len(sys.argv) > 1:
    configFile = sys.argv[1]
print("Using config from %s"%configFile)
    
config = yaml.load(open(configFile,mode='r'))

logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s:%(message)s', filename='tfbrew.log', filemode='w')
logger = logging.getLogger(__name__)

console = logging.StreamHandler()
console.setLevel(config.get('consoleLoglevel', 'WARNING'))
logging.getLogger('').addHandler(console)

sys.path.append(os.path.join(os.path.dirname(__file__), "plugins"))

for componentType in ['sensors', 'actors', 'extensions']:
    for component in config[componentType]:
        for name, attribs in component.items():
            logging.info("setting up %s"%name)
            plugin = importlib.import_module('plugins.%s'%attribs['plugin'])
            components[name] = plugin.factory(name, attribs)
for ctrl in config['controllers']:
    for name, attribs in ctrl.items():
        logger.info("setting up %s"%name)
        logicPlugin = importlib.import_module('plugins.%s'%attribs['plugin'])
        logic = logicPlugin.factory(name, attribs['logicCoeffs'])
        sensor = components[attribs['sensor']]
        actor = components[attribs['actor']]
        agitator = components.get(attribs.get('agitator', ''),None)
        initialSetpoint = attribs.get('initialSetpoint', 67.0)
        initiallyEnabled = True if attribs.get('initialState', 'on') == 'on' else 'off'
        components[name] = controller.Controller(name, sensor, actor, logic, agitator, initialSetpoint, initiallyEnabled)


for conn in config['connections']:
    (sendEvent, recvEvent) = conn.split('=>')
    (sendComponent, sendType) = sendEvent.split('.')
    (recvComponent, recvType) = recvEvent.split('.')
    event.register(sendEvent, lambda event, rc=recvComponent, rt=recvType: components[rc].callback(rt, event))

async def start_background_tasks(app):
    pass

async def cleanup_background_tasks(app):
    pass

app.on_startup.append(start_background_tasks)
app.on_cleanup.append(cleanup_background_tasks)

isWebUIenabled = config.get('enableWebUI',False)
async def rootRouteHandler(request):
    if isWebUIenabled:
        return web.HTTPFound('/static/index.html')
    else:
        return web.Response(text="Web UI not enabled in config.yaml")

app.router.add_get('/', rootRouteHandler)

if isWebUIenabled:
    app.router.add_static('/static', 'static/')

web.run_app(app, port=config.get('port',8080))