Statistics
| Branch: | Tag: | Revision:

root / snf-pithos-app / pithos / api / management / commands / storagequota.py @ f4fbb0fa

History | View | Annotate | Download (3.6 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

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

    
38
from pithos.api.settings import (BACKEND_DB_MODULE, BACKEND_DB_CONNECTION,
39
                                 BACKEND_BLOCK_MODULE, BACKEND_BLOCK_PATH,
40
                                 BACKEND_BLOCK_UMASK,
41
                                 BACKEND_QUEUE_MODULE, BACKEND_QUEUE_HOSTS,
42
                                 BACKEND_QUEUE_EXCHANGE,
43
                                 BACKEND_QUOTA, BACKEND_VERSIONING)
44
from pithos.backends import connect_backend
45

    
46

    
47
class Command(BaseCommand):
48
    args = "<user>"
49
    help = "Get/set a user's quota"
50

    
51
    option_list = BaseCommand.option_list + (
52
        make_option('--set-quota',
53
                    dest='quota',
54
                    metavar='BYTES',
55
                    help="Set user's quota"),
56
    )
57

    
58
    def handle(self, *args, **options):
59
        if len(args) != 1:
60
            raise CommandError("Please provide a user")
61

    
62
        user = args[0]
63
        quota = options.get('quota')
64
        if quota is not None:
65
            try:
66
                quota = int(quota)
67
            except ValueError:
68
                raise CommandError("Invalid quota")
69

    
70
        backend = connect_backend(db_module=BACKEND_DB_MODULE,
71
                                  db_connection=BACKEND_DB_CONNECTION,
72
                                  block_module=BACKEND_BLOCK_MODULE,
73
                                  block_path=BACKEND_BLOCK_PATH,
74
                                  block_umask=BACKEND_BLOCK_UMASK,
75
                                  queue_module=BACKEND_QUEUE_MODULE,
76
                                  queue_connection=BACKEND_QUEUE_HOSTS,
77
                                  queue_exchange=BACKEND_QUEUE_EXCHANGE)
78
        backend.default_policy['quota'] = BACKEND_QUOTA
79
        backend.default_policy['versioning'] = BACKEND_VERSIONING
80
        if quota is not None:
81
            backend.update_account_policy(user, user, {'quota': quota})
82
        else:
83
            self.stdout.write("Quota for %s: %s\n" % (
84
                user, backend.get_account_policy(user, user)['quota']))
85
        backend.close()