Produits Entr'ouvert » Corbo

# corbo - Announces Manager

# Copyright (C) 2016 Entr'ouvert

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from hobo.agent.common.management.commands import hobo_notify

from corbo.models import Subscription

class Command(hobo_notify.Command):

    @classmethod

    def process_notification(cls, tenant, notification):

        super(CommonCommand, self).process_notification(cls, tenant, notification)

        object_type = notification['objects']['@type']

        full = notification['full'] if 'full' in notification else False

        if object_type != 'user':

            return

        data = notification['objects']['data']

        action = notification['@type']

        if full:

            uuids = [obj['uuid'] for obj in data]

            Subscription.objects.exclude(uuid__in=uuids).delete()

        for obj in data:

            for subscription in Subscription.objects.filter(uuid=obj['uuid']):

                if action == 'provision':

                    if subscription.identifier.startswith('mailto:'):

                        subscription.identifier = 'mailto:%s' % obj['email']

                    elif subscription.identifier.startswith('sms:'):

                        if obj['mobile']:

                            subscription.identifier = 'sms:%s' % obj['mobile']

                        else:

                            subscription.delete()

                            continue

                    subscription.save()

                elif action == 'deprovision':

                    subscription.delete()

INSTALLED_APPS = ('corbo.hobo_agent', ) + INSTALLED_APPS