Source code for q2_sdk.hq.db.notification_type

from argparse import _SubParsersAction
from functools import partial
from q2_sdk.hq.table_row import TableRow

from .db_object import DbObject




class NotificationTypeRow(TableRow):
    NotificationTypeID: int
    ShortName: str
    DisplayName: str





class NotificationType(DbObject):
    GET_BY_NAME_KEY = "ShortName"
    NAME = "NotificationType"

    REPRESENTATION_ROW_CLASS = NotificationTypeRow



    def add_arguments(self, parser: _SubParsersAction):
        subparser = parser.add_parser("get_notification_types")
        subparser.set_defaults(
            parser="get_notification_types",
            func=partial(self.get, serialize_for_cli=True),
        )




    async def get(self, serialize_for_cli=False) -> list[NotificationTypeRow]:
        response = await self.call_hq("sdk_GetNotificationTypes")
        if serialize_for_cli:
            response = self.serialize_for_cli(
                response,
                fields_to_display=["NotificationTypeID", "ShortName", "DisplayName"],
            )
        return response