| @@ 121-137 (lines=17) @@ | ||
| 118 | return kwargs |
|
| 119 | ||
| 120 | ||
| 121 | class UserServiceMixin(object): |
|
| 122 | """ |
|
| 123 | Mixin for UpdateView and DeleteView |
|
| 124 | """ |
|
| 125 | queryset = UserService.objects.all() |
|
| 126 | ||
| 127 | @method_decorator(login_required) |
|
| 128 | def dispatch(self, *args, **kwargs): |
|
| 129 | return super(UserServiceMixin, self).dispatch(*args, **kwargs) |
|
| 130 | ||
| 131 | def get_queryset(self): |
|
| 132 | # get the Service of the connected user |
|
| 133 | if self.request.user.is_authenticated(): |
|
| 134 | return self.queryset.filter(user=self.request.user, |
|
| 135 | id=self.kwargs.get('pk')) |
|
| 136 | # otherwise return nothing |
|
| 137 | return UserService.objects.none() |
|
| 138 | ||
| 139 | ||
| 140 | class UserServiceUpdateView(UserServiceMixin, UpdateView): |
|
| @@ 316-331 (lines=16) @@ | ||
| 313 | return context |
|
| 314 | ||
| 315 | ||
| 316 | class TriggerServiceMixin(object): |
|
| 317 | """ |
|
| 318 | Mixin for UpdateView and DeleteView |
|
| 319 | """ |
|
| 320 | queryset = TriggerService.objects.all() |
|
| 321 | ||
| 322 | @method_decorator(login_required) |
|
| 323 | def dispatch(self, *args, **kwargs): |
|
| 324 | return super(TriggerServiceMixin, self).dispatch(*args, **kwargs) |
|
| 325 | ||
| 326 | def get_queryset(self): |
|
| 327 | # get the trigger of the connected user |
|
| 328 | if self.request.user.is_authenticated(): |
|
| 329 | return self.queryset.filter(user=self.request.user, id=self.kwargs.get('pk')) |
|
| 330 | # otherwise return nothing |
|
| 331 | return TriggerService.objects.none() |
|
| 332 | ||
| 333 | ||
| 334 | class TriggerUpdateView(TriggerServiceMixin, UpdateView): |
|