Synnefo

# Copyright 2012, 2013 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 django.core.management.base import CommandError

from optparse import make_option

from astakos.im.models import AstakosUser, get_latest_terms, Chain

from astakos.im.quotas import list_user_quotas

from synnefo.lib.ordereddict import OrderedDict

from synnefo.webproject.management.commands import SynnefoCommand

from synnefo.webproject.management import utils

from ._common import show_quotas, style_options, check_style

import uuid

class Command(SynnefoCommand):

    args = "<user ID or email or uuid>"

    help = "Show user info"

    option_list = SynnefoCommand.option_list + (

        make_option('--quota',

                    action='store_true',

                    dest='list_quotas',

                    default=False,

                    help="Also list user quota"),

        make_option('--unit-style',

                    default='mb',

                    help=("Specify display unit for resource values "

                          "(one of %s); defaults to mb") % style_options),

        make_option('--projects',

                    action='store_true',

                    dest='list_projects',

                    default=False,

                    help="Also list project memberships"),

    )

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

        if len(args) != 1:

            raise CommandError("Please provide a user ID or email")

        identifier = args[0]

        if identifier.isdigit():

            users = AstakosUser.objects.filter(id=int(identifier))

        else:

            try:

                uuid.UUID(identifier)

            except:

                users = AstakosUser.objects.filter(email__iexact=identifier)

            else:

                users = AstakosUser.objects.filter(uuid=identifier)

        if users.count() == 0:

            field = 'id' if identifier.isdigit() else 'email'

            msg = "Unknown user with %s '%s'" % (field, identifier)

            raise CommandError(msg)

        for user in users:

            kv = OrderedDict(

                [

                    ('id', user.id),

                    ('uuid', user.uuid),

                    ('status', user.status_display),

                    ('email', user.email),

                    ('first name', user.first_name),

                    ('last name', user.last_name),

                    ('active', user.is_active),

                    ('admin', user.is_superuser),

                    ('last login', user.last_login),

                    ('date joined', user.date_joined),

                    ('last update', user.updated),

                    #('token', user.auth_token),

                    ('token expiration', user.auth_token_expires),

                    ('invitations', user.invitations),

                    ('invitation level', user.level),

                    ('providers', user.auth_providers_display),

                    ('verified', user.is_verified),

                    ('groups', [elem.name for elem in user.groups.all()]),

                    ('permissions', [elem.codename

                                     for elem in user.user_permissions.all()]),

                    ('group permissions', user.get_group_permissions()),

                    ('email verified', user.email_verified),

                    ('username', user.username),

                    ('activation_sent_date', user.activation_sent),

                ])

            if get_latest_terms():

                has_signed_terms = user.signed_terms

                kv['has_signed_terms'] = has_signed_terms

                if has_signed_terms:

                    kv['date_signed_terms'] = user.date_signed_terms

            utils.pprint_table(self.stdout, [kv.values()], kv.keys(),

                               options["output_format"], vertical=True)

            if options["list_quotas"]:

                unit_style = options["unit_style"]

                check_style(unit_style)

                quotas, initial = list_user_quotas([user])

                if quotas:

                    self.stdout.write("\n")

                    print_data, labels = show_quotas(quotas, initial,

                                                     style=unit_style)

                    utils.pprint_table(self.stdout, print_data, labels,

                                       options["output_format"],

                                       title="User Quota")

            if options["list_projects"]:

                print_data, labels = ownerships(user)

                if print_data:

                    self.stdout.write("\n")

                    utils.pprint_table(self.stdout, print_data, labels,

                                       options["output_format"],

                                       title="Owned Projects")

                print_data, labels = memberships(user)

                if print_data:

                    self.stdout.write("\n")

                    utils.pprint_table(self.stdout, print_data, labels,

                                       options["output_format"],

                                       title="Project Memberships")

def memberships(user):

    ms = user.projectmembership_set.all()

    print_data = []

    labels = ('project id', 'project name', 'status')

    for m in ms:

        project = m.project

        print_data.append((project.id,

                           project.application.name,

                           m.state_display(),

                           ))

    return print_data, labels

def ownerships(user):

    chain_dict = Chain.objects.all_full_state()

    chain_dict = filter_by(is_owner(user), chain_dict)

    return chain_info(chain_dict)

def is_owner(user):

    def f(state, project, app):

        return user == app.owner

    return f

def filter_by(f, chain_dict):

    d = {}

    for chain, tpl in chain_dict.iteritems():

        if f(*tpl):

            d[chain] = tpl

    return d

def chain_info(chain_dict):

    labels = ('project id', 'project name', 'status', 'pending app id')

    l = []

    for chain, (state, project, app) in chain_dict.iteritems():

        status = Chain.state_display(state)

        if state in Chain.PENDING_STATES:

            appid = str(app.id)

        else:

            appid = ""

        t = (chain,

             project.application.name if project else app.name,

             status,

             appid,

             )

        l.append(t)

    return l, labels