vmcelery.py

""" Celery module for libvirt RPC calls. """
from celery import Celery
from kombu import Queue, Exchange
from socket import gethostname
from os import getenv


HOSTNAME = gethostname()
AMQP_URI = getenv('AMQP_URI')
CACHE_URI = getenv('CACHE_URI')


def to_bool(value):
    return value.lower() in ("true", "yes", "y", "t")

lib_connection = None

celery = Celery('vmcelery',
                broker=AMQP_URI,
                include=['vmdriver'])

celery.conf.update(
    CELERY_RESULT_BACKEND='cache',
    CELERY_CACHE_BACKEND=CACHE_URI,
    CELERY_TASK_RESULT_EXPIRES=300,
    CELERY_QUEUES=(
        Queue(HOSTNAME + '.vm', Exchange(
            'vmdriver', type='direct'), routing_key="vmdriver"),
    )
)

if to_bool(getenv('LIBVIRT_KEEPALIVE', "False")):
    import libvirt
    lib_connection = libvirt.open(getenv('LIBVIRT_URI'))