invitation-list.py 2.98 KB
Newer Older
1
# Copyright 2012 GRNET S.A. All rights reserved.
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#
# 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

Sofia Papagiannaki's avatar
Sofia Papagiannaki committed
36
from django.core.management.base import NoArgsCommand, CommandError
37
38
39

from astakos.im.models import Invitation

40
41
from ._common import format_bool

42

Sofia Papagiannaki's avatar
Sofia Papagiannaki committed
43
class Command(NoArgsCommand):
44
    help = "List invitations"
45

Sofia Papagiannaki's avatar
Sofia Papagiannaki committed
46
    option_list = NoArgsCommand.option_list + (
47
        make_option('-c',
48
49
50
51
52
53
                    action='store_true',
                    dest='csv',
                    default=False,
                    help="Use pipes to separate values"),
    )

Sofia Papagiannaki's avatar
Sofia Papagiannaki committed
54
    def handle_noargs(self, **options):
55
        invitations = Invitation.objects.all().order_by('id')
56

57
        labels = ('id', 'inviter', 'email', 'real name', 'code', 'consumed')
Sofia Papagiannaki's avatar
Sofia Papagiannaki committed
58
        columns = (3, 24, 24, 24, 20, 4, 8)
59

60
61
62
63
64
        if not options['csv']:
            line = ' '.join(l.rjust(w) for l, w in zip(labels, columns))
            self.stdout.write(line + '\n')
            sep = '-' * len(line)
            self.stdout.write(sep + '\n')
65

66
67
68
        for invitation in invitations:
            id = str(invitation.id)
            code = str(invitation.code)
69
            consumed = format_bool(invitation.is_consumed)
70
71
72
            fields = (format(elem) for elem in \
                        (id, invitation.inviter.email, invitation.username,
                         invitation.realname, code, consumed))
73

74
75
76
77
            if options['csv']:
                line = '|'.join(fields)
            else:
                line = ' '.join(f.rjust(w) for f, w in zip(fields, columns))
78

79
            self.stdout.write(line + '\n')