Fixed FC-Gifu

2024-08-26 09:02:01 +00:00
parent 95b787c819
commit 80ccaace3d
6 changed files with 108 additions and 29 deletions
--- a/rog/permissions.py
+++ b/rog/permissions.py
@ -14,3 +14,12 @@ class IsTeamOwner(permissions.BasePermission):
        elif isinstance(obj, Member):
            return obj.team.owner == request.user
        return False
+
+class IsTeamOwnerOrMember(permissions.BasePermission):
+    def has_permission(self, request, view):
+        team_id = view.kwargs.get('team_id')
+        if not team_id:
+            return False
+        team = Team.objects.get(id=team_id)
+        return team.owner == request.user or team.members.filter(user=request.user).exists()
+