craftbeerpi3/modules/core/baseapi.py

from proptypes import *

class BaseAPI(object):

    def __init__(self, cbpi):
        self.cbpi = cbpi
        self.cbpi.cache[self.key] = {}


    def init(self):
        for name, value in self.cbpi.cache[self.key].iteritems():

            value["class"].init_global()

    def parseProps(self, key, cls, **options):

        name = cls.__name__
        tmpObj = cls()

        try:
            doc = tmpObj.__doc__.strip()
        except:
            doc = ""

        if self.cbpi.cache.get(key) is None:
            self.cbpi.cache[key] = {}
        self.cbpi.logger.debug(name)
        self.cbpi.cache.get(key)[name] = {"name": name, "class": cls, "description":doc, "properties": [], "actions": []}
        self.cbpi.cache.get(key)[name].update(options)
        members = [attr for attr in dir(tmpObj) if not callable(getattr(tmpObj, attr)) and not attr.startswith("__")]
        for m in members:
            t = tmpObj.__getattribute__(m)

            if isinstance(t, Property.Number):
                self.cbpi.cache.get(key)[name]["properties"].append({"name": m, "label": t.label, "type": "number", "configurable": t.configurable, "description": t.description, "default_value": t.default_value, "unit": t.unit})
            elif isinstance(t, Property.Text):
                self.cbpi.cache.get(key)[name]["properties"].append({"name": m, "label": t.label, "type": "text", "required": t.required, "configurable": t.configurable, "description": t.description, "default_value": t.default_value})
            elif isinstance(t, Property.Select):
                self.cbpi.cache.get(key)[name]["properties"].append({"name": m, "label": t.label, "type": "select",  "configurable": True, "options": t.options, "description": t.description})
            elif isinstance(t, Property.Actor):
                self.cbpi.cache.get(key)[name]["properties"].append({"name": m, "label": t.label, "type": "actor",  "configurable": True, "description": t.description})
            elif isinstance(t, Property.Sensor):
                self.cbpi.cache.get(key)[name]["properties"].append({"name": m, "label": t.label, "type": "sensor", "configurable": True, "description": t.description})
            elif isinstance(t, Property.Kettle):
                self.cbpi.cache.get(key)[name]["properties"].append({"name": m, "label": t.label, "type": "kettle", "configurable": True, "description": t.description})

        for method_name, method in cls.__dict__.iteritems():
            if hasattr(method, "action"):
                label = method.__getattribute__("label")
                parameters = method.__getattribute__("parameters")
                props = []
                if parameters is not None:
                    for k, t in parameters.iteritems():
                        if isinstance(t, Property.Number):
                            props.append({"name": k, "label": t.label, "type": "number", "configurable": t.configurable, "description": t.description, "default_value": t.default_value, "unit": t.unit})
                        elif isinstance(t, Property.Text):
                            props.append({"name": k, "label": t.label, "type": "text", "required": t.required, "configurable": t.configurable, "description": t.description, "default_value": t.default_value})
                        elif isinstance(t, Property.Select):
                            props.append({"name": k, "label": t.label, "type": "select", "configurable": True, "options": t.options, "description": t.description})
                        elif isinstance(t, Property.Actor):
                            props.append({"name": k, "label": t.label, "type": "actor", "configurable": True, "description": t.description})
                        elif isinstance(t, Property.Sensor):
                            props.append({"name": k, "label": t.label, "type": "sensor", "configurable": True, "description": t.description})
                        elif isinstance(t, Property.Kettle):
                            props.append({"name": k, "label": t.label, "type": "kettle", "configurable": True, "description": t.description})

                self.cbpi.cache.get(key)[name]["actions"].append({"method": method_name, "label": label, "properties":props })
        return cls


class SensorAPI(BaseAPI):

    key = "sensor_types"

    def type(self, description="Step", **options):
       def decorator(f):
            BaseAPI.parseProps(self, self.key,f, description=description)
            return f
       return decorator

    def action(self, label, parameters=None):

        def real_decorator(func):
            func.action = True
            func.parameters = parameters
            func.label = label
            return func
        return real_decorator


