Ahenk/opt/ahenk/ahenkd.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Author: İsmail BAŞARAN <ismail.basaran@tubitak.gov.tr> <basaran.ismaill@gmail.com>
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>

from base.config.ConfigManager import ConfigManager
from base.deamon.BaseDeamon import BaseDaemon
from base.logger.AhenkLogger import Logger
from base.Scope import Scope
from base.messaging.Messaging import Messaging
from base.messaging.MessageReceiver import MessageReceiver
from base.messaging.MessageSender import MessageSender
from base.execution.ExecutionManager import ExecutionManager
from base.registration.Registration import Registration
from base.messaging.MessageResponseQueue import MessageResponseQueue
from base.event.EventManager import EventManager
from base.plugin.PluginManager import PluginManager
from base.task.TaskManager import TaskManager
from multiprocessing import Process
import sys


class AhenkDeamon(BaseDaemon):
    """docstring for AhenkDeamon"""

    def reload(self,msg):
        # reload service here
        pass

    def run(self):
        print ("Ahenk running...")

        globalscope = Scope()
        globalscope.setInstance(globalscope)

        configFilePath='/etc/ahenk/ahenk.conf'
        configfileFolderPath='/etc/ahenk/config.d/'

        #configuration manager must be first load
        configManager = ConfigManager(configFilePath,configfileFolderPath)
        config = configManager.read()
        globalscope.setConfigurationManager(config)

        # Logger must be second
        logger = Logger()
        logger.info("[AhenkDeamon] Log was set")
        globalscope.setLogger(logger)

        eventManager = EventManager()
        globalscope.setEventManager(eventManager)
        logger.info("[AhenkDeamon] Event Manager was set")

        messageManager = Messaging()
        globalscope.setMessageManager(messageManager)
        logger.info("[AhenkDeamon] Message Manager was set")

        pluginManager = PluginManager()
        pluginManager.loadPlugins()
        globalscope.setPluginManager(pluginManager)
        logger.info("[AhenkDeamon] Plugin Manager was set")

        taskManger = TaskManager()
        globalscope.setTaskManager(taskManger)
        logger.info("[AhenkDeamon] Task Manager was set")

        registration=Registration()
        globalscope.setRegistration(registration)
        logger.info("[AhenkDeamon] Registration was set")

        execution_manager=ExecutionManager()
        globalscope.setExecutionManager(execution_manager)
        logger.info("[AhenkDeamon] Execution Manager was set")


        while registration.is_registered() is False:
            logger.debug("[AhenkDeamon] Attempting to register")
            registration.registration_request()

        logger.info("[AhenkDeamon] Ahenk is registered")

        message_receiver = MessageReceiver()
        rec_process = Process(target=message_receiver.connect_to_server)
        rec_process.start()
        logger.info("[AhenkDeamon] Receiver was set")

        if registration.is_ldap_registered() is False:
            logger.debug("[AhenkDeamon] Attempting to registering ldap")
            registration.ldap_registration_request() #TODO bu mesaj daha kısa olabilir

        logger.info("[AhenkDeamon] LDAP registration of Ahenk is completed")

		#login
        logger.info("[AhenkDeamon] Logining...")
        message_sender=MessageSender(messageManager.login_msg(),None)
        message_sender.connect_to_server()


        #logout
        #message_sender=MessageSender(messageManager.logout_msg(),None)
        #message_sender.connect_to_server()

        #rec_process.terminate()

        """
            this is must be created after message services
            responseQueue = queue.Queue()
            messageResponseQueue = MessageResponseQueue(responseQueue)
            messageResponseQueue.setDaemon(True)
            messageResponseQueue.start()
            globalscope.setResponseQueue(responseQueue)
        """


if __name__ == '__main__':

    pidfilePath='/var/run/ahenk.pid'

    ahenkdaemon = AhenkDeamon(pidfilePath)

    print (sys.argv)

    if len(sys.argv) == 2:
        if sys.argv[1] == "start":
            print ("starting")
            ahenkdaemon.run()
            #print (ahenkdaemon.get_pid())
        elif sys.argv[1] == 'stop':
            ahenkdaemon.stop()
        elif sys.argv[1] == 'restart':
            ahenkdaemon.restart()
        elif sys.argv[1] == 'status':
            # print (status)
            pass
        else:
            print ('Unknown command. Usage : %s start|stop|restart|status' % sys.argv[0])
            sys.exit(2)
        sys.exit(0)
    else:
        print ('Usage : %s start|stop|restart|status' % sys.argv[0])
        sys.exit(2)