| @@ 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): |
|
| @@ 333-349 (lines=17) @@ | ||
| 330 | return context |
|
| 331 | ||
| 332 | ||
| 333 | class TriggerServiceMixin(object): |
|
| 334 | """ |
|
| 335 | Mixin for UpdateView and DeleteView |
|
| 336 | """ |
|
| 337 | queryset = TriggerService.objects.all() |
|
| 338 | ||
| 339 | @method_decorator(login_required) |
|
| 340 | def dispatch(self, *args, **kwargs): |
|
| 341 | return super(TriggerServiceMixin, self).dispatch(*args, **kwargs) |
|
| 342 | ||
| 343 | def get_queryset(self): |
|
| 344 | # get the trigger of the connected user |
|
| 345 | if self.request.user.is_authenticated(): |
|
| 346 | return self.queryset.filter(user=self.request.user, |
|
| 347 | id=self.kwargs.get('pk')) |
|
| 348 | # otherwise return nothing |
|
| 349 | return TriggerService.objects.none() |
|
| 350 | ||
| 351 | ||
| 352 | class TriggerUpdateView(TriggerServiceMixin, UpdateView): |
|