Statistics
| Branch: | Tag: | Revision:

root / logic / credits.py @ 583bfaa0

History | View | Annotate | Download (5.5 kB)

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

    
30
# Business Logic for all Credit related activity
31

    
32
from datetime import datetime
33
from django.db import transaction
34

    
35
from synnefo.db.models import Debit, FlavorCost
36

    
37
@transaction.commit_on_success
38
def debit_account(user , amount, vm, description):
39
    """Charges the user with the specified amount of credits for a vm (resource)"""
40
    date_now = datetime.now()
41
    user.credit = user.credit - amount
42
    user.save()
43

    
44
    # then write the debit entry
45
    debit = Debit()
46
    debit.user = user
47
    debit.vm = vm
48
    debit.when = date_now
49
    debit.description = description
50
    debit.save()
51

    
52

    
53
@transaction.commit_on_success
54
def credit_account(self, amount, creditor, description):
55
    """No clue :)"""
56
    return
57

    
58

    
59
@transaction.commit_on_success
60
def charge(vm):
61
    """Charges the owner of this VM.
62

63
    Charges the owner of a VM for the period
64
    from vm.charged to datetime.now(), based on the
65
    current operating state.
66

67
    """
68
    charged_states = ('STARTED', 'STOPPED')
69

    
70
    start_datetime = vm.charged
71
    vm.charged = datetime.now()
72

    
73
    # Only charge for a specific set of states
74
    if vm.operstate in charged_states:
75
        cost_list = []
76

    
77
        # remember, we charge only for Started and Stopped
78
        if vm.operstate == 'STARTED':
79
            cost_list = get_cost_active(vm.flavor, start_datetime, vm.charged)
80
        elif vm.operstate == 'STOPPED':
81
            cost_list = get_cost_inactive(vm.flavor, start_datetime, vm.charged)
82

    
83
        # find the total vost
84
        total_cost = sum([x[1] for x in cost_list])
85

    
86
        # add the debit entry
87
        description = "Server = %s, charge = %d for state: %s" % (vm.name, total_cost, vm.operstate)
88
        debit_account(vm.owner, total_cost, vm, description)
89

    
90
    vm.save()
91

    
92

    
93
def get_costs(vm, start_datetime, end_datetime, active):
94
    """Return a list with FlavorCost objects for the specified duration"""
95
    def between(enh_fc, a_date):
96
        """Checks if a date is between a FlavorCost duration"""
97
        if enh_fc.effective_from <= a_date and enh_fc.effective_to is None:
98
            return True
99

    
100
        return enh_fc.effective_from <= a_date and enh_fc.effective_to >= a_date
101

    
102
    # Get the related FlavorCost objects, sorted.
103
    price_list = FlavorCost.objects.filter(flavor=vm).order_by('effective_from')
104

    
105
    # add the extra field FlavorCost.effective_to
106
    for idx in range(0, len(price_list)):
107
        if idx + 1 == len(price_list):
108
            price_list[idx].effective_to = None
109
        else:
110
            price_list[idx].effective_to = price_list[idx + 1].effective_from
111

    
112
    price_result = []
113
    found_start = False
114

    
115
    # Find the affected FlavorCost, according to the
116
    # dates, and put them in price_result
117
    for p in price_list:
118
        if between(p, start_datetime):
119
            found_start = True
120
            p.effective_from = start_datetime
121
        if between(p, end_datetime):
122
            p.effective_to = end_datetime
123
            price_result.append(p)
124
            break
125
        if found_start:
126
            price_result.append(p)
127

    
128
    results = []
129

    
130
    # Create the list and the result tuples
131
    for p in price_result:
132
        if active:
133
            cost = p.cost_active
134
        else:
135
            cost = p.cost_inactive
136

    
137
        results.append( ( p.effective_from, calculate_cost(p.effective_from, p.effective_to, cost)) )
138

    
139
    return results
140

    
141

    
142
def calculate_cost(start_date, end_date, cost):
143
    """Calculate the total cost for the specified duration"""
144
    td = end_date - start_date
145
    sec = float(td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / float(10**6)
146
    total_hours = float(sec) / float(60.0*60.0)
147
    total_cost = float(cost)*total_hours
148

    
149
    return round(total_cost)
150

    
151

    
152
def get_cost_active(vm, start_datetime, end_datetime):
153
    """Returns a list with the active costs for the specified duration"""
154
    return get_costs(vm, start_datetime, end_datetime, True)
155

    
156

    
157
def get_cost_inactive(vm, start_datetime, end_datetime):
158
    """Returns a list with the inactive costs for the specified duration"""
159
    return get_costs(vm, start_datetime, end_datetime, False)