mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-11-29 02:04:15 +01:00
199 lines
8.6 KiB
Python
199 lines
8.6 KiB
Python
import logging
|
|
import os
|
|
from importlib import import_module
|
|
from importlib_metadata import version, metadata
|
|
import datetime
|
|
import aiohttp
|
|
import asyncio
|
|
import yaml
|
|
import subprocess
|
|
import sys
|
|
from cbpi.api import *
|
|
from cbpi.utils.utils import load_config
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class PluginController():
|
|
modules = {}
|
|
types = {}
|
|
|
|
def __init__(self, cbpi):
|
|
self.cbpi = cbpi
|
|
|
|
def load_plugins(self):
|
|
|
|
this_directory = os.sep.join(os.path.abspath(__file__).split(os.sep)[:-1])
|
|
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("active") is True and 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:
|
|
|
|
logger.error(e)
|
|
|
|
def load_plugins_from_evn(self):
|
|
|
|
for p in self.cbpi.static_config.get("plugins", []):
|
|
|
|
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)
|
|
|
|
def register(self, name, clazz) -> None:
|
|
'''
|
|
Register a new actor type
|
|
:param name: actor name
|
|
:param clazz: actor class
|
|
:return: None
|
|
'''
|
|
logger.debug("Register %s Class %s" % (name, clazz.__name__))
|
|
|
|
if issubclass(clazz, CBPiActor):
|
|
self.cbpi.actor.types[name] = self._parse_step_props(clazz, name)
|
|
|
|
if issubclass(clazz, CBPiKettleLogic):
|
|
self.cbpi.kettle.types[name] = self._parse_step_props(clazz, name)
|
|
|
|
if issubclass(clazz, CBPiSensor):
|
|
self.cbpi.sensor.types[name] = self._parse_step_props(clazz, name)
|
|
|
|
if issubclass(clazz, CBPiStep):
|
|
self.cbpi.step.types[name] = self._parse_step_props(clazz, name)
|
|
|
|
if issubclass(clazz, CBPiExtension):
|
|
self.c = clazz(self.cbpi)
|
|
|
|
def _parse_property_object(self, p):
|
|
if isinstance(p, Property.Number):
|
|
return {"label": p.label, "type": "number", "configurable": p.configurable, "description": p.description,
|
|
"default_value": p.default_value}
|
|
elif isinstance(p, Property.Text):
|
|
return {"label": p.label, "type": "text", "configurable": p.configurable, "default_value": p.default_value,
|
|
"description": p.description}
|
|
elif isinstance(p, Property.Select):
|
|
return {"label": p.label, "type": "select", "configurable": True, "options": p.options,
|
|
"description": p.description}
|
|
elif isinstance(p, Property.Actor):
|
|
return {"label": p.label, "type": "actor", "configurable": p.configurable, "description": p.description}
|
|
elif isinstance(p, Property.Sensor):
|
|
return {"label": p.label, "type": "sensor", "configurable": p.configurable, "description": p.description}
|
|
elif isinstance(p, Property.Kettle):
|
|
return {"label": p.label, "type": "kettle", "configurable": p.configurable, "description": p.description}
|
|
|
|
def _parse_step_props(self, cls, name):
|
|
|
|
result = {"name": name, "class": cls, "properties": [], "actions": []}
|
|
|
|
if hasattr(cls, "cbpi_parameters"):
|
|
parameters = []
|
|
for p in cls.cbpi_parameters:
|
|
parameters.append(self._parse_property_object(p))
|
|
result["properties"] = parameters
|
|
for method_name, method in cls.__dict__.items():
|
|
if hasattr(method, "action"):
|
|
key = method.__getattribute__("key")
|
|
parameters = []
|
|
for p in method.__getattribute__("parameters"):
|
|
parameters.append(self._parse_property_object(p))
|
|
result["actions"].append({"method": method_name, "label": key, "parameters": parameters})
|
|
|
|
return result
|
|
|
|
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
|
|
|
|
async def load_plugin_list(self):
|
|
result = []
|
|
try:
|
|
with open(os.path.join(".", 'config', "config.yaml"), 'rt') as f:
|
|
data = yaml.load(f, Loader=yaml.FullLoader)
|
|
|
|
for p in data["plugins"]:
|
|
try:
|
|
p_metadata= metadata(p)
|
|
p_name = p_metadata['name']
|
|
p_version = p_metadata['Version']
|
|
p_summary = p_metadata['Summary']
|
|
p_homepage= p_metadata['Home-page']
|
|
p_author = p_metadata['Author']
|
|
p_author_email = p_metadata['Author-email']
|
|
p_license = p_metadata['License']
|
|
p_description = p_metadata['Description']
|
|
plugin_data = {'Name': p_name,'Version': p_version,'Summary': p_summary,'Homepage':p_homepage,'Author':p_author,'Email': p_author_email,'License': p_license,'Description': p_description}
|
|
result.append(plugin_data)
|
|
except:
|
|
pass
|
|
# print("- ({})\t{}".format(p_version,p))
|
|
except Exception as e:
|
|
print(e)
|
|
return []
|
|
pass
|
|
return result
|