cloud-computing-msc-ai-examples/faafo/bin/faafo-worker

#!/usr/bin/env python

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import os
import sys

import kombu
from oslo_config import cfg
from oslo_log import log

from faafo.worker import service as worker
from faafo import version

LOG = log.getLogger('faafo.worker')
CONF = cfg.CONF

# If ../faafo/__init__.py exists, add ../ to Python search path, so that
# it will override what happens to be installed in /usr/(local/)lib/python...
possible_topdir = os.path.normpath(os.path.join(os.path.abspath(sys.argv[0]),
                                   os.pardir,
                                   os.pardir))
if os.path.exists(os.path.join(possible_topdir, 'faafo', '__init__.py')):
    sys.path.insert(0, possible_topdir)

if __name__ == '__main__':
    log.register_options(CONF)
    log.set_defaults()

    CONF(project='worker', prog='faafo-worker',
         default_config_files=['/etc/faafo/faafo.conf'],
         version=version.version_info.version_string())

    log.setup(CONF, 'worker',
              version=version.version_info.version_string())

    connection = kombu.Connection(CONF.transport_url)
    server = worker.Worker(connection)
    try:
        server.run()
    except KeyboardInterrupt:
        LOG.info("Caught keyboard interrupt. Exiting.")