88from application .models import ApplicationApiKey
99from application .serializers .application_api_key import ApplicationKeySerializer
1010from common .auth import TokenAuth
11+ from common .auth .authentication import has_permissions
12+ from common .constants .permission_constants import PermissionConstants
1113from common .log .log import log
1214from common .result import result , success
1315
@@ -34,10 +36,11 @@ class ApplicationKey(APIView):
3436 )
3537 @log (menu = 'Application' , operate = "Add ApiKey" ,
3638 get_operation_object = lambda r , k : get_application_operation_object (k .get ('application_api_key_id' )))
37- def post (self ,request : Request , application_id : str , workspace_id : str ):
39+ @has_permissions (PermissionConstants .APPLICATION_OVERVIEW_API_KEY .get_workspace_application_permission ())
40+ def post (self , request : Request , application_id : str , workspace_id : str ):
3841 return result .success (ApplicationKeySerializer (
39- data = {'application_id' : application_id , 'user_id' : request .user .id ,
40- 'workspace_id' :workspace_id }).generate ())
42+ data = {'application_id' : application_id , 'user_id' : request .user .id ,
43+ 'workspace_id' : workspace_id }).generate ())
4144
4245 @extend_schema (
4346 methods = ['GET' ],
@@ -47,11 +50,11 @@ def post(self,request: Request, application_id: str, workspace_id: str):
4750 parameters = ApplicationKeyCreateAPI .get_parameters (),
4851 tags = [_ ('Application Api Key' )] # type: ignore
4952 )
50- def get ( self , request : Request , application_id : str , workspace_id : str ):
51- return result , success ( ApplicationKeySerializer (
52- data = { 'application_id' : application_id , 'user_id' : request . user . id ,
53- 'workspace_id' : workspace_id }). list ())
54-
53+ @ has_permissions ( PermissionConstants . APPLICATION_OVERVIEW_API_KEY . get_workspace_application_permission ())
54+ def get ( self , request : Request , application_id : str , workspace_id : str ):
55+ return result , success ( ApplicationKeySerializer (
56+ data = { 'application_id' : application_id , 'user_id' : request . user . id ,
57+ 'workspace_id' : workspace_id }). list ())
5558
5659 class Operate (APIView ):
5760 authentication_classes = [TokenAuth ]
@@ -64,9 +67,6 @@ class Operate(APIView):
6467 parameters = ApplicationKeyCreateAPI .get_parameters (),
6568 tags = [_ ('Application Api Key' )] # type: ignore
6669 )
70+ @has_permissions (PermissionConstants .APPLICATION_OVERVIEW_API_KEY .get_workspace_application_permission ())
6771 def put (self , request : Request , application_id : str , workspace_id : str ):
68- return result .success (ApplicationKeySerializer .Operate (
69-
70- )
71- )
72-
72+ return result .success (ApplicationKeySerializer .Operate ())
0 commit comments