blob: 516fb298ad9991848250a871ebe7adf7c3179f66 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
from django.contrib.auth.mixins import UserPassesTestMixin
class BookingUserTestMixin(UserPassesTestMixin):
# Test if a user has permission to book this Pod
def test_func(self):
user = self.request.user
# Check if User is troubleshooter / admin
if user.has_perm(('dashboard.add_booking')):
return True
# Check if User owns this resource
user_resources = user.userresource_set.get_queryset()
for user_resource in user_resources:
if user_resource.resource_id == self.resource.resource_id:
return True
return False
|
