def handle(self, *args, **options):

        if not options["task_code"]:
            log.error("Expected a task_code argument, use --help.")
            return

        numeric_level = getattr(logging, options["log_level"].upper(), None)
        if not isinstance(numeric_level, int):
            log.error("Invalid log level: %s" % options["log_level"])
            numeric_level = 10

        broker = BrokerConnection()
        connection = broker.connect_to_broker()

        removeFileHandlers()
        setFileHandler(options["worker_nr"])
        setLevelToAllHandlers(numeric_level)

        if connection is None:
            log.error("Could not connect to broker.")
            return

        action = ActionTask(options["task_code"], options["worker_nr"])

        logging.handlers.AMQPMessageHandler = AMQPMessageHandler
        broker_logging_handler = logging.handlers.AMQPMessageHandler(
            action, numeric_level)
        action.set_broker_logging_handler(broker_logging_handler)

        task_worker = Worker(connection,
                             options["task_code"],
                             action,
                             options["worker_nr"])
        task_worker.run_worker()
        removeFileHandlers()
 def test_action(self, child, task_code, worker_nr):
     #action = ActionTask('120', '10')
     from flooding_worker.worker.worker import Worker
     from flooding_worker.worker.broker_connection import BrokerConnection
     connection = BrokerConnection().connect_to_broker()
     action = ActionTask(task_code, worker_nr)
     worker = Worker(connection, task_code, action, worker_nr)
     worker.run_worker()
    def handle(self, *args, **options):

        broker = BrokerConnection()
        connection = broker.connect_to_broker()

        removeFileHandlers()
        setFileHandler('logging')

        if connection is None:
            log.error("Could not connect to broker.")
            return

        action = ActionLogging()

        logging_worker = Worker(connection,
                                options["task_code"],
                                action)
        logging_worker.run_worker()
Пример #4
0
    def handle(self, *args, **options):

        numeric_level = getattr(logging, options["log_level"].upper(), None)
        if not isinstance(numeric_level, int):
            log.error("Invalid log level: %s" % options["log_level"])
            numeric_level = 10

        broker = BrokerConnection()
        connection = broker.connect_to_broker()

        removeFileHandlers()
        setFileHandler('queue')

        if connection is None:
            log.error("Could not connect to broker.")
            return
        # TODO CREATE ActionPreority
        action = ActionPriority(connection,
                                options["task_code"])

        queue_worker = Worker(connection,
                              options["task_code"],
                              action)
        queue_worker.run_worker()