from rest_framework.viewsets import ModelViewSet
from . import models

from rest_framework.permissions import BasePermission

class IsOwnerPermission(BasePermission):
    def has_object_permission(self, request, view, obj):
        return request.user == obj.owner

class TeamViewSet(ModelViewSet):
    model = models.Team
    permission_classes = (IsOwnerPermission, )

class PlayerViewSet(ModelViewSet):
    model = models.Player

from rest_framework.viewsets import ModelViewSet
from . import models
from . import permisssions

class TeamViewSet(ModelViewSet):
    model = models.Team
    permission_classes = (permisssions.IsOwnerPermission, )

class PlayerViewSet(ModelViewSet):
    model = models.Player

Move to a module permissions
Move to a module permissions

  from rest_framework.viewsets import ModelViewSet
  from . import models
+ from . import permisssions
- 
- from rest_framework.permissions import BasePermission
- 
- class IsOwnerPermission(BasePermission):
-     def has_object_permission(self, request, view, obj):
-         return request.user == obj.owner
  
  class TeamViewSet(ModelViewSet):
      model = models.Team
-     permission_classes = (IsOwnerPermission, )
+     permission_classes = (permisssions.IsOwnerPermission, )
  
  class PlayerViewSet(ModelViewSet):
      model = models.Player
  
Move to a module permissions
## Code Before:
from rest_framework.viewsets import ModelViewSet
from . import models

from rest_framework.permissions import BasePermission

class IsOwnerPermission(BasePermission):
    def has_object_permission(self, request, view, obj):
        return request.user == obj.owner

class TeamViewSet(ModelViewSet):
    model = models.Team
    permission_classes = (IsOwnerPermission, )

class PlayerViewSet(ModelViewSet):
    model = models.Player

## Instruction:
Move to a module permissions
## Code After:
from rest_framework.viewsets import ModelViewSet
from . import models
from . import permisssions

class TeamViewSet(ModelViewSet):
    model = models.Team
    permission_classes = (permisssions.IsOwnerPermission, )

class PlayerViewSet(ModelViewSet):
    model = models.Player
