#88 Feat: 팀장-> 팀 완료 API 작성

whatTeamNaeTeam · Aug 28, 2024 · 76ca0ef · 76ca0ef
1 parent 8616f92
commit 76ca0ef
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/wtnt/user/profile/service.py b/wtnt/user/profile/service.py
@@ -251,11 +251,29 @@ def get_my_team_detail(self):
 
         return data
 
+    def finish_project(self):
+        team_id = self.kwargs.get("team_id")
+        user_id = self.request.user.id
+        try:
+            team = Team.objects.select_related("leader").get(id=team_id)
+        except Team.DoesNotExist:
+            raise notfound_exception.TeamNotFoundError()
+
+        self.check_leader(user_id, team.leader.id)
+
+        team.is_accomplished = True
+        team.save()
+
+        return {"detail": f"{team.title} is accomplished."}
+
     def ban_user_from_team(self):
         team_id = self.kwargs.get("team_id")
         user_id = self.request.user.id
 
-        team = Team.objects.select_related("leader").get(id=team_id)
+        try:
+            team = Team.objects.select_related("leader").get(id=team_id)
+        except Team.DoesNotExist:
+            raise notfound_exception.TeamNotFoundError()
 
         self.check_leader(user_id, team.leader.id)