class StepAPI(BaseAPI):

    key = "step_types"

    def init(self):
        pass

    def type(self, description="Step", **options):
       def decorator(f):
            BaseAPI.parseProps(self, self.key,f, description=description)
            return f
       return decorator

    def action(self, label, parameters=None):
        def real_decorator(func):
            func.action = True
            func.label = label
            func.parameters = parameters
            return func
        return real_decorator




class ActorAPI(BaseAPI):

    key = "actor_types"

    def type(self, description="", **options):
       def decorator(f):
            BaseAPI.parseProps(self, self.key, f, description=description)
            return f
       return decorator

    def action(self, label, parameters=None):
        def real_decorator(func):
            func.action = True
            func.label = label
            func.parameters = parameters
            return func
        return real_decorator



class KettleAPI(BaseAPI):

    key = "controller_types"

    def controller(self, description="", **options):
       def decorator(f):
            BaseAPI.parseProps(self, self.key,f,description=description)
            return f
       return decorator

    def action(self, label, parameters=None):
        def real_decorator(func):
            func.action = True
            func.label = label
            func.parameters = parameters
            return func
        return real_decorator

class FermenterAPI(BaseAPI):

    key = "fermentation_controller_types"

    def controller(self, description="Step", **options):
       def decorator(f):
            BaseAPI.parseProps(self, self.key,f,description=description)
            return f
       return decorator

    def action(self, label, parameters=None):
        def real_decorator(func):
            func.action = True
            func.label = label
            func.parameters = parameters
            return func
        return real_decorator

class CoreAPI(BaseAPI):

    key = "core"

    def __init__(self, cbpi):
        self.cbpi = cbpi
        self.cbpi.cache["actions"] = {}
        self.cbpi.cache["init"] = []
        self.cbpi.cache["js"] = {}
        self.cbpi.cache["background"] = []
        self.cbpi.cache["web_menu"] =[]

    def init(self):
        self.cbpi.cache["init"] = sorted(self.cbpi.cache["init"], key=lambda k: k['order'])
        for value in self.cbpi.cache.get("init"):


            value["function"](self.cbpi)

        def job(interval, method):
            while True:
                try:
                    method(self.cbpi)
                except Exception as e:
                    self.cbpi.logger.error(e)
                self.cbpi.sleep(interval)

        for value in self.cbpi.cache.get("background"):
            t = self.cbpi.start_background_task(target=job,  interval=value.get("interval"),  method=value.get("function"))

    def action(self, **options):
       def decorator(f):
            BaseAPI.parseProps(self, "actions", f)
            return f
       return decorator

    def add_js(self, name, file):
        self.cbpi.cache["js"][name] = file

    def add_menu_link(self, name, path):
        self.cbpi.cache["web_menu"].append(dict(name=name, path=path))

    def initializer(self, order=0, **options):
        def decorator(f):
             self.cbpi.cache.get("init").append({"function": f, "order": order})
             return f
        return decorator

    def backgroundtask(self, key, interval, **options):
        def decorator(f):
             self.cbpi.cache.get("background").append({"function": f, "key": key, "interval": interval})
             return f
        return decorator

    def listen(self, name, method=None, async=False):

        if method is not None:
            if self.cbpi.eventbus.get(name) is None:
                self.cbpi.eventbus[name] = []
            self.cbpi.eventbus[name].append({"function": method, "async": async})
        else:
            def real_decorator(function):
                if self.cbpi.eventbus.get(name) is None:
                    self.cbpi.eventbus[name] = []
                self.cbpi.eventbus[name].append({"function": function, "async": async})
                def wrapper(*args, **kwargs):
                    return function(*args, **kwargs)
                return wrapper
            return real_decorator


class Buzzer(object):

    def beep(self):
        pass

class Addon(object):
    def __init__(self, cbpi):
        self.step = StepAPI(cbpi)
        self.actor = ActorAPI(cbpi)
        self.sensor = SensorAPI(cbpi)
        self.kettle = KettleAPI(cbpi)
        self.fermenter = FermenterAPI(cbpi)
        self.core = CoreAPI(cbpi)

    def init(self):
        self.core.init()
        self.step.init()
        self.actor.init()
        self.sensor.init()