Core.py

from PluginManager import PluginManager
from PluginDispatcher import PluginDispatcher
from Configuration import ConfigFile
from Util import call
from re import match
from sys import path
from os import getcwd
from Util import dictJoin
from Logging import LogFile

path.append(getcwd())

log = LogFile("Core")


class Core:
    _PluginManager = None
    _PluginDispatcher = None
    _ResponseObject = None
    _Connector = None
    _Config = None

    def _LoadConnector(self, ConName):
        try:
            con = __import__("%s.Connector" % ConName,
                    globals(), locals(), "Connector")
            log.debug("Got connector:", con)
            cls = getattr(con, "Connector", None)
        except :
            log.exception("Exception while loading connector")
            cls = None
        log.debug("Connectors class", cls)
        if cls:
            c = cls()
            log.debug("Connector constructed")
            return c

        log.critical("No connector")
        return cls

    def HandleEvent(self, event):
        log.dict(event,"HandleEvent")

        pm = self._PluginManager
        if not pm:
            log.warning("No plugin manager")
            return

        pd = self._PluginDispatcher
        if not pd:
            log.warning("No plugin dispatcher")
            return

        ro = self._ResponseObject
        if not ro:
            log.warning("no response object")
            pass

        matches = pm.GetMatchingFunctions(event)
        log.debug("Matched %i hook(s)." % len(matches))

        for inst, func, args, servs in matches:
            newEvent = dictJoin(event, dictJoin(args,
                {"self": inst, "response": ro}))
            log.debug("Services found for plugin:", servs)
            if servs:
                log.debug("Event before processing:", newEvent)

            servDict={}
            servDict["event"]=newEvent
            servDict["pm"]=self._PluginManager
            servDict["pd"]=self._PluginDispatcher
            servDict["ro"]=self._ResponseObject
            servDict["c"]=self._Connector
            servDict["core"]=self
            servDict["config"]=self._Config
            for servName in servs:
                serv = pm.GetService(servName)
                log.debug("Processing service",servName,serv)
                call(serv.onEvent,servDict)

            if servs:
                log.dict(newEvent,"Event after processing:")
            #issue 5 fix goes here
            newEvent.update(servDict)
            pd.Enqueue((func, newEvent))

    def __init__(self):
        self._Config = ConfigFile("Core")
        if not self._Config:
            log.critical("No log file loaded!")
            return
        ConName = self._Config["Core", "Provider"]
        if ConName == None:
            log.critical("No Core:Provider in Core.cfg")
            del self._Connector
            return 
            
        self._Connector=self._LoadConnector(ConName) 
    	if self._Connector:
            self._PluginManager = PluginManager(ConName)
            self._PluginDispatcher = PluginDispatcher()
            self._Connector.SetEventHandler(self.HandleEvent)
            self._ResponseObject = self._Connector.GetResponseObject()
            self._PluginDispatcher.SetResponseHandler(
                self._Connector.HandleResponse)

    def Start(self):
        if not self._Connector:
            log.warning("Could not start, no connector.")
            return

        log.debug("Starting")
        log.debug("Auto loading plugins")
        self.AutoLoad()
        log.debug("Auto load complete")

        if self._Connector:
            log.debug("Connector starting")
            self._Connector.Start()
        #else log error?

    def Stop(self):
        log.debug("Stopping")
        if self._PluginDispatcher:
            self._PluginDispatcher.Stop()
        if self._PluginManager:
            self._PluginManager.Stop()
        if self._Connector:
            self._Connector.Stop()

    def AutoLoad(self):
        if not self._PluginManager:
            return
        pm = self._PluginManager
        log.note("Starting autoload", "Root:" + pm.root)
        cf = ConfigFile(pm.root, "Autoload")
        lines = ["Configuration:"]
        for i in cf:
            lines.append(i)
            for j in cf[i]:
                lines.append("  %s=%s"%(j,cf[i,j]))
        log.debug(*lines)
        if cf:
            log.debug("Autoloading plugins.")

            names = cf["Plugins", "Names"]
            log.debug("Autoloading plugins", names)
            if names:
                for name in names.split():
                    pm.LoadPlugin(name)
                log.debug("Autoloading finished.")
                pd=self._PluginDispatcher
                handler = pd.GetResponseHandler()
                log.debug("Updating dedicated thread pool",self._ResponseObject,handler)
                pd.EnsureDedicated(pm.GetDedicated(),self._ResponseObject,handler)
                    
        else:
            log.note("No Autoload configuration file")

 

if __name__ == "__main__":
    try:
        c = Core()
        try:
            c.Start()
        except:
            log.exception("Exception while starting.")
        c.Stop()
    except:
        log.exception("Exception while stopping.")
    log.debug("End of core")