70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
# Copyright (C) CVAT.ai Corporation
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from __future__ import annotations
|
|
|
|
from cvat_sdk.api_client import apis, models
|
|
from cvat_sdk.core.helpers import get_paginated_collection
|
|
from cvat_sdk.core.proxies.model_proxy import (
|
|
ModelCreateMixin,
|
|
ModelDeleteMixin,
|
|
ModelListMixin,
|
|
ModelRetrieveMixin,
|
|
ModelUpdateMixin,
|
|
build_model_bases,
|
|
)
|
|
|
|
_CommentEntityBase, _CommentRepoBase = build_model_bases(
|
|
models.CommentRead, apis.CommentsApi, api_member_name="comments_api"
|
|
)
|
|
|
|
|
|
class Comment(
|
|
models.ICommentRead,
|
|
_CommentEntityBase,
|
|
ModelUpdateMixin[models.IPatchedCommentWriteRequest],
|
|
ModelDeleteMixin,
|
|
):
|
|
_model_partial_update_arg = "patched_comment_write_request"
|
|
|
|
|
|
class CommentsRepo(
|
|
_CommentRepoBase,
|
|
ModelListMixin[Comment],
|
|
ModelCreateMixin[Comment, models.ICommentWriteRequest],
|
|
ModelRetrieveMixin[Comment],
|
|
):
|
|
_entity_type = Comment
|
|
|
|
|
|
_IssueEntityBase, _IssueRepoBase = build_model_bases(
|
|
models.IssueRead, apis.IssuesApi, api_member_name="issues_api"
|
|
)
|
|
|
|
|
|
class Issue(
|
|
models.IIssueRead,
|
|
_IssueEntityBase,
|
|
ModelUpdateMixin[models.IPatchedIssueWriteRequest],
|
|
ModelDeleteMixin,
|
|
):
|
|
_model_partial_update_arg = "patched_issue_write_request"
|
|
|
|
def get_comments(self) -> list[Comment]:
|
|
return [
|
|
Comment(self._client, m)
|
|
for m in get_paginated_collection(
|
|
self._client.api_client.comments_api.list_endpoint, issue_id=self.id
|
|
)
|
|
]
|
|
|
|
|
|
class IssuesRepo(
|
|
_IssueRepoBase,
|
|
ModelListMixin[Issue],
|
|
ModelCreateMixin[Issue, models.IIssueWriteRequest],
|
|
ModelRetrieveMixin[Issue],
|
|
):
|
|
_entity_type = Issue
|