fix: Cleanup plugin.py
All checks were successful
Build Docker / Build Image (push) Successful in 1m13s

This commit is contained in:
Nathan Woodburn 2024-11-21 19:35:33 +11:00
parent 40f520ba5e
commit 3290240a25
Signed by: nathanwoodburn
GPG Key ID: 203B000478AD0EF1

View File

@ -5,7 +5,6 @@ import sys
import hashlib
def listPlugins():
plugins = []
for file in os.listdir("plugins"):
@ -45,6 +44,7 @@ def pluginExists(plugin: str):
return True
return False
def verifyPlugin(plugin: str):
signatures = []
try:
@ -75,12 +75,6 @@ def hashPlugin(plugin: str):
return sha256.hexdigest()
def getPluginData(pluginStr: str):
plugin = importlib.import_module("plugins."+pluginStr)
@ -104,10 +98,12 @@ def getPluginData(pluginStr: str):
return info
def getPluginFunctions(plugin: str):
plugin = importlib.import_module("plugins."+plugin)
return plugin.functions
def runPluginFunction(plugin: str, function: str, params: dict, authentication: str):
plugin_module = importlib.import_module("plugins."+plugin)
if function not in plugin_module.functions:
@ -134,7 +130,6 @@ def runPluginFunction(plugin: str, function: str, params: dict, authentication:
if pluginHash not in signatures:
return {"error": "Plugin not verified"}
# Call the function with provided parameters
try:
result = plugin_function(params, authentication)
@ -144,14 +139,17 @@ def runPluginFunction(plugin: str, function: str, params: dict, authentication:
return {"error": str(e)}
# return plugin.runFunction(function, params, authentication)
def getPluginFunctionInputs(plugin: str, function: str):
plugin = importlib.import_module("plugins."+plugin)
return plugin.functions[function]["params"]
def getPluginFunctionReturns(plugin: str, function: str):
plugin = importlib.import_module("plugins."+plugin)
return plugin.functions[function]["returns"]
def getDomainFunctions():
plugins = listPlugins()
domainFunctions = []
@ -166,6 +164,7 @@ def getDomainFunctions():
})
return domainFunctions
def getSearchFunctions():
plugins = listPlugins()
searchFunctions = []
@ -180,6 +179,7 @@ def getSearchFunctions():
})
return searchFunctions
def getDashboardFunctions():
plugins = listPlugins()
dashboardFunctions = []