Source code for MainDaemon

#!/usr/bin/env python
"""  Ockle PDU and servers manager
The main daemon that runs it all

Created on Apr 25, 2012

@author: Guy Sheffer <guy.sheffer at mail.huji.ac.il>
"""
import json
import codecs
import os,sys

from common.common import loadConfig
from common.common import appendProjectPath

from networkTree.ServerNetwork import *

from straight.plugin import load
from plugins.ModuleTemplate import ModuleTemplate
from networkTree.ServerNetworkFactory import ServerNetworkFactory

from common.CommunicationHandler import CommunicationHandler

config,ETC_DIR = loadConfig()





PLUGIN_DIR = "plugins"
try:
    PLUGIN_DIR =config.get('main', 'PLUGIN_DIR')
except:
    pass

LOG_FILE_PATH = "log.txt"
try:
    LOG_FILE_PATH = appendProjectPath(config.get('main', 'LOG_FILE_PATH'))
except:
    pass

pluginList=[]
try:
    pluginList= json.loads(config.get("plugins","pluginList"))
except:
    pass


class MainDaemon(object):
    ''' The Main Daemon runs the Ockle Core, and controls the Server network,
    It loads the plugins which decide what the network behavior should be'''
    def debug(self,message):
        self.f = codecs.open(LOG_FILE_PATH,'a')
        print "DEBUG: " + str(message)
        self.f.write((time.strftime("%Y/%m/%d %H:%M:%S ", time.localtime()) + str(message.encode("utf-8")) + "\n"));
        #self.f.close()
        return
        
    def __init__(self):
        self.ETC_DIR = ETC_DIR
        OUTLET_DIR = config.get('main', 'OUTLET_DIR')
        self.OUTLETS_DIR = os.path.join(ETC_DIR,OUTLET_DIR)
        
        TESTER_DIR = config.get('main', 'TESTER_DIR')
        self.TESTERS_DIR = os.path.join(ETC_DIR,TESTER_DIR)
        
        CONTROLLER_DIR = config.get('main', 'CONTROLLER_DIR')
        self.CONTROLLERS_DIR = os.path.join(ETC_DIR,CONTROLLER_DIR) 
        
        SERVER_DIR = config.get('main', 'SERVER_DIR')
        self.SERVERS_DIR = os.path.join(ETC_DIR,SERVER_DIR)
        
        def makeSureDirExists(path):
            if not os.path.isdir(path):
                os.mkdir(path)
            return
        
        makeSureDirExists(self.OUTLETS_DIR)
        makeSureDirExists(self.CONTROLLERS_DIR)
        makeSureDirExists(self.TESTERS_DIR)
        makeSureDirExists(self.SERVERS_DIR)
        
        #config framework
        self.config = config #get it from the global
        
        #Handle the serverTree
        #Daemon.__init__(self, pidfile, stdin, stdout, stderr)
        
        factory = ServerNetworkFactory(self)
        self.servers=factory.buildNetwork(ETC_DIR)
        #self.servers.allOff()
        #servers.initiateStartup()
        
        #Communication handling system
        self.communicationHandler =CommunicationHandler(self)
        self.communicationHandler.AddCommandToList("getAvailablePluginsList", lambda dataDict: self.getAvailablePluginsListIndex(dataDict))
        self.communicationHandler.AddCommandToList("getAvailablePDUsList", lambda dataDict: {"PDUs" : json.dumps(factory.getOutletsDictIndex())})
        self.communicationHandler.AddCommandToList("getAvailableTestersList", lambda dataDict: {"Testers" : json.dumps(factory.getTestersDictIndex())})
        self.communicationHandler.AddCommandToList("getAvailableControllersList", lambda dataDict: {"Controllers" : json.dumps(factory.getControllersDictIndex())})
        self.communicationHandler.AddCommandToList("restart", lambda dataDict: self.reload(dataDict))
        
        
        self.running= True
        
        #plugin init
        self.plugins = []
        plugins = self.getPluginList()
        
        for plugin in plugins:
            if plugin.__name__ in pluginList:
                #self.debug("Loaded: " + plugin.__name__)
                self.plugins.append(plugin(self))
        
        #plugin run
        for plugin in self.plugins:
            plugin.start()
        
        #main loop
        try:
            while self.running:
                #self.debug("In main loop")
                time.sleep(10)
        except KeyboardInterrupt:
            self.debug("Got KeyboardInterrupt, exiting")
            self.shutdown()
        return
    
    def getPluginList(self):
        '''
        Get a list of all class plugins
        @return: a list of all class plugins
        '''
        return load(PLUGIN_DIR,subclasses=ModuleTemplate)
    
    def getAvailablePluginsListIndex(self,dict={}):
        ''' Get an Index of available plugins
        @return: a dict with available plugins with their name as the index, and the description as their value
        '''
        returnValue={}
        plugins = self.getPluginList()
        for plugin in plugins:
            returnValue[plugin.__name__] = plugin.__doc__
        return returnValue
    
    def reload(self,dataDict):
        ''' A general function to reload everything
        '''        
        self.shutdown()
        self.debug("Ockle is restarting")
        os.execl(sys.executable,sys.executable,__file__,"restart")
        return
    
    def shutdown(self):
        ''' Shutdown Ockle '''
        self.running = False
        for plugin in self.plugins:
            plugin.stop()
        return


pluginLoadList=["PingTester"]
    
if __name__ == "__main__":
    try:
        if sys.argv[1] == "restart":
            print "restart, wait 5 seconds"
            time.sleep(5)
    except:
        pass
    servers = MainDaemon()