root / snf-astakos-app / astakos / api / quotas.py @ bbf79074
History | View | Annotate | Download (8.2 kB)
1 | 18f21257 | Giorgos Korfiatis | # Copyright 2013 GRNET S.A. All rights reserved.
|
---|---|---|---|
2 | 18f21257 | Giorgos Korfiatis | #
|
3 | 18f21257 | Giorgos Korfiatis | # Redistribution and use in source and binary forms, with or
|
4 | 18f21257 | Giorgos Korfiatis | # without modification, are permitted provided that the following
|
5 | 18f21257 | Giorgos Korfiatis | # conditions are met:
|
6 | 18f21257 | Giorgos Korfiatis | #
|
7 | 18f21257 | Giorgos Korfiatis | # 1. Redistributions of source code must retain the above
|
8 | 18f21257 | Giorgos Korfiatis | # copyright notice, this list of conditions and the following
|
9 | 18f21257 | Giorgos Korfiatis | # disclaimer.
|
10 | 18f21257 | Giorgos Korfiatis | #
|
11 | 18f21257 | Giorgos Korfiatis | # 2. Redistributions in binary form must reproduce the above
|
12 | 18f21257 | Giorgos Korfiatis | # copyright notice, this list of conditions and the following
|
13 | 18f21257 | Giorgos Korfiatis | # disclaimer in the documentation and/or other materials
|
14 | 18f21257 | Giorgos Korfiatis | # provided with the distribution.
|
15 | 18f21257 | Giorgos Korfiatis | #
|
16 | 18f21257 | Giorgos Korfiatis | # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
|
17 | 18f21257 | Giorgos Korfiatis | # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
18 | 18f21257 | Giorgos Korfiatis | # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
19 | 18f21257 | Giorgos Korfiatis | # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
|
20 | 18f21257 | Giorgos Korfiatis | # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
21 | 18f21257 | Giorgos Korfiatis | # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
22 | 18f21257 | Giorgos Korfiatis | # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
23 | 18f21257 | Giorgos Korfiatis | # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
24 | 18f21257 | Giorgos Korfiatis | # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
25 | 18f21257 | Giorgos Korfiatis | # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
26 | 18f21257 | Giorgos Korfiatis | # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
27 | 18f21257 | Giorgos Korfiatis | # POSSIBILITY OF SUCH DAMAGE.
|
28 | 18f21257 | Giorgos Korfiatis | #
|
29 | 18f21257 | Giorgos Korfiatis | # The views and conclusions contained in the software and
|
30 | 18f21257 | Giorgos Korfiatis | # documentation are those of the authors and should not be
|
31 | 18f21257 | Giorgos Korfiatis | # interpreted as representing official policies, either expressed
|
32 | 18f21257 | Giorgos Korfiatis | # or implied, of GRNET S.A.
|
33 | 18f21257 | Giorgos Korfiatis | |
34 | 18f21257 | Giorgos Korfiatis | from django.utils import simplejson as json |
35 | 18f21257 | Giorgos Korfiatis | from django.views.decorators.csrf import csrf_exempt |
36 | 18f21257 | Giorgos Korfiatis | from django.http import HttpResponse |
37 | 18f21257 | Giorgos Korfiatis | |
38 | b830f774 | Giorgos Korfiatis | from snf_django.lib.db.transaction import commit_on_success_strict |
39 | 18f21257 | Giorgos Korfiatis | from astakos.api.util import json_response |
40 | befa5997 | Giorgos Korfiatis | |
41 | befa5997 | Giorgos Korfiatis | from snf_django.lib import api |
42 | befa5997 | Giorgos Korfiatis | from snf_django.lib.api.faults import BadRequest, InternalServerError |
43 | befa5997 | Giorgos Korfiatis | |
44 | 1028e568 | Giorgos Korfiatis | from astakos.im.api import api_method as generic_api_method |
45 | befa5997 | Giorgos Korfiatis | from astakos.im.api.user import user_from_token |
46 | befa5997 | Giorgos Korfiatis | from astakos.im.api.service import service_from_token |
47 | befa5997 | Giorgos Korfiatis | |
48 | 08043775 | Giorgos Korfiatis | from astakos.im.resources import get_resources |
49 | 08043775 | Giorgos Korfiatis | from astakos.im.quotas import get_user_quotas, get_service_quotas |
50 | 18f21257 | Giorgos Korfiatis | |
51 | 18f21257 | Giorgos Korfiatis | import astakos.quotaholder.exception as qh_exception |
52 | 948e15bc | Giorgos Korfiatis | import astakos.quotaholder.callpoint as qh |
53 | 18f21257 | Giorgos Korfiatis | |
54 | 18f21257 | Giorgos Korfiatis | |
55 | befa5997 | Giorgos Korfiatis | @api.api_method(http_method='GET', token_required=True, user_required=False) |
56 | befa5997 | Giorgos Korfiatis | @user_from_token
|
57 | 18f21257 | Giorgos Korfiatis | def quotas(request, user=None): |
58 | 18f21257 | Giorgos Korfiatis | result = get_user_quotas(user) |
59 | 18f21257 | Giorgos Korfiatis | return json_response(result)
|
60 | 18f21257 | Giorgos Korfiatis | |
61 | 18f21257 | Giorgos Korfiatis | |
62 | bb005bc2 | Giorgos Korfiatis | @api.api_method(http_method='GET', token_required=True, user_required=False) |
63 | bb005bc2 | Giorgos Korfiatis | @service_from_token
|
64 | bb005bc2 | Giorgos Korfiatis | def service_quotas(request): |
65 | 5afce44d | Giorgos Korfiatis | user = request.GET.get('user')
|
66 | 5afce44d | Giorgos Korfiatis | users = [user] if user is not None else None |
67 | 5afce44d | Giorgos Korfiatis | result = get_service_quotas(request.service_instance, users=users) |
68 | bb005bc2 | Giorgos Korfiatis | return json_response(result)
|
69 | bb005bc2 | Giorgos Korfiatis | |
70 | bb005bc2 | Giorgos Korfiatis | |
71 | befa5997 | Giorgos Korfiatis | @api.api_method(http_method='GET', token_required=False, user_required=False) |
72 | 1028e568 | Giorgos Korfiatis | def resources(request): |
73 | 1028e568 | Giorgos Korfiatis | result = get_resources() |
74 | 1028e568 | Giorgos Korfiatis | return json_response(result)
|
75 | 1028e568 | Giorgos Korfiatis | |
76 | 1028e568 | Giorgos Korfiatis | |
77 | 18f21257 | Giorgos Korfiatis | @csrf_exempt
|
78 | 18f21257 | Giorgos Korfiatis | def commissions(request): |
79 | 18f21257 | Giorgos Korfiatis | method = request.method |
80 | 18f21257 | Giorgos Korfiatis | if method == 'GET': |
81 | 18f21257 | Giorgos Korfiatis | return get_pending_commissions(request)
|
82 | 18f21257 | Giorgos Korfiatis | elif method == 'POST': |
83 | 18f21257 | Giorgos Korfiatis | return issue_commission(request)
|
84 | 18f21257 | Giorgos Korfiatis | else:
|
85 | 18f21257 | Giorgos Korfiatis | raise BadRequest('Method not allowed.') |
86 | 18f21257 | Giorgos Korfiatis | |
87 | 18f21257 | Giorgos Korfiatis | |
88 | befa5997 | Giorgos Korfiatis | @api.api_method(http_method='GET', token_required=True, user_required=False) |
89 | befa5997 | Giorgos Korfiatis | @service_from_token
|
90 | 18f21257 | Giorgos Korfiatis | def get_pending_commissions(request): |
91 | 18f21257 | Giorgos Korfiatis | data = request.GET |
92 | a4398c8c | Giorgos Korfiatis | client_key = str(request.service_instance)
|
93 | 18f21257 | Giorgos Korfiatis | |
94 | 18f21257 | Giorgos Korfiatis | result = qh.get_pending_commissions(clientkey=client_key) |
95 | 18f21257 | Giorgos Korfiatis | return json_response(result)
|
96 | 18f21257 | Giorgos Korfiatis | |
97 | 18f21257 | Giorgos Korfiatis | |
98 | bbf79074 | Giorgos Korfiatis | def _provisions_to_list(provisions): |
99 | bbf79074 | Giorgos Korfiatis | lst = [] |
100 | bbf79074 | Giorgos Korfiatis | for provision in provisions: |
101 | bbf79074 | Giorgos Korfiatis | try:
|
102 | bbf79074 | Giorgos Korfiatis | holder = provision['holder']
|
103 | bbf79074 | Giorgos Korfiatis | source = provision['source']
|
104 | bbf79074 | Giorgos Korfiatis | resource = provision['resource']
|
105 | bbf79074 | Giorgos Korfiatis | quantity = provision['quantity']
|
106 | bbf79074 | Giorgos Korfiatis | key = (holder, source, resource) |
107 | bbf79074 | Giorgos Korfiatis | lst.append((key, quantity)) |
108 | bbf79074 | Giorgos Korfiatis | except KeyError: |
109 | bbf79074 | Giorgos Korfiatis | raise ValueError("Malformed provision") |
110 | bbf79074 | Giorgos Korfiatis | return lst
|
111 | bbf79074 | Giorgos Korfiatis | |
112 | bbf79074 | Giorgos Korfiatis | |
113 | 18f21257 | Giorgos Korfiatis | @csrf_exempt
|
114 | befa5997 | Giorgos Korfiatis | @api.api_method(http_method='POST', token_required=True, user_required=False) |
115 | befa5997 | Giorgos Korfiatis | @service_from_token
|
116 | 18f21257 | Giorgos Korfiatis | def issue_commission(request): |
117 | 18f21257 | Giorgos Korfiatis | data = request.raw_post_data |
118 | 18f21257 | Giorgos Korfiatis | input_data = json.loads(data) |
119 | 18f21257 | Giorgos Korfiatis | |
120 | a4398c8c | Giorgos Korfiatis | client_key = str(request.service_instance)
|
121 | 18f21257 | Giorgos Korfiatis | provisions = input_data['provisions']
|
122 | bbf79074 | Giorgos Korfiatis | provisions = _provisions_to_list(provisions) |
123 | 18f21257 | Giorgos Korfiatis | force = input_data.get('force', False) |
124 | 18f21257 | Giorgos Korfiatis | auto_accept = input_data.get('auto_accept', False) |
125 | 18f21257 | Giorgos Korfiatis | |
126 | 18f21257 | Giorgos Korfiatis | try:
|
127 | 18f21257 | Giorgos Korfiatis | result = _issue_commission(clientkey=client_key, |
128 | 18f21257 | Giorgos Korfiatis | provisions=provisions, |
129 | 18f21257 | Giorgos Korfiatis | force=force, |
130 | 18f21257 | Giorgos Korfiatis | accept=auto_accept) |
131 | 18f21257 | Giorgos Korfiatis | data = {"serial": result}
|
132 | 18f21257 | Giorgos Korfiatis | status_code = 201
|
133 | 18f21257 | Giorgos Korfiatis | except (qh_exception.NoCapacityError,
|
134 | 18f21257 | Giorgos Korfiatis | qh_exception.NoQuantityError) as e:
|
135 | 18f21257 | Giorgos Korfiatis | status_code = 413
|
136 | 18f21257 | Giorgos Korfiatis | body = {"message": e.message,
|
137 | 18f21257 | Giorgos Korfiatis | "code": status_code,
|
138 | 18f21257 | Giorgos Korfiatis | "data": e.data,
|
139 | 18f21257 | Giorgos Korfiatis | } |
140 | 18f21257 | Giorgos Korfiatis | data = {"overLimit": body}
|
141 | 18f21257 | Giorgos Korfiatis | except qh_exception.NoHoldingError as e: |
142 | 18f21257 | Giorgos Korfiatis | status_code = 404
|
143 | 18f21257 | Giorgos Korfiatis | body = {"message": e.message,
|
144 | 18f21257 | Giorgos Korfiatis | "code": status_code,
|
145 | 18f21257 | Giorgos Korfiatis | "data": e.data,
|
146 | 18f21257 | Giorgos Korfiatis | } |
147 | 18f21257 | Giorgos Korfiatis | data = {"itemNotFound": body}
|
148 | 18f21257 | Giorgos Korfiatis | except qh_exception.InvalidDataError as e: |
149 | 18f21257 | Giorgos Korfiatis | status_code = 400
|
150 | 18f21257 | Giorgos Korfiatis | body = {"message": e.message,
|
151 | 18f21257 | Giorgos Korfiatis | "code": status_code,
|
152 | 18f21257 | Giorgos Korfiatis | } |
153 | 18f21257 | Giorgos Korfiatis | data = {"badRequest": body}
|
154 | 18f21257 | Giorgos Korfiatis | |
155 | 18f21257 | Giorgos Korfiatis | return json_response(data, status_code=status_code)
|
156 | 18f21257 | Giorgos Korfiatis | |
157 | 18f21257 | Giorgos Korfiatis | |
158 | 18f21257 | Giorgos Korfiatis | @commit_on_success_strict()
|
159 | 18f21257 | Giorgos Korfiatis | def _issue_commission(clientkey, provisions, force, accept): |
160 | 18f21257 | Giorgos Korfiatis | serial = qh.issue_commission(clientkey=clientkey, |
161 | 18f21257 | Giorgos Korfiatis | provisions=provisions, |
162 | 18f21257 | Giorgos Korfiatis | force=force) |
163 | 18f21257 | Giorgos Korfiatis | if accept:
|
164 | 18f21257 | Giorgos Korfiatis | done = qh.accept_commission(clientkey=clientkey, |
165 | 18f21257 | Giorgos Korfiatis | serial=serial) |
166 | 18f21257 | Giorgos Korfiatis | |
167 | 18f21257 | Giorgos Korfiatis | return serial
|
168 | 18f21257 | Giorgos Korfiatis | |
169 | 18f21257 | Giorgos Korfiatis | |
170 | 6c0f4562 | Giorgos Korfiatis | def notFoundCF(serial): |
171 | 6c0f4562 | Giorgos Korfiatis | body = {"code": 404, |
172 | 6c0f4562 | Giorgos Korfiatis | "message": "serial %s does not exist" % serial, |
173 | 6c0f4562 | Giorgos Korfiatis | } |
174 | 6c0f4562 | Giorgos Korfiatis | return {"itemNotFound": body} |
175 | 6c0f4562 | Giorgos Korfiatis | |
176 | 6c0f4562 | Giorgos Korfiatis | |
177 | 6c0f4562 | Giorgos Korfiatis | def conflictingCF(serial): |
178 | 6c0f4562 | Giorgos Korfiatis | body = {"code": 400, |
179 | 6c0f4562 | Giorgos Korfiatis | "message": "cannot both accept and reject serial %s" % serial, |
180 | 6c0f4562 | Giorgos Korfiatis | } |
181 | 6c0f4562 | Giorgos Korfiatis | return {"badRequest": body} |
182 | 18f21257 | Giorgos Korfiatis | |
183 | 18f21257 | Giorgos Korfiatis | |
184 | 18f21257 | Giorgos Korfiatis | @csrf_exempt
|
185 | befa5997 | Giorgos Korfiatis | @api.api_method(http_method='POST', token_required=True, user_required=False) |
186 | befa5997 | Giorgos Korfiatis | @service_from_token
|
187 | 18f21257 | Giorgos Korfiatis | @commit_on_success_strict()
|
188 | 18f21257 | Giorgos Korfiatis | def resolve_pending_commissions(request): |
189 | 18f21257 | Giorgos Korfiatis | data = request.raw_post_data |
190 | 18f21257 | Giorgos Korfiatis | input_data = json.loads(data) |
191 | 18f21257 | Giorgos Korfiatis | |
192 | a4398c8c | Giorgos Korfiatis | client_key = str(request.service_instance)
|
193 | 18f21257 | Giorgos Korfiatis | accept = input_data.get('accept', [])
|
194 | 18f21257 | Giorgos Korfiatis | reject = input_data.get('reject', [])
|
195 | 18f21257 | Giorgos Korfiatis | |
196 | 18f21257 | Giorgos Korfiatis | result = qh.resolve_pending_commissions(clientkey=client_key, |
197 | 18f21257 | Giorgos Korfiatis | accept_set=accept, |
198 | 18f21257 | Giorgos Korfiatis | reject_set=reject) |
199 | 6c0f4562 | Giorgos Korfiatis | accepted, rejected, notFound, conflicting = result |
200 | 6c0f4562 | Giorgos Korfiatis | notFound = [(serial, notFoundCF(serial)) for serial in notFound] |
201 | 6c0f4562 | Giorgos Korfiatis | conflicting = [(serial, conflictingCF(serial)) for serial in conflicting] |
202 | 6c0f4562 | Giorgos Korfiatis | cloudfaults = notFound + conflicting |
203 | 18f21257 | Giorgos Korfiatis | data = {'accepted': accepted,
|
204 | 18f21257 | Giorgos Korfiatis | 'rejected': rejected,
|
205 | 18f21257 | Giorgos Korfiatis | 'failed': cloudfaults
|
206 | 18f21257 | Giorgos Korfiatis | } |
207 | 18f21257 | Giorgos Korfiatis | |
208 | 18f21257 | Giorgos Korfiatis | return json_response(data)
|
209 | 18f21257 | Giorgos Korfiatis | |
210 | 18f21257 | Giorgos Korfiatis | |
211 | befa5997 | Giorgos Korfiatis | @api.api_method(http_method='GET', token_required=True, user_required=False) |
212 | befa5997 | Giorgos Korfiatis | @service_from_token
|
213 | 18f21257 | Giorgos Korfiatis | def get_commission(request, serial): |
214 | 18f21257 | Giorgos Korfiatis | data = request.GET |
215 | a4398c8c | Giorgos Korfiatis | client_key = str(request.service_instance)
|
216 | 18f21257 | Giorgos Korfiatis | serial = int(serial)
|
217 | 18f21257 | Giorgos Korfiatis | |
218 | 18f21257 | Giorgos Korfiatis | try:
|
219 | 18f21257 | Giorgos Korfiatis | data = qh.get_commission(clientkey=client_key, |
220 | 18f21257 | Giorgos Korfiatis | serial=serial) |
221 | 18f21257 | Giorgos Korfiatis | status_code = 200
|
222 | 18f21257 | Giorgos Korfiatis | return json_response(data, status_code)
|
223 | 18f21257 | Giorgos Korfiatis | except qh_exception.NoCommissionError as e: |
224 | 18f21257 | Giorgos Korfiatis | return HttpResponse(status=404) |
225 | 18f21257 | Giorgos Korfiatis | |
226 | 18f21257 | Giorgos Korfiatis | |
227 | 18f21257 | Giorgos Korfiatis | @csrf_exempt
|
228 | befa5997 | Giorgos Korfiatis | @api.api_method(http_method='POST', token_required=True, user_required=False) |
229 | befa5997 | Giorgos Korfiatis | @service_from_token
|
230 | 18f21257 | Giorgos Korfiatis | @commit_on_success_strict()
|
231 | 18f21257 | Giorgos Korfiatis | def serial_action(request, serial): |
232 | 18f21257 | Giorgos Korfiatis | data = request.raw_post_data |
233 | 18f21257 | Giorgos Korfiatis | input_data = json.loads(data) |
234 | 18f21257 | Giorgos Korfiatis | serial = int(serial)
|
235 | 18f21257 | Giorgos Korfiatis | |
236 | a4398c8c | Giorgos Korfiatis | client_key = str(request.service_instance)
|
237 | 18f21257 | Giorgos Korfiatis | |
238 | 18f21257 | Giorgos Korfiatis | accept = 'accept' in input_data |
239 | 18f21257 | Giorgos Korfiatis | reject = 'reject' in input_data |
240 | 18f21257 | Giorgos Korfiatis | |
241 | 18f21257 | Giorgos Korfiatis | if accept == reject:
|
242 | 18f21257 | Giorgos Korfiatis | raise BadRequest('Specify either accept or reject action.') |
243 | 18f21257 | Giorgos Korfiatis | |
244 | 6c0f4562 | Giorgos Korfiatis | result = qh.resolve_pending_commission(clientkey=client_key, |
245 | 6c0f4562 | Giorgos Korfiatis | serial=serial, |
246 | 6c0f4562 | Giorgos Korfiatis | accept=accept) |
247 | 18f21257 | Giorgos Korfiatis | response = HttpResponse() |
248 | 18f21257 | Giorgos Korfiatis | if not result: |
249 | 18f21257 | Giorgos Korfiatis | response.status_code = 404
|
250 | 18f21257 | Giorgos Korfiatis | |
251 | 18f21257 | Giorgos Korfiatis | return response |