Add IpLocationPlugin to PluginManager

This commit is contained in:
mac-zhou
2024-03-19 19:46:44 +08:00
parent dc51eb0ff1
commit 23b5cde07f
2 changed files with 47 additions and 1 deletions

View File

@@ -15,6 +15,7 @@ from plugins.deepl import DeeplTranslatePlugin
from plugins.worldtimeapi import WorldTimeApiPlugin
from plugins.whois_ import WhoisPlugin
from plugins.webshot import WebshotPlugin
from plugins.iplocation import IpLocationPlugin
class PluginManager:
@@ -40,6 +41,7 @@ class PluginManager:
'auto_tts': AutoTextToSpeech,
'whois': WhoisPlugin,
'webshot': WebshotPlugin,
'iplocation': IpLocationPlugin,
}
self.plugins = [plugin_mapping[plugin]() for plugin in enabled_plugins if plugin in plugin_mapping]
@@ -69,4 +71,4 @@ class PluginManager:
def __get_plugin_by_function_name(self, function_name):
return next((plugin for plugin in self.plugins
if function_name in map(lambda spec: spec.get('name'), plugin.get_spec())), None)
if function_name in map(lambda spec: spec.get('name'), plugin.get_spec())), None)