nkey/colio
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 Fix: [#30] check_user_portfolio_rel 시 try except

Browse Source
This commit is contained in:
sm4640
2025-04-12 15:53:55 +09:00
parent 493622ab7e
commit a7dc98ee74
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
portfolios/views.py
View File

@@ -77,8 +77,11 @@ class PortfolioChangeState(APIView):
portfolio = get_object_or_404(Portfolio, pk=pk)
user = request.user
action_type = request.query_params.get('type')
if PortfolioBeforeRelCheckService.check_user_portfolio_rel(action_type, portfolio, user):
return Response({"message": "already done"}, status=status.HTTP_400_BAD_REQUEST)
try:
if PortfolioBeforeRelCheckService.check_user_portfolio_rel(action_type, portfolio, user):
return Response({"message": "already done"}, status=status.HTTP_400_BAD_REQUEST)
except ValueError as e:
return Response({'message': str(e)}, status=status.HTTP_400_BAD_REQUEST)
return self._handle_action(action_type, portfolio, user, add=True)
@transaction.atomic
@@ -86,8 +89,11 @@ class PortfolioChangeState(APIView):
portfolio = get_object_or_404(Portfolio, pk=pk)
user = request.user
action_type = request.query_params.get('type')
if not PortfolioBeforeRelCheckService.check_user_portfolio_rel(action_type, portfolio, user):
return Response({"message": "never done before"}, status=status.HTTP_400_BAD_REQUEST)
try:
if not PortfolioBeforeRelCheckService.check_user_portfolio_rel(action_type, portfolio, user):
return Response({"message": "never done before"}, status=status.HTTP_400_BAD_REQUEST)
except ValueError as e:
return Response({'message': str(e)}, status=status.HTTP_400_BAD_REQUEST)
return self._handle_action(action_type, portfolio, user, add=False)
def _handle_action(self, action_type, portfolio, user, add=True):