Source code for q2_sdk.hq.db.address_type

from argparse import _SubParsersAction
from functools import partial
from typing import List

from q2_sdk.hq.table_row import TableRow

from .db_object import DbObject




class AddressTypeRow(TableRow):
    AddressTypeID: int
    AddressTypeName: str





class AddressType(DbObject):
    GET_BY_NAME_KEY = "AddressTypeName"
    NAME = "AddressType"
    REPRESENTATION_ROW_CLASS = AddressTypeRow



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




    async def get(self, serialize_for_cli=False) -> List[AddressTypeRow]:
        response = await self.call_hq("sdk_GetAddressType")

        if serialize_for_cli:
            columns = ["AddressTypeID", "AddressTypeName"]
            response = self.serialize_for_cli(response, columns)

        return response