Skip to content

Commit 0a1d5c4

Browse files
wxg0103wxg0103
committed
refactor: permission constants
1 parent fc0599f commit 0a1d5c4

File tree

10 files changed

+4405
-4207
lines changed

10 files changed

+4405
-4207
lines changed

‎apps/common/constants/exception_code_constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,3 +41,4 @@ class ExceptionCodeConstants(Enum):
4141
_('The username cannot be empty and must be between 6 and 20 characters long.'))
4242
PASSWORD_NOT_EQ_RE_PASSWORD = ExceptionCodeConstantsValue(1007,
4343
_('Password and confirmation password are inconsistent'))
44+
NICKNAME_IS_EXIST = ExceptionCodeConstantsValue(1008, _('The nickname is already registered'))

‎apps/common/constants/permission_constants.py

Lines changed: 217 additions & 72 deletions
Large diffs are not rendered by default.

‎apps/knowledge/views/document.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ class SyncWeb(APIView):
188188
responses=SyncWebAPI.get_response(),
189189
tags=[_('Knowledge Base/Documentation')] # type: ignore
190190
)
191-
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
191+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_SYNC.get_workspace_permission())
192192
def get(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str):
193193
return result.success(DocumentSerializers.Sync(
194194
data={'document_id': document_id, 'knowledge_id': knowledge_id, 'workspace_id': workspace_id}
@@ -207,7 +207,7 @@ class Refresh(APIView):
207207
responses=RefreshAPI.get_response(),
208208
tags=[_('Knowledge Base/Documentation')] # type: ignore
209209
)
210-
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
210+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_VECTOR.get_workspace_permission())
211211
def put(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str):
212212
return result.success(DocumentSerializers.Operate(
213213
data={'document_id': document_id, 'knowledge_id': knowledge_id, 'workspace_id': workspace_id}
@@ -285,7 +285,7 @@ class BatchSync(APIView):
285285
tags=[_('Knowledge Base/Documentation')] # type: ignore
286286
)
287287
@has_permissions(
288-
PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(),
288+
PermissionConstants.KNOWLEDGE_DOCUMENT_SYNC.get_workspace_permission(),
289289
PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(),
290290
)
291291
def put(self, request: Request, workspace_id: str, knowledge_id: str):
@@ -307,7 +307,7 @@ class BatchDelete(APIView):
307307
tags=[_('Knowledge Base/Documentation')] # type: ignore
308308
)
309309
@has_permissions(
310-
PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(),
310+
PermissionConstants.KNOWLEDGE_DOCUMENT_DELETE.get_workspace_permission(),
311311
PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(),
312312
)
313313
def put(self, request: Request, workspace_id: str, knowledge_id: str):
@@ -328,7 +328,7 @@ class BatchRefresh(APIView):
328328
tags=[_('Knowledge Base/Documentation')] # type: ignore
329329
)
330330
@has_permissions(
331-
PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(),
331+
PermissionConstants.KNOWLEDGE_DOCUMENT_VECTOR.get_workspace_permission(),
332332
PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(),
333333
)
334334
def put(self, request: Request, workspace_id: str, knowledge_id: str):
@@ -351,7 +351,7 @@ class BatchGenerateRelated(APIView):
351351
tags=[_('Knowledge Base/Documentation')] # type: ignore
352352
)
353353
@has_permissions(
354-
PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission(),
354+
PermissionConstants.KNOWLEDGE_DOCUMENT_GENERATE.get_workspace_permission(),
355355
PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission(),
356356
)
357357
def put(self, request: Request, workspace_id: str, knowledge_id: str):

