Source code for q2_sdk.hq.db.notification_type

from argparse import _SubParsersAction
from functools import partial

from .db_object import DbObject




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



    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):
        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