Statistics
| Branch: | Tag: | Revision:

root / snf-astakos-app / astakos / im / management / commands / astakos-quota-verify.py @ f2c27eb6

History | View | Annotate | Download (3.8 kB)

1
# Copyright 2012 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or
4
# without modification, are permitted provided that the following
5
# conditions are met:
6
#
7
#   1. Redistributions of source code must retain the above
8
#      copyright notice, this list of conditions and the following
9
#      disclaimer.
10
#
11
#   2. Redistributions in binary form must reproduce the above
12
#      copyright notice, this list of conditions and the following
13
#      disclaimer in the documentation and/or other materials
14
#      provided with the distribution.
15
#
16
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27
# POSSIBILITY OF SUCH DAMAGE.
28
#
29
# The views and conclusions contained in the software and
30
# documentation are those of the authors and should not be
31
# interpreted as representing official policies, either expressed
32
# or implied, of GRNET S.A.
33

    
34
from optparse import make_option
35
from django.core.management.base import BaseCommand, CommandError
36
from django.db import transaction
37

    
38
from astakos.im.models import sync_all_users, sync_projects
39
from astakos.im.functions import get_user_by_uuid
40

    
41
import logging
42
logger = logging.getLogger(__name__)
43

    
44
class Command(BaseCommand):
45
    help = "Inspect quotaholder status"
46

    
47
    option_list = BaseCommand.option_list + (
48
        make_option('--check',
49
                    action='store_true',
50
                    dest='check',
51
                    default=True,
52
                    help="Check if quotaholder is in sync with astakos (default)"),
53
        make_option('--sync',
54
                    action='store_true',
55
                    dest='sync',
56
                    default=False,
57
                    help="Sync quotaholder"),
58
    )
59

    
60
    @transaction.commit_on_success
61
    def handle(self, *args, **options):
62
        sync = options['sync']
63

    
64
        try:
65
            log = sync_all_users(sync=sync)
66
            existing, nonexisting, registered_quotas, astakos_quotas = log
67

    
68
            if nonexisting:
69
                self.stdout.write("Users not registered in quotaholder:\n")
70
                for user in nonexisting:
71
                    self.stdout.write("%s\n" % (user))
72
                self.stdout.write("\n")
73

    
74
            diffs = 0
75
            for holder, local in astakos_quotas.iteritems():
76
                registered = registered_quotas.pop(holder, None)
77
                if registered is None:
78
                    diffs += 1
79
                    self.stdout.write("No quotas for %s in quotaholder.\n\n" %
80
                                      (get_user_by_uuid(holder)))
81
                elif local != registered:
82
                    diffs += 1
83
                    self.stdout.write("Quotas differ for %s:\n" % (get_user_by_uuid(holder)))
84
                    self.stdout.write("Quotas according to quotaholder:\n")
85
                    self.stdout.write("%s\n" % (registered))
86
                    self.stdout.write("Quotas according to astakos:\n")
87
                    self.stdout.write("%s\n\n" % (local))
88

    
89
            if diffs:
90
                self.stdout.write("Quotas differ for %d users.\n" % (diffs))
91
        except BaseException, e:
92
            logger.exception(e)
93
            raise CommandError("Syncing failed.")