Produits Entr'ouvert » Hobo

from django.db.models import post_save, post_delete

from django_rbac.utils import get_role_model

from hobo.agent.common import notify_agents

from authentic2.utils import to_list

from authentic2.saml.models import LibertyProvider

@to_list

def get_audience():

    for provider in LibertyProvider.objects.all():

        yield provider.entity_id

def notify_roles_post_save(sender, instance, created, **kwargs):

    notify_agents({

        '@type': 'provision',

        'audience': get_audience(),

        'objects': [

            {

                '@type': 'role',

                'uuid': instance.uuid,

                'name': instance.name,

                'slug': instance.slug,

                'description': instance.description,

            }

        ]

    })

def notify_roles_post_delete(sender, instance, **kwargs):

    notify_agents({

        '@type': 'deprovision',

        'audience': get_audience(),

        'objects': [

            {

                '@type': 'role',

                'uuid': instance.uuid,

                'name': instance.name,

                'slug': instance.slug,

                'description': instance.description,

            }

        ]

    })

    def ready(self):

        Role = get_role_model()

        post_save.connect(notify_roles_post_save, Role)

        post_delete.connect(notify_roles_post_delete, Role)

from celery import Celery

from kombu.common import Broadcast

from django.conf import settings

from django.db import connection

def notify_agents(data):

    '''Send notifications to all other tenants'''

    notification = {

        'tenant': connection.get_tenant().domain_url,

        'data': data,

    }

    with Celery('hobo', broker=settings.BROKER_URL) as app:

        app.conf.update(

            CELERY_TASK_SERIALIZER='json',

            CELERY_ACCEPT_CONTENT=['json'],

            CELERY_RESULT_SERIALIZER='json',

            CELERY_QUEUES=(Broadcast('broadcast_tasks'), )

        )

        # see called method in hobo.agent.worker.celery

        app.send_task('hobo-notify',

                      (notification,),

                      expires=settings.BROKER_TASK_EXPIRES,

                      queue='broadcast_tasks')

from . import notify

@app.task(name='hobo-notify', bind=True, acks_late=True)

def hobo_notify(self, data):

    # do something with data

    notify.notify(data)

import os.path

import subprocess

import json

from . import settings

def notify(data):

    for command in settings.NOTIFY_COMMANDS:

        if not os.path.exists(command):

            continue

        what = command + ' notify -'

        cmd_process = subprocess.Popen(what, shell=True, stdin=subprocess.PIPE,

                                       stdout=subprocess.PIPE)

        cmd_process.communicate(input=json.dumps(data))

        if cmd_process.returncode != 0:

            raise RuntimeError('command "%s" failed' % what)

NOTIFY_COMMANDS = ['/usr/sbin/wcsctl']