Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / quotas / util.py @ 06ffd11f

History | View | Annotate | Download (4.4 kB)

1
# Copyright 2012, 2013 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, Q
35

    
36
from synnefo.db.models import VirtualMachine, Network, FloatingIP
37
from synnefo.quotas import Quotaholder, ASTAKOS_TOKEN
38

    
39

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

    
44
    vms = VirtualMachine.objects.filter(deleted=False)
45
    networks = Network.objects.filter(deleted=False)
46
    floating_ips = FloatingIP.objects.filter(deleted=False)
47

    
48
    if user is not None:
49
        vms = vms.filter(userid=user)
50
        networks = networks.filter(userid=user)
51
        floating_ips = floating_ips.filter(userid=user)
52

    
53
    # Get resources related with VMs
54
    vm_resources = vms.values("userid").annotate(num=Count("id"),
55
                                                 ram=Sum("flavor__ram"),
56
                                                 cpu=Sum("flavor__cpu"),
57
                                                 disk=Sum("flavor__disk"))
58
    vm_active_resources = \
59
        vms.values("userid")\
60
           .filter(Q(operstate="STARTED") | Q(operstate="BUILD") |
61
                   Q(operstate="ERROR"))\
62
           .annotate(active_ram=Sum("flavor__ram"),
63
                     active_cpu=Sum("flavor__cpu"))
64

    
65
    for vm_res in vm_resources.iterator():
66
        user = vm_res['userid']
67
        res = {"cyclades.vm": vm_res["num"],
68
               "cyclades.cpu": vm_res["cpu"],
69
               "cyclades.disk": 1073741824 * vm_res["disk"],
70
               "cyclades.ram": 1048576 * vm_res["ram"]}
71
        holdings[user] = res
72

    
73
    for vm_res in vm_active_resources.iterator():
74
        user = vm_res['userid']
75
        holdings[user]["cyclades.active_cpu"] = vm_res["active_cpu"]
76
        holdings[user]["cyclades.active_ram"] = 1048576 * vm_res["active_ram"]
77

    
78
    # Get resources related with networks
79
    net_resources = networks.values("userid")\
80
                            .annotate(num=Count("id"))
81
    for net_res in net_resources.iterator():
82
        user = net_res['userid']
83
        holdings.setdefault(user, {})
84
        holdings[user]["cyclades.network.private"] = net_res["num"]
85

    
86
    floating_ips_resources = floating_ips.values("userid")\
87
                                         .annotate(num=Count("id"))
88
    for floating_ip_res in floating_ips_resources.iterator():
89
        user = floating_ip_res["userid"]
90
        holdings.setdefault(user, {})
91
        holdings[user]["cyclades.floating_ip"] = floating_ip_res["num"]
92

    
93
    return holdings
94

    
95

    
96
def get_quotaholder_holdings(user=None):
97
    """Get quotas from Quotaholder for all Cyclades resources.
98

99
    Returns quotas for all users, unless a single user is specified.
100
    """
101
    qh = Quotaholder.get()
102
    return qh.service_get_quotas(ASTAKOS_TOKEN, user)
103

    
104

    
105
def transform_quotas(quotas):
106
    d = {}
107
    for resource, counters in quotas.iteritems():
108
        used = counters['usage']
109
        limit = counters['limit']
110
        pending = counters['pending']
111
        d[resource] = (used, limit, pending)
112
    return d