|
1
|
|
|
# coding: utf-8 |
|
2
|
|
|
from __future__ import unicode_literals |
|
3
|
|
|
|
|
4
|
|
|
import arrow |
|
5
|
|
|
# django |
|
6
|
|
|
from django.core import management |
|
7
|
|
|
from django.core.cache import caches |
|
8
|
|
|
from django.core.urlresolvers import reverse_lazy, reverse |
|
9
|
|
|
from django.conf import settings |
|
10
|
|
|
from django.shortcuts import render, redirect, get_object_or_404 |
|
11
|
|
|
from django.contrib.auth.decorators import login_required |
|
12
|
|
|
from django.contrib.auth import logout |
|
13
|
|
|
from django.http import HttpResponseRedirect |
|
14
|
|
|
from django.views.generic import TemplateView, UpdateView, ListView, DeleteView |
|
15
|
|
|
from django.db.models import Q |
|
16
|
|
|
from django.utils.decorators import method_decorator |
|
17
|
|
|
from django.utils.translation import ugettext as _ |
|
18
|
|
|
|
|
19
|
|
|
|
|
20
|
|
|
# trigger_happy |
|
21
|
|
|
from django_th.models import TriggerService, UserService, ServicesActivated |
|
22
|
|
|
from django_th.tools import get_service |
|
23
|
|
|
from django_th.forms.base import TriggerServiceForm |
|
24
|
|
|
|
|
25
|
|
|
import logging |
|
26
|
|
|
# Get an instance of a logger |
|
27
|
|
|
logger = logging.getLogger(__name__) |
|
28
|
|
|
cache = caches['django_th'] |
|
29
|
|
|
|
|
30
|
|
|
""" |
|
31
|
|
|
Part I : Trigger Part |
|
32
|
|
|
""" |
|
33
|
|
|
|
|
34
|
|
|
# ************************ |
|
35
|
|
|
# FBV : simple actions * |
|
36
|
|
|
# ************************ |
|
37
|
|
|
|
|
38
|
|
|
|
|
39
|
|
|
def can_modify_trigger(request, provider, consumer): |
|
40
|
|
|
# do not permit to edit the details of one trigger |
|
41
|
|
|
# if the provider or consumer is disabled |
|
42
|
|
|
if provider and consumer: |
|
43
|
|
|
return False |
|
44
|
|
|
else: |
|
45
|
|
|
from django.contrib import messages |
|
46
|
|
|
messages.warning(request, 'You cant modify a disabled trigger') |
|
47
|
|
|
return True |
|
48
|
|
|
|
|
49
|
|
|
|
|
50
|
|
|
def logout_view(request): |
|
51
|
|
|
""" |
|
52
|
|
|
logout the user then redirect him to the home page |
|
53
|
|
|
""" |
|
54
|
|
|
logout(request) |
|
55
|
|
|
return HttpResponseRedirect(reverse('base')) |
|
56
|
|
|
|
|
57
|
|
|
|
|
58
|
|
|
def trigger_on_off(request, trigger_id): |
|
59
|
|
|
""" |
|
60
|
|
|
enable/disable the status of the trigger then go back home |
|
61
|
|
|
:param request: request object |
|
62
|
|
|
:param trigger_id: the trigger ID to switch the status to True or False |
|
63
|
|
|
:type request: HttpRequest object |
|
64
|
|
|
:type trigger_id: int |
|
65
|
|
|
:return render |
|
66
|
|
|
:rtype HttpResponse |
|
67
|
|
|
""" |
|
68
|
|
|
now = arrow.utcnow().to(settings.TIME_ZONE).format( |
|
69
|
|
|
'YYYY-MM-DD HH:mm:ssZZ') |
|
70
|
|
|
trigger = get_object_or_404(TriggerService, pk=trigger_id) |
|
71
|
|
|
if trigger.status: |
|
72
|
|
|
title = 'disabled' |
|
73
|
|
|
title_trigger = _('Set this trigger on') |
|
74
|
|
|
btn = 'success' |
|
75
|
|
|
trigger.status = False |
|
76
|
|
|
else: |
|
77
|
|
|
title = _('Edit your service') |
|
78
|
|
|
title_trigger = _('Set this trigger off') |
|
79
|
|
|
btn = 'primary' |
|
80
|
|
|
trigger.status = True |
|
81
|
|
|
# set the trigger to the current date when the |
|
82
|
|
|
# the trigger is back online |
|
83
|
|
|
trigger.date_triggered = now |
|
84
|
|
|
trigger.save() |
|
85
|
|
|
|
|
86
|
|
|
return render(request, 'triggers/trigger_line.html', |
|
87
|
|
|
{'trigger': trigger, |
|
88
|
|
|
'title': title, |
|
89
|
|
|
'title_trigger': title_trigger, |
|
90
|
|
|
'btn': btn}) |
|
91
|
|
|
|
|
92
|
|
|
|
|
93
|
|
|
def fire_trigger(request, trigger_id): |
|
94
|
|
|
""" |
|
95
|
|
|
start the handling of only ONE trigger |
|
96
|
|
|
:param request: request object |
|
97
|
|
|
:param trigger_id: the trigger ID to switch the status to True or False |
|
98
|
|
|
:type request: HttpRequest object |
|
99
|
|
|
:type trigger_id: int |
|
100
|
|
|
:return render |
|
101
|
|
|
:rtype HttpResponse |
|
102
|
|
|
""" |
|
103
|
|
|
date = '' |
|
104
|
|
|
|
|
105
|
|
|
if cache.get('django_th' + '_fire_trigger_' + str(trigger_id)): |
|
106
|
|
|
template = 'triggers/fire_trigger_ko.html' |
|
107
|
|
|
trigger = TriggerService.objects.get(id=trigger_id) |
|
108
|
|
|
kwargs = {'trigger': trigger} |
|
109
|
|
|
else: |
|
110
|
|
|
now = arrow.utcnow().to(settings.TIME_ZONE).format( |
|
111
|
|
|
'YYYY-MM-DD HH:mm:ssZZ') |
|
112
|
|
|
|
|
113
|
|
|
cache.set('django_th' + '_fire_trigger_' + str(trigger_id), '*') |
|
114
|
|
|
management.call_command('read_n_pub', trigger_id=trigger_id) |
|
115
|
|
|
|
|
116
|
|
|
trigger = TriggerService.objects.get(id=trigger_id) |
|
117
|
|
|
date_result = arrow.get(trigger.date_result).to(settings.TIME_ZONE)\ |
|
118
|
|
|
.format('YYYY-MM-DD HH:mm:ssZZ') |
|
119
|
|
|
date_triggered = arrow.get(trigger.date_triggered).\ |
|
120
|
|
|
to(settings.TIME_ZONE).format('YYYY-MM-DD HH:mm:ssZZ') |
|
121
|
|
|
|
|
122
|
|
|
if date_result < date_triggered and date_triggered > now: |
|
123
|
|
|
date = '*' |
|
124
|
|
|
|
|
125
|
|
|
template = 'triggers/fire_trigger.html' |
|
126
|
|
|
kwargs = {'trigger': trigger, 'date': date} |
|
127
|
|
|
|
|
128
|
|
|
return render(request, template, kwargs) |
|
129
|
|
|
|
|
130
|
|
|
|
|
131
|
|
|
def service_related_triggers_switch_to(request, user_service_id, switch): |
|
132
|
|
|
""" |
|
133
|
|
|
switch the status of all the triggers related to the service, |
|
134
|
|
|
then go back home |
|
135
|
|
|
:param request: request object |
|
136
|
|
|
:param user_service_id: the service ID to switch the status to |
|
137
|
|
|
True or False of all the related trigger |
|
138
|
|
|
:param switch: the switch value |
|
139
|
|
|
:type request: HttpRequest object |
|
140
|
|
|
:type user_service_id: int |
|
141
|
|
|
:type switch: string off or on |
|
142
|
|
|
""" |
|
143
|
|
|
status = True |
|
144
|
|
|
if switch == 'off': |
|
145
|
|
|
status = False |
|
146
|
|
|
|
|
147
|
|
|
TriggerService.objects.filter(provider__id=user_service_id).update( |
|
148
|
|
|
status=status) |
|
149
|
|
|
TriggerService.objects.filter(consumer__id=user_service_id).update( |
|
150
|
|
|
status=status) |
|
151
|
|
|
|
|
152
|
|
|
return HttpResponseRedirect(reverse('user_services')) |
|
153
|
|
|
|
|
154
|
|
|
|
|
155
|
|
|
def trigger_switch_all_to(request, switch): |
|
156
|
|
|
""" |
|
157
|
|
|
switch the status of all the "my" triggers then go back home |
|
158
|
|
|
:param request: request object |
|
159
|
|
|
:param switch: the switch value |
|
160
|
|
|
:type request: HttpRequest object |
|
161
|
|
|
:type switch: string off or on |
|
162
|
|
|
""" |
|
163
|
|
|
now = arrow.utcnow().to(settings.TIME_ZONE).format('YYYY-MM-DD HH:mm:ss') |
|
164
|
|
|
status = True |
|
165
|
|
|
if switch == 'off': |
|
166
|
|
|
status = False |
|
167
|
|
|
if status: |
|
168
|
|
|
TriggerService.objects.filter(user=request.user).update( |
|
169
|
|
|
status=status, date_triggered=now) |
|
170
|
|
|
else: |
|
171
|
|
|
TriggerService.objects.filter(user=request.user).update(status=status) |
|
172
|
|
|
|
|
173
|
|
|
return HttpResponseRedirect(reverse('base')) |
|
174
|
|
|
|
|
175
|
|
|
|
|
176
|
|
|
def list_services(request, step): |
|
177
|
|
|
""" |
|
178
|
|
|
get the activated services added from the administrator |
|
179
|
|
|
:param request: request object |
|
180
|
|
|
:param step: the step which is proceeded |
|
181
|
|
|
:type request: HttpRequest object |
|
182
|
|
|
:type step: string |
|
183
|
|
|
:return the activated services added from the administrator |
|
184
|
|
|
""" |
|
185
|
|
|
all_datas = [] |
|
186
|
|
|
|
|
187
|
|
|
if step == '0': |
|
188
|
|
|
services = ServicesActivated.objects.filter(status=1) |
|
189
|
|
|
elif step == '3': |
|
190
|
|
|
services = ServicesActivated.objects.filter(status=1, |
|
191
|
|
|
id__iexact=request.id) |
|
192
|
|
|
for class_name in services: |
|
193
|
|
|
all_datas.append({class_name: class_name.name.rsplit('Service', 1)[1]}) |
|
194
|
|
|
|
|
195
|
|
|
return all_datas |
|
196
|
|
|
|
|
197
|
|
|
|
|
198
|
|
|
def trigger_edit(request, trigger_id, edit_what): |
|
199
|
|
|
""" |
|
200
|
|
|
edit the provider |
|
201
|
|
|
:param request: request object |
|
202
|
|
|
:param trigger_id: ID of the trigger to edit |
|
203
|
|
|
:param edit_what: edit a 'Provider' or 'Consumer' ? |
|
204
|
|
|
:type request: HttpRequest object |
|
205
|
|
|
:type trigger_id: int |
|
206
|
|
|
:type edit_what: string |
|
207
|
|
|
:return render |
|
208
|
|
|
:rtype HttpResponse |
|
209
|
|
|
""" |
|
210
|
|
|
if edit_what not in ('Provider', 'Consumer'): |
|
211
|
|
|
# bad request |
|
212
|
|
|
return redirect('base') |
|
213
|
|
|
|
|
214
|
|
|
form_name = edit_what + 'Form' |
|
215
|
|
|
|
|
216
|
|
|
# get the trigger object |
|
217
|
|
|
service = TriggerService.objects.get(id=trigger_id) |
|
218
|
|
|
|
|
219
|
|
|
if can_modify_trigger(request, |
|
220
|
|
|
service.provider.name.status, |
|
221
|
|
|
service.consumer.name.status): |
|
222
|
|
|
return HttpResponseRedirect(reverse('base')) |
|
223
|
|
|
|
|
224
|
|
|
if edit_what == 'Consumer': |
|
225
|
|
|
my_service = service.consumer.name.name |
|
226
|
|
|
else: |
|
227
|
|
|
my_service = service.provider.name.name |
|
228
|
|
|
|
|
229
|
|
|
# get the service name |
|
230
|
|
|
service_name = str(my_service).split('Service')[1] |
|
231
|
|
|
# get the model of this service |
|
232
|
|
|
model = get_service(my_service) |
|
233
|
|
|
|
|
234
|
|
|
# get the data of this service linked to that trigger |
|
235
|
|
|
data = model.objects.get(trigger_id=trigger_id) |
|
236
|
|
|
|
|
237
|
|
|
template = service_name.lower() + '/edit_' + edit_what.lower() + ".html" |
|
238
|
|
|
|
|
239
|
|
|
if request.method == 'POST': |
|
240
|
|
|
form = get_service(my_service, 'forms', form_name)( |
|
241
|
|
|
request.POST, instance=data) |
|
242
|
|
|
if form.is_valid(): |
|
243
|
|
|
form.save() |
|
244
|
|
|
return HttpResponseRedirect(reverse('trigger_edit_thanks')) |
|
245
|
|
|
else: |
|
246
|
|
|
form = get_service(my_service, 'forms', form_name)(instance=data) |
|
247
|
|
|
|
|
248
|
|
|
context = {'description': service.description, 'edit_what': edit_what} |
|
249
|
|
|
return render(request, template, {'form': form, 'context': context}) |
|
250
|
|
|
|
|
251
|
|
|
|
|
252
|
|
|
class TriggerListView(ListView): |
|
253
|
|
|
""" |
|
254
|
|
|
list of Triggers |
|
255
|
|
|
the list can be filtered by service |
|
256
|
|
|
""" |
|
257
|
|
|
context_object_name = "triggers_list" |
|
258
|
|
|
queryset = TriggerService.objects.all() |
|
259
|
|
|
template_name = "home.html" |
|
260
|
|
|
paginate_by = 3 |
|
261
|
|
|
|
|
262
|
|
|
def get_paginate_by(self, queryset): |
|
263
|
|
|
""" |
|
264
|
|
|
Get the number of items to paginate by, |
|
265
|
|
|
from the settings |
|
266
|
|
|
""" |
|
267
|
|
|
paginate_by = 3 |
|
268
|
|
|
if hasattr(settings, 'DJANGO_TH'): |
|
269
|
|
|
if settings.DJANGO_TH.get('paginate_by'): |
|
270
|
|
|
paginate_by = settings.DJANGO_TH['paginate_by'] |
|
271
|
|
|
return paginate_by |
|
272
|
|
|
|
|
273
|
|
|
def get_queryset(self): |
|
274
|
|
|
filtered_by = None |
|
275
|
|
|
# by default, sort by date_created |
|
276
|
|
|
ordered_by = (str('-date_triggered'), ) |
|
277
|
|
|
# get the Trigger of the connected user |
|
278
|
|
|
if self.request.user.is_authenticated(): |
|
279
|
|
|
# if the user selected a filter, get its ID |
|
280
|
|
|
if self.kwargs.get('trigger_filtered_by'): |
|
281
|
|
|
filtered_by = UserService.objects.filter( |
|
282
|
|
|
user=self.request.user, |
|
283
|
|
|
name=self.kwargs.get('trigger_filtered_by'))[0].id |
|
284
|
|
|
|
|
285
|
|
|
if self.kwargs.get('trigger_ordered_by'): |
|
286
|
|
|
""" |
|
287
|
|
|
sort by 'name' property in the related model UserService |
|
288
|
|
|
""" |
|
289
|
|
|
order_by = str(self.kwargs.get('trigger_ordered_by') + "__name") |
|
290
|
|
|
# append to the tuple, the selected 'trigger_ordered_by' |
|
291
|
|
|
# choosen in the dropdown |
|
292
|
|
|
ordered_by = (order_by, ) + ordered_by |
|
293
|
|
|
|
|
294
|
|
|
# no filter selected |
|
295
|
|
|
if filtered_by is None: |
|
296
|
|
|
return self.queryset.filter(user=self.request.user).order_by( |
|
297
|
|
|
*ordered_by).select_related('consumer__name', |
|
298
|
|
|
'provider__name') |
|
299
|
|
|
|
|
300
|
|
|
# filter selected : display all related triggers |
|
301
|
|
|
else: |
|
302
|
|
|
# here the queryset will do : |
|
303
|
|
|
# 1) get trigger of the connected user AND |
|
304
|
|
|
# 2) get the triggers where the provider OR the consumer match |
|
305
|
|
|
# the selected service |
|
306
|
|
|
return self.queryset.filter(user=self.request.user).filter( |
|
307
|
|
|
Q(provider=filtered_by) | |
|
308
|
|
|
Q(consumer=filtered_by)).order_by( |
|
309
|
|
|
*ordered_by).select_related('consumer__name', |
|
310
|
|
|
'provider__name') |
|
311
|
|
|
# otherwise return nothing when user is not connected |
|
312
|
|
|
return TriggerService.objects.none() |
|
313
|
|
|
|
|
314
|
|
|
def get_context_data(self, **kw): |
|
315
|
|
|
""" |
|
316
|
|
|
get the data of the view |
|
317
|
|
|
|
|
318
|
|
|
data are : |
|
319
|
|
|
1) number of triggers enabled |
|
320
|
|
|
2) number of triggers disabled |
|
321
|
|
|
3) number of activated services |
|
322
|
|
|
4) list of activated services by the connected user |
|
323
|
|
|
""" |
|
324
|
|
|
triggers_enabled = triggers_disabled = services_activated = () |
|
325
|
|
|
|
|
326
|
|
|
context = super(TriggerListView, self).get_context_data(**kw) |
|
327
|
|
|
|
|
328
|
|
|
if self.kwargs.get('trigger_filtered_by'): |
|
329
|
|
|
page_link = reverse('trigger_filter_by', |
|
330
|
|
|
kwargs={'trigger_filtered_by': |
|
331
|
|
|
self.kwargs.get('trigger_filtered_by')}) |
|
332
|
|
|
elif self.kwargs.get('trigger_ordered_by'): |
|
333
|
|
View Code Duplication |
page_link = reverse('trigger_order_by', |
|
|
|
|
|
|
334
|
|
|
kwargs={'trigger_ordered_by': |
|
335
|
|
|
self.kwargs.get('trigger_ordered_by')}) |
|
336
|
|
|
else: |
|
337
|
|
|
page_link = reverse('home') |
|
338
|
|
|
|
|
339
|
|
|
if self.request.user.is_authenticated(): |
|
340
|
|
|
# get the enabled triggers |
|
341
|
|
|
triggers_enabled = TriggerService.objects.filter( |
|
342
|
|
|
user=self.request.user, status=1).count() |
|
343
|
|
|
# get the disabled triggers |
|
344
|
|
|
triggers_disabled = TriggerService.objects.filter( |
|
345
|
|
|
user=self.request.user, status=0).count() |
|
346
|
|
|
# get the activated services |
|
347
|
|
|
user_service = UserService.objects.filter(user=self.request.user) |
|
348
|
|
|
""" |
|
349
|
|
|
List of triggers activated by the user |
|
350
|
|
|
""" |
|
351
|
|
|
context['trigger_filter_by'] = user_service |
|
352
|
|
|
""" |
|
353
|
|
|
number of service activated for the current user |
|
354
|
|
|
""" |
|
355
|
|
|
services_activated = user_service.count() |
|
356
|
|
|
|
|
357
|
|
|
""" |
|
358
|
|
|
which triggers are enabled/disabled |
|
359
|
|
|
""" |
|
360
|
|
|
context['nb_triggers'] = {'enabled': triggers_enabled, |
|
361
|
|
|
'disabled': triggers_disabled} |
|
362
|
|
|
""" |
|
363
|
|
|
Number of services activated |
|
364
|
|
|
""" |
|
365
|
|
|
context['nb_services'] = services_activated |
|
366
|
|
|
|
|
367
|
|
|
context['page_link'] = page_link |
|
368
|
|
|
|
|
369
|
|
|
return context |
|
370
|
|
|
|
|
371
|
|
|
|
|
372
|
|
|
class TriggerServiceMixin(object): |
|
373
|
|
|
""" |
|
374
|
|
|
Mixin for UpdateView and DeleteView |
|
375
|
|
|
""" |
|
376
|
|
|
queryset = TriggerService.objects.all() |
|
377
|
|
|
|
|
378
|
|
|
@method_decorator(login_required) |
|
379
|
|
|
def dispatch(self, *args, **kwargs): |
|
380
|
|
|
return super(TriggerServiceMixin, self).dispatch(*args, **kwargs) |
|
381
|
|
|
|
|
382
|
|
|
def get_queryset(self): |
|
383
|
|
|
# get the trigger of the connected user |
|
384
|
|
|
if self.request.user.is_authenticated(): |
|
385
|
|
|
return self.queryset.filter(user=self.request.user, |
|
386
|
|
|
id=self.kwargs.get('pk')) |
|
387
|
|
|
# otherwise return nothing |
|
388
|
|
|
return TriggerService.objects.none() |
|
389
|
|
|
|
|
390
|
|
|
|
|
391
|
|
|
class TriggerUpdateView(TriggerServiceMixin, UpdateView): |
|
392
|
|
|
""" |
|
393
|
|
|
Form to update description |
|
394
|
|
|
""" |
|
395
|
|
|
form_class = TriggerServiceForm |
|
396
|
|
|
template_name = "triggers/edit_description_trigger.html" |
|
397
|
|
|
success_url = reverse_lazy("trigger_edit_thanks") |
|
398
|
|
|
|
|
399
|
|
|
def get_context_data(self, **kw): |
|
400
|
|
|
return super(TriggerUpdateView, self).get_context_data(**kw) |
|
401
|
|
|
|
|
402
|
|
|
def get(self, *args, **kwargs): |
|
403
|
|
|
# Go through keyword arguments, and either save their values to our |
|
404
|
|
|
# instance, or raise an error. |
|
405
|
|
|
self.object = self.get_object() |
|
406
|
|
|
status = can_modify_trigger(self.request, |
|
407
|
|
|
self.object.provider.name.status, |
|
408
|
|
|
self.object.consumer.name.status) |
|
409
|
|
|
if status: |
|
410
|
|
|
return HttpResponseRedirect(reverse('base')) |
|
411
|
|
|
else: |
|
412
|
|
|
return super(TriggerUpdateView, self).get( |
|
413
|
|
|
self.request, *args, **kwargs) |
|
414
|
|
|
|
|
415
|
|
|
|
|
416
|
|
|
class TriggerEditedTemplateView(TemplateView): |
|
417
|
|
|
""" |
|
418
|
|
|
just a simple form to say thanks :P |
|
419
|
|
|
""" |
|
420
|
|
|
template_name = "triggers/thanks_trigger.html" |
|
421
|
|
|
|
|
422
|
|
|
def get_context_data(self, **kw): |
|
423
|
|
|
context = super(TriggerEditedTemplateView, self).get_context_data(**kw) |
|
424
|
|
|
context['sentence'] = 'Your trigger has been successfully modified' |
|
425
|
|
|
return context |
|
426
|
|
|
|
|
427
|
|
|
|
|
428
|
|
|
class TriggerDeleteView(TriggerServiceMixin, DeleteView): |
|
429
|
|
|
""" |
|
430
|
|
|
page to delete a trigger |
|
431
|
|
|
""" |
|
432
|
|
|
template_name = "triggers/delete_trigger.html" |
|
433
|
|
|
success_url = reverse_lazy("trigger_delete_thanks") |
|
434
|
|
|
|
|
435
|
|
|
|
|
436
|
|
|
class TriggerDeletedTemplateView(TemplateView): |
|
437
|
|
|
""" |
|
438
|
|
|
just a simple form to say thanks :P |
|
439
|
|
|
""" |
|
440
|
|
|
template_name = "triggers/thanks_trigger.html" |
|
441
|
|
|
|
|
442
|
|
|
def get_context_data(self, **kw): |
|
443
|
|
|
context = super(TriggerDeletedTemplateView, self).\ |
|
444
|
|
|
get_context_data(**kw) |
|
445
|
|
|
context['sentence'] = 'Your trigger has been successfully deleted' |
|
446
|
|
|
return context |
|
447
|
|
|
|