Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / quotas / util.py @ 94ad2d07

History | View | Annotate | Download (4 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 django.db.models import Sum, Count
35

    
36
from synnefo.db.models import VirtualMachine, Network
37
from synnefo.quotas import get_quota_holder, NoEntityError
38

    
39
def get_db_holdings(users=None):
40
    """Get holdings from Cyclades DB."""
41
    holdings = {}
42

    
43
    vms = VirtualMachine.objects.filter(deleted=False)
44
    networks = Network.objects.filter(deleted=False)
45

    
46
    if users:
47
        assert(type(users) is list)
48
        vms = vms.filter(userid__in=users)
49
        networks = networks.filter(userid__in=users)
50

    
51
    # Get resources related with VMs
52
    vm_resources = vms.values("userid").annotate(num=Count("id"),
53
                                                 ram=Sum("flavor__ram"),
54
                                                 cpu=Sum("flavor__cpu"),
55
                                                 disk=Sum("flavor__disk"))
56
    for vm_res in vm_resources:
57
        user = vm_res['userid']
58
        res = {"vm": vm_res["num"],
59
               "cpu": vm_res["cpu"],
60
               "disk": 1073741824 * vm_res["disk"],
61
               "ram": 1048576 * vm_res["ram"]}
62
        holdings[user] = res
63

    
64
    # Get resources related with networks
65
    net_resources = networks.values("userid")\
66
                            .annotate(num=Count("id"))
67
    for net_res in net_resources:
68
        user = net_res['userid']
69
        if user not in holdings:
70
            holdings[user] = {}
71
        holdings[user]["network.private"] = net_res["num"]
72

    
73
    return holdings
74

    
75

    
76
def get_quotaholder_holdings(users=[]):
77
    """Get holdings from Quotaholder.
78

79
    If the entity for the user does not exist in quotaholder, no holding
80
    is returned.
81
    """
82
    users = filter(lambda u: not u is None, users)
83
    holdings = {}
84
    with get_quota_holder() as qh:
85
        list_holdings = [(user, "1") for user in users]
86
        (qh_holdings, rejected) = qh.list_holdings(context={},
87
                                                   list_holdings=list_holdings)
88
        found_users = filter(lambda u: not u in rejected, users)
89
    for user, user_holdings in zip(found_users, qh_holdings):
90
        if not user_holdings:
91
            continue
92
        for h in user_holdings:
93
            assert(h[0] == user)
94
        user_holdings = filter(lambda x: x[1].startswith("cyclades."),
95
                               user_holdings)
96
        holdings[user] = dict(map(decode_holding, user_holdings))
97
    return holdings
98

    
99

    
100
def decode_holding(holding):
101
    entity, resource, imported, exported, returned, released = holding
102
    res = resource.replace("cyclades.", "")
103
    return (res, imported - exported + returned - released)