Synnefo

ASTAKOS_AQUARIUM_URL                        ''                                                                              The billing (aquarium) URI

                                                                                                                            e.g. ``http://localhost:8888/user``

To update user credibility from the billing system (Aquarium), enable the queue, install snf-pithos-tools and use ``pithos-dispatcher``::

    pithos-dispatcher --exchange=aquarium --callback=astakos.im.endpoints.aquarium.consumer.on_creditevent

# Copyright 2011-2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

import requests

from django.utils import simplejson as json

from astakos.im.settings import AQUARIUM_URL

class AquariumClient():

    def get_billing(self, user, start, end):

        if not AQUARIUM_URL:

            return

        url = AQUARIUM_URL.rstrip('/')

        url = '%s/%s/bill/%d/%d' % (url, user, start, end)

        r = requests.get(url)

        try:

            return r.status_code, json.loads(r.text)

        except ValueError:

            pass

        return r.status_code, None

# Copyright 2011-2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

import logging

logging.basicConfig(format='%(asctime)s [%(levelname)s] %(name)s %(message)s',

                    datefmt='%Y-%m-%d %H:%M:%S'

                    )

logger = logging.getLogger('endpoint.aquarium')

from astakos.im.models import AstakosUser

def on_creditevent(msg):

    """

    Queue handler for updating AstakosUser has_credits

    """

    try:

        email = msg.get('userid')

        has_credits = msg.get('status') == 'on' or False

        user = AstakosUser.objects.get(email=email, is_active=True)

        user.has_credits = has_credits

        user.save()

    except BaseException, e:

        logger.exception(e)

# Copyright 2011-2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

import logging

from functools import wraps

from urlparse import urlparse

from astakos.im.settings import QUEUE_CONNECTION

if QUEUE_CONNECTION:

    from synnefo.lib.queue import (exchange_connect, exchange_send,

                                   exchange_close, UserEvent, Receipt

                                   )

QUEUE_CLIENT_ID = '3'  # Astakos.

INSTANCE_ID = '1'

RESOURCE = 'addcredits'

DEFAULT_CREDITS = 1000

logging.basicConfig(format='%(asctime)s [%(levelname)s] %(name)s %(message)s',

                    datefmt='%Y-%m-%d %H:%M:%S'

                    )

logger = logging.getLogger('endpoint.aquarium')

def call(func):

    @wraps(func)

    def wrapper(*args, **kwargs):

        if not QUEUE_CONNECTION:

            return

        try:

            body, key = func(*args, **kwargs)

            conn = exchange_connect(QUEUE_CONNECTION)

            parts = urlparse(QUEUE_CONNECTION)

            exchange = parts.path[1:]

            routing_key = key % exchange

            exchange_send(conn, routing_key, body)

            exchange_close(conn)

        except BaseException, e:

            logger.exception(e)

    return wrapper

@call

def report_user_event(user, create=False):

    eventType = 'create' if not create else 'modify'

    body = UserEvent(QUEUE_CLIENT_ID, user.email, user.is_active, eventType, {}

                     ).format()

    routing_key = '%s.user'

    return body, routing_key

@call

def report_user_credits_event(user):

    body = Receipt(QUEUE_CLIENT_ID, user.email, INSTANCE_ID, RESOURCE,

                   DEFAULT_CREDITS, details={}

                   ).format()

    routing_key = '%s.resource'

    return body, routing_key

def report_credits_event():

    # better approach?

    from astakos.im.models import AstakosUser

    map(report_user_credits_event, AstakosUser.objects.all())

# Copyright 2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from optparse import make_option

from django.core.management.base import BaseCommand, CommandError

from django.db import transaction

from astakos.im.models import sync_projects

import logging

logger = logging.getLogger(__name__)

class Command(BaseCommand):

    help = """

    Check for and perform pending project synchronization.

    Synchronization is automatic when Astakos is running.

    This command provides manual access to the synchronization.

    Details:

    Changes in project resources and memberships (such as approving

    a project or accepting a member into it) are changes in policy

    that are recorded in Astakos. These changes have to be synchronized

    with the quota system so that they come in effect.

    The quotaholder service must be running.

    Astakos is not required, but there should be no harm if it is

    running, even if under online traffic.

    """

    option_list = BaseCommand.option_list + (

        make_option('--check',

                    action='store_true',

                    dest='check',

                    default=True,

                    help="Check if projects are in sync with quotaholder (default)"),

        make_option('--trigger',

                    action='store_true',

                    dest='trigger',

                    default=False,

                    help="Sync projects to quotaholder"),

    )

    @transaction.commit_on_success

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

        trigger = options['trigger']

        try:

            log = sync_projects(sync=trigger)

            pending, (modified, reactivating, deactivating) = log

            if pending:

                self.stdout.write("Memberships pending sync:\n")

                for m in pending:

                    self.stdout.write("%s\n" % (m))

                self.stdout.write("\n")

            if modified:

                self.stdout.write("Modified projects:\n")

                for p in modified:

                    self.stdout.write("%s\n" % (p))

                self.stdout.write("\n")

            if reactivating:

                self.stdout.write("Reactivating projects:\n")

                for p in reactivating:

                    self.stdout.write("%s\n" % (p))

                self.stdout.write("\n")

            if deactivating:

                self.stdout.write("Deactivating projects:\n")

                for p in deactivating:

                    self.stdout.write("%s\n" % (p))

                self.stdout.write("\n")

        except BaseException, e:

            logger.exception(e)

            raise CommandError("Syncing failed.")

# Set the billing URI

AQUARIUM_URL = getattr(settings, 'ASTAKOS_AQUARIUM_URL', '')