import asyncio import logging import os from importlib import import_module import aiohttp import yaml from aiohttp import web from cbpi.api import * from cbpi.utils.utils import load_config, json_dumps logger = logging.getLogger(__name__) import subprocess import sys class PluginController(): modules = {} types = {} def __init__(self, cbpi): self.cbpi = cbpi self.cbpi.register(self, "/plugin") @classmethod async def load_plugin_list(self): async with aiohttp.ClientSession() as session: async with session.get('https://raw.githubusercontent.com/Manuel83/craftbeerpi-plugins/master/plugins_v4.yaml') as resp: if (resp.status == 200): data = yaml.load(await resp.text()) return data def load_plugins(self): this_directory = os.path.dirname(__file__) for filename in os.listdir(os.path.join(this_directory, "../extension")): if os.path.isdir(os.path.join(this_directory, "../extension/") + filename) is False or filename == "__pycache__": continue try: logger.info("Trying to load plugin %s" % filename) data = load_config(os.path.join(this_directory, "../extension/%s/config.yaml" % filename)) if (data.get("version") == 4): self.modules[filename] = import_module("cbpi.extension.%s" % (filename)) self.modules[filename].setup(self.cbpi) logger.info("Plugin %s loaded successful" % filename) else: logger.warning("Plugin %s is not supporting version 4" % filename) except Exception as e: print(e) logger.error(e) def load_plugins_from_evn(self): plugins = [] this_directory = os.path.dirname(__file__) with open("./config/plugin_list.txt") as f: plugins = f.read().splitlines() plugins = list(set(plugins)) for p in plugins: logger.debug("Load Plugin %s" % p) try: logger.info("Try to load plugin: %s " % p) self.modules[p] = import_module(p) self.modules[p].setup(self.cbpi) logger.info("Plugin %s loaded successfully" % p) except Exception as e: logger.error("FAILED to load plugin %s " % p) logger.error(e) @on_event("job/plugins_install/done") async def done(self, **kwargs): self.cbpi.notify(key="p", message="Plugin installed ", type="success") print("DONE INSTALL PLUGIN", kwargs) @request_mapping(path="/install", method="GET", auth_required=False) async def install_plugin(self, request): """ --- description: Install Plugin tags: - Plugin produces: - application/json responses: "204": description: successful operation. Return "pong" text "405": description: invalid HTTP Method """ async def install(name): await asyncio.sleep(5) subprocess.call([sys.executable, "-m", "pip", "install", name]) print("OK") await self.cbpi.job.start_job(install('requests'), "requests", "plugins_install") return web.Response(status=204) @request_mapping(path="/list", method="GET", auth_required=False) async def get_plugins(self, request): """ --- description: Get a list of avialable plugins tags: - Plugin produces: - application/json responses: "200": description: successful operation. Return "pong" text "405": description: invalid HTTP Method """ return web.json_response(await self.load_plugin_list(), dumps=json_dumps) def register(self, name, clazz) -> None: ''' Register a new actor type :param name: actor name :param clazz: actor class :return: None ''' logger.info("Register %s Class %s" % (name, clazz.__name__)) if issubclass(clazz, CBPiActor): # self.cbpi.actor.types[name] = {"class": clazz, "config": self._parse_props(clazz)} self.cbpi.actor.types[name] = self._parse_props(clazz) if issubclass(clazz, CBPiSensor): self.cbpi.sensor.types[name] = self._parse_props(clazz) if issubclass(clazz, CBPiKettleLogic): self.cbpi.kettle.types[name] = self._parse_props(clazz) if issubclass(clazz, CBPiSimpleStep): self.cbpi.step.types[name] = self._parse_props(clazz) if issubclass(clazz, CBPiExtension): self.c = clazz(self.cbpi) def _parse_props(self, cls): name = cls.__name__ result = {"name": name, "class": cls, "properties": [], "actions": []} tmpObj = cls(cbpi=None, managed_fields=None) members = [attr for attr in dir(tmpObj) if not callable(getattr(tmpObj, attr)) and not attr.startswith("__")] for m in members: if isinstance(tmpObj.__getattribute__(m), Property.Number): t = tmpObj.__getattribute__(m) result["properties"].append( {"name": m, "label": t.label, "type": "number", "configurable": t.configurable, "description": t.description, "default_value": t.default_value}) elif isinstance(tmpObj.__getattribute__(m), Property.Text): t = tmpObj.__getattribute__(m) result["properties"].append( {"name": m, "label": t.label, "type": "text", "configurable": t.configurable, "default_value": t.default_value, "description": t.description}) elif isinstance(tmpObj.__getattribute__(m), Property.Select): t = tmpObj.__getattribute__(m) result["properties"].append( {"name": m, "label": t.label, "type": "select", "configurable": True, "options": t.options, "description": t.description}) elif isinstance(tmpObj.__getattribute__(m), Property.Actor): t = tmpObj.__getattribute__(m) result["properties"].append({"name": m, "label": t.label, "type": "actor", "configurable": t.configurable, "description": t.description}) elif isinstance(tmpObj.__getattribute__(m), Property.Sensor): t = tmpObj.__getattribute__(m) result["properties"].append({"name": m, "label": t.label, "type": "sensor", "configurable": t.configurable, "description": t.description}) elif isinstance(tmpObj.__getattribute__(m), Property.Kettle): t = tmpObj.__getattribute__(m) result["properties"].append({"name": m, "label": t.label, "type": "kettle", "configurable": t.configurable, "description": t.description}) for method_name, method in cls.__dict__.items(): if hasattr(method, "action"): key = method.__getattribute__("key") parameters = method.__getattribute__("parameters") result["actions"].append({"method": method_name, "label": key, "parameters": parameters}) return result