Synnefo

# Copyright 2012-2014 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 django.db.models import Q

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

from astakos.im.quotas import get_user_quotas

from synnefo.lib.ordereddict import OrderedDict

from snf_django.management.commands import SynnefoCommand

from snf_django.management import utils

from ._common import show_user_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),

                    ('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"] and user.is_accepted():

                unit_style = options["unit_style"]

                check_style(unit_style)

                quotas = get_user_quotas(user)

                if quotas:

                    self.stdout.write("\n")

                    print_data, labels = show_user_quotas(quotas,

                                                          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.uuid,

                           project.realname,

                           m.state_display(),

                           ))

    return print_data, labels

def ownerships(user):

    chains = Project.objects.select_related("last_application").\

        filter(owner=user)

    return chain_info(chains)

def chain_info(chains):

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

    l = []

    for project in chains:

        status = project.state_display()

        app = project.last_application

        pending_appid = app.id if app and app.state == app.PENDING else ""

        t = (project.uuid,

             project.realname,

             status,

             pending_appid,

             )

        l.append(t)

    return l, labels