‎apps/knowledge/views/knowledge.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ class SyncWeb(APIView):
125125
responses=SyncWebAPI.get_response(),
126126
tags=[_('Knowledge Base')] # type: ignore
127127
)
128-
@has_permissions(PermissionConstants.KNOWLEDGE_EDIT.get_workspace_permission())
128+
@has_permissions(PermissionConstants.KNOWLEDGE_SYNC.get_workspace_permission())
129129
def put(self, request: Request, workspace_id: str, knowledge_id: str):
130130
return result.success(KnowledgeSerializer.SyncWeb(
131131
data={
@@ -176,7 +176,7 @@ class Embedding(APIView):
176176
responses=EmbeddingAPI.get_response(),
177177
tags=[_('Knowledge Base')] # type: ignore
178178
)
179-
@has_permissions(PermissionConstants.KNOWLEDGE_EDIT.get_workspace_permission())
179+
@has_permissions(PermissionConstants.KNOWLEDGE_VECTOR.get_workspace_permission())
180180
def put(self, request: Request, workspace_id: str, knowledge_id: str):
181181
return result.success(KnowledgeSerializer.Operate(
182182
data={'knowledge_id': knowledge_id, 'workspace_id': workspace_id, 'user_id': request.user.id}
@@ -195,7 +195,7 @@ class GenerateRelated(APIView):
195195
responses=GenerateRelatedAPI.get_response(),
196196
tags=[_('Knowledge Base')] # type: ignore
197197
)
198-
@has_permissions(PermissionConstants.KNOWLEDGE_EDIT.get_workspace_permission())
198+
@has_permissions(PermissionConstants.KNOWLEDGE_GENERATE.get_workspace_permission())
199199
def put(self, request: Request, workspace_id: str, knowledge_id: str):
200200
return result.success(KnowledgeSerializer.Operate(
201201
data={'knowledge_id': knowledge_id, 'workspace_id': workspace_id, 'user_id': request.user.id}

‎apps/knowledge/views/paragraph.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ class ParagraphView(APIView):
2525
responses=ParagraphReadAPI.get_response(),
2626
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
2727
)
28-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_READ.get_workspace_permission())
28+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_READ.get_workspace_permission())
2929
def get(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str):
3030
q = ParagraphSerializers.Query(
3131
data={
@@ -46,7 +46,7 @@ def get(self, request: Request, workspace_id: str, knowledge_id: str, document_i
4646
responses=ParagraphCreateAPI.get_response(),
4747
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
4848
)
49-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_CREATE.get_workspace_permission())
49+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_CREATE.get_workspace_permission())
5050
def post(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str):
5151
return result.success(ParagraphSerializers.Create(
5252
data={'workspace_id': workspace_id, 'knowledge_id': knowledge_id, 'document_id': document_id}
@@ -65,7 +65,7 @@ class BatchDelete(APIView):
6565
responses=ParagraphBatchDeleteAPI.get_response(),
6666
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
6767
)
68-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
68+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
6969
def put(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str):
7070
return result.success(ParagraphSerializers.Batch(
7171
data={'workspace_id': workspace_id, 'knowledge_id': knowledge_id, 'document_id': document_id}
@@ -84,7 +84,7 @@ class BatchGenerateRelated(APIView):
8484
responses=ParagraphBatchGenerateRelatedAPI.get_response(),
8585
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
8686
)
87-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
87+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_GENERATE.get_workspace_permission())
8888
def put(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str):
8989
return result.success(ParagraphSerializers.Batch(
9090
data={'workspace_id': workspace_id, 'knowledge_id': knowledge_id, 'document_id': document_id}
@@ -103,7 +103,7 @@ class Operate(APIView):
103103
responses=ParagraphEditAPI.get_response(),
104104
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
105105
)
106-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
106+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
107107
def put(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str, paragraph_id: str):
108108
o = ParagraphSerializers.Operate(
109109
data={
@@ -125,7 +125,7 @@ def put(self, request: Request, workspace_id: str, knowledge_id: str, document_i
125125
responses=ParagraphGetAPI.get_response(),
126126
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
127127
)
128-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
128+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
129129
def get(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str, paragraph_id: str):
130130
o = ParagraphSerializers.Operate(
131131
data={
@@ -146,7 +146,7 @@ def get(self, request: Request, workspace_id: str, knowledge_id: str, document_i
146146
parameters=ParagraphGetAPI.get_parameters(),
147147
responses=ParagraphGetAPI.get_response(),
148148
tags=[_('Knowledge Base/Documentation/Paragraph')]) # type: ignore
149-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
149+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
150150
def delete(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str, paragraph_id: str):
151151
o = ParagraphSerializers.Operate(
152152
data={
@@ -172,7 +172,7 @@ class Problem(APIView):
172172
responses=ProblemCreateAPI.get_response(),
173173
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
174174
)
175-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
175+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
176176
def post(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str, paragraph_id: str):
177177
return result.success(ParagraphSerializers.Problem(
178178
data={
@@ -192,7 +192,7 @@ def post(self, request: Request, workspace_id: str, knowledge_id: str, document_
192192
responses=ParagraphGetAPI.get_response(),
193193
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
194194
)
195-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
195+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
196196
def get(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str, paragraph_id: str):
197197
return result.success(ParagraphSerializers.Problem(
198198
data={
@@ -216,7 +216,7 @@ class UnAssociation(APIView):
216216
responses=UnAssociationAPI.get_response(),
217217
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
218218
)
219-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
219+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
220220
def put(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str):
221221
return result.success(ParagraphSerializers.Association(
222222
data={
@@ -241,7 +241,7 @@ class Association(APIView):
241241
responses=AssociationAPI.get_response(),
242242
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
243243
)
244-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
244+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
245245
def put(self, request: Request, workspace_id: str, knowledge_id: str, document_id: str):
246246
return result.success(ParagraphSerializers.Association(
247247
data={
@@ -265,7 +265,7 @@ class Page(APIView):
265265
responses=ParagraphPageAPI.get_response(),
266266
tags=[_('Knowledge Base/Documentation/Paragraph')] # type: ignore
267267
)
268-
@has_permissions(PermissionConstants.KNOWLEDGE_PARAGRAPH_EDIT.get_workspace_permission())
268+
@has_permissions(PermissionConstants.KNOWLEDGE_DOCUMENT_EDIT.get_workspace_permission())
269269
def get(self, request: Request,
270270
workspace_id: str, knowledge_id: str, document_id: str, current_page: int, page_size: int):
271271
d = ParagraphSerializers.Query(

0 commit comments

Comments
 (0)