mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-14 08:17:13 +03:00
119 lines
3.0 KiB
Python
119 lines
3.0 KiB
Python
"""
|
|
Plugin hooks for EDMC - Ian Norton
|
|
"""
|
|
import os
|
|
import imp
|
|
import sys
|
|
|
|
from config import config
|
|
|
|
"""
|
|
Dictionary of loaded plugin modules.
|
|
"""
|
|
PLUGINS = dict()
|
|
|
|
|
|
def find_plugins():
|
|
"""
|
|
Look for plugin entry points.
|
|
:return:
|
|
"""
|
|
found = dict()
|
|
plug_folders = os.listdir(config.plugin_dir)
|
|
for name in plug_folders:
|
|
loadfile = os.path.join(config.plugin_dir, name, "load.py")
|
|
if os.path.isfile(loadfile):
|
|
found[name] = loadfile
|
|
return found
|
|
|
|
|
|
def load_plugins():
|
|
"""
|
|
Load all found plugins
|
|
:return:
|
|
"""
|
|
found = find_plugins()
|
|
imp.acquire_lock()
|
|
for plugname in found:
|
|
try:
|
|
with open(found[plugname], "rb") as plugfile:
|
|
plugmod = imp.load_module(plugname, plugfile, found[plugname],
|
|
(".py", "r", imp.PY_SOURCE))
|
|
if "plugin_start" in dir(plugmod):
|
|
plugmod.plugin_start()
|
|
PLUGINS[plugname] = plugmod
|
|
|
|
except Exception as plugerr:
|
|
sys.stderr.write('%s\n' % plugerr) # appears in %TMP%/EDMarketConnector.log in packaged Windows app
|
|
|
|
imp.release_lock()
|
|
|
|
|
|
def _get_plugin_func(plugname, funcname):
|
|
"""
|
|
Get a function from a plugin, else return None if it isn't implemented.
|
|
:param plugname:
|
|
:param funcname:
|
|
:return:
|
|
"""
|
|
if funcname in dir(PLUGINS[plugname]):
|
|
return getattr(PLUGINS[plugname], funcname)
|
|
return None
|
|
|
|
|
|
def get_plugin_app(plugname, parent):
|
|
"""
|
|
If the plugin provides mainwindow content create and return it.
|
|
:param plugname: name of the plugin
|
|
:param parent: the parent frame for this entry.
|
|
:return:
|
|
"""
|
|
plugin_app = _get_plugin_func(plugname, "plugin_app")
|
|
if plugin_app:
|
|
return plugin_app(parent)
|
|
return None
|
|
|
|
|
|
def get_plugin_pref(plugname, parent):
|
|
"""
|
|
If the plugin provides a prefs frame, create and return it.
|
|
:param plugname: name of the plugin
|
|
:param parent: the parent frame for this preference tab.
|
|
:return:
|
|
"""
|
|
plugin_prefs = _get_plugin_func(plugname, "plugin_prefs")
|
|
if plugin_prefs:
|
|
return plugin_prefs(parent)
|
|
return None
|
|
|
|
|
|
def notify_system_changed(timestamp, system):
|
|
"""
|
|
Send notification data to each plugin when we arrive at a new system.
|
|
:param timestamp:
|
|
:param system:
|
|
:return:
|
|
"""
|
|
for plugname in PLUGINS:
|
|
system_changed = _get_plugin_func(plugname, "system_changed")
|
|
if system_changed:
|
|
try:
|
|
system_changed(timestamp, system)
|
|
except Exception as plugerr:
|
|
print plugerr
|
|
|
|
|
|
def notify_newdata(data):
|
|
"""
|
|
Send the latest EDMC data from the FD servers to each plugin
|
|
:param data:
|
|
:return:
|
|
"""
|
|
for plugname in PLUGINS:
|
|
cmdr_data = _get_plugin_func(plugname, "cmdr_data")
|
|
if cmdr_data:
|
|
try:
|
|
cmdr_data(data)
|
|
except Exception as plugerr:
|
|
print plugerr
|