| @@ 316-331 (lines=16) @@ | ||
| 313 | ||
| 314 | return context |
|
| 315 | ||
| 316 | ||
| 317 | class TriggerServiceMixin(object): |
|
| 318 | """ |
|
| 319 | Mixin for UpdateView and DeleteView |
|
| 320 | """ |
|
| 321 | queryset = TriggerService.objects.all() |
|
| 322 | ||
| 323 | @method_decorator(login_required) |
|
| 324 | def dispatch(self, *args, **kwargs): |
|
| 325 | return super(TriggerServiceMixin, self).dispatch(*args, **kwargs) |
|
| 326 | ||
| 327 | def get_queryset(self): |
|
| 328 | # get the trigger of the connected user |
|
| 329 | if self.request.user.is_authenticated(): |
|
| 330 | return self.queryset.filter(user=self.request.user, |
|
| 331 | id=self.kwargs.get('pk')) |
|
| 332 | # otherwise return nothing |
|
| 333 | return TriggerService.objects.none() |
|
| 334 | ||
| @@ 137-153 (lines=17) @@ | ||
| 134 | return kwargs |
|
| 135 | ||
| 136 | ||
| 137 | class UserServiceMixin(object): |
|
| 138 | """ |
|
| 139 | Mixin for UpdateView and DeleteView |
|
| 140 | """ |
|
| 141 | queryset = UserService.objects.all() |
|
| 142 | ||
| 143 | @method_decorator(login_required) |
|
| 144 | def dispatch(self, *args, **kwargs): |
|
| 145 | return super(UserServiceMixin, self).dispatch(*args, **kwargs) |
|
| 146 | ||
| 147 | def get_queryset(self): |
|
| 148 | # get the Service of the connected user |
|
| 149 | if self.request.user.is_authenticated(): |
|
| 150 | return self.queryset.filter(user=self.request.user, |
|
| 151 | id=self.kwargs.get('pk')) |
|
| 152 | # otherwise return nothing |
|
| 153 | return UserService.objects.none() |
|
| 154 | ||
| 155 | ||
| 156 | class UserServiceUpdateView(UserServiceMixin, UpdateView): |
|