ruff reformatted

pull/20023/head
lizb 1 year ago
parent 1ac3245ef2
commit 641b28dd02

@ -339,8 +339,11 @@ class DatasetTagsApi(DatasetApiResource):
parser = reqparse.RequestParser() parser = reqparse.RequestParser()
parser.add_argument( parser.add_argument(
"name", nullable=False, required=True, help="Name must be between 1 to 50 characters.", "name",
type=DatasetTagsApi._validate_tag_name nullable=False,
required=True,
help="Name must be between 1 to 50 characters.",
type=DatasetTagsApi._validate_tag_name,
) )
args = parser.parse_args() args = parser.parse_args()
@ -358,12 +361,13 @@ class DatasetTagsApi(DatasetApiResource):
parser = reqparse.RequestParser() parser = reqparse.RequestParser()
parser.add_argument( parser.add_argument(
"name", nullable=False, required=True, help="Name must be between 1 to 50 characters.", "name",
type=DatasetTagsApi._validate_tag_name nullable=False,
) required=True,
parser.add_argument( help="Name must be between 1 to 50 characters.",
"tag_id", nullable=False, required=True, help="Id of a tag.", type=str type=DatasetTagsApi._validate_tag_name,
) )
parser.add_argument("tag_id", nullable=False, required=True, help="Id of a tag.", type=str)
args = parser.parse_args() args = parser.parse_args()
tag = TagService.update_tags(args, args.get("tag_id")) tag = TagService.update_tags(args, args.get("tag_id"))
@ -379,9 +383,7 @@ class DatasetTagsApi(DatasetApiResource):
if not current_user.is_editor: if not current_user.is_editor:
raise Forbidden() raise Forbidden()
parser = reqparse.RequestParser() parser = reqparse.RequestParser()
parser.add_argument( parser.add_argument("tag_id", nullable=False, required=True, help="Id of a tag.", type=str)
"tag_id", nullable=False, required=True, help="Id of a tag.", type=str
)
args = parser.parse_args() args = parser.parse_args()
TagService.delete_tag(args.get("tag_id")) TagService.delete_tag(args.get("tag_id"))
@ -395,7 +397,6 @@ class DatasetTagsApi(DatasetApiResource):
class DatasetTagBindingApi(DatasetApiResource): class DatasetTagBindingApi(DatasetApiResource):
@validate_dataset_token @validate_dataset_token
def post(self, _, dataset_id): def post(self, _, dataset_id):
# The role of the current user in the ta table must be admin, owner, editor, or dataset_operator # The role of the current user in the ta table must be admin, owner, editor, or dataset_operator
@ -407,8 +408,7 @@ class DatasetTagBindingApi(DatasetApiResource):
"tag_ids", type=list, nullable=False, required=True, location="json", help="Tag IDs is required." "tag_ids", type=list, nullable=False, required=True, location="json", help="Tag IDs is required."
) )
parser.add_argument( parser.add_argument(
"target_id", type=str, nullable=False, required=True, location="json", "target_id", type=str, nullable=False, required=True, location="json", help="Target Dataset ID is required."
help="Target Dataset ID is required."
) )
args = parser.parse_args() args = parser.parse_args()
@ -419,7 +419,6 @@ class DatasetTagBindingApi(DatasetApiResource):
class DatasetTagUnbindingApi(DatasetApiResource): class DatasetTagUnbindingApi(DatasetApiResource):
@validate_dataset_token @validate_dataset_token
def post(self, _, dataset_id): def post(self, _, dataset_id):
# The role of the current user in the ta table must be admin, owner, editor, or dataset_operator # The role of the current user in the ta table must be admin, owner, editor, or dataset_operator
@ -438,7 +437,6 @@ class DatasetTagUnbindingApi(DatasetApiResource):
class DatasetTagsBindingStatusApi(DatasetApiResource): class DatasetTagsBindingStatusApi(DatasetApiResource):
@validate_dataset_token @validate_dataset_token
def get(self, _, *args, **kwargs): def get(self, _, *args, **kwargs):
"""Get all knowledge type tags.""" """Get all knowledge type tags."""
@ -448,6 +446,7 @@ class DatasetTagsBindingStatusApi(DatasetApiResource):
response = {"data": tags_list, "total": len(tags)} response = {"data": tags_list, "total": len(tags)}
return response, 200 return response, 200
api.add_resource(DatasetListApi, "/datasets") api.add_resource(DatasetListApi, "/datasets")
api.add_resource(DatasetApi, "/datasets/<uuid:dataset_id>") api.add_resource(DatasetApi, "/datasets/<uuid:dataset_id>")
api.add_resource(DatasetTagsApi, "/datasets/tags") api.add_resource(DatasetTagsApi, "/datasets/tags")

Loading…
Cancel
Save