from argparse import _SubParsersAction
from functools import partial
from q2_sdk.core.dynamic_imports import (
api_ExecuteStoredProcedure as ExecuteStoredProcedure,
)
from q2_sdk.hq.table_row import TableRow
from .db_object import DbObject
[docs]
class ServiceChargePlanRow(TableRow):
ServiceChargePlanID: int
PlanName: str
Description: str
MonthsFree: int
BaseFee: float
MaxUsageFee: float
MaxTxnFee: float
SumUsageAndTxnFees: bool
NoMaxUsageFee: bool
NoMaxTxnFee: bool
[docs]
class ServiceChargePlan(DbObject):
GET_BY_NAME_KEY = "PlanName"
NAME = "ServiceChargePlan"
REPRESENTATION_ROW_CLASS = ServiceChargePlanRow
[docs]
def add_arguments(self, parser: _SubParsersAction):
subparser = parser.add_parser("get_service_charge_plan")
subparser.set_defaults(parser="get")
subparser.set_defaults(func=partial(self.get, serialize_for_cli=True))
[docs]
async def get(self, serialize_for_cli=False) -> list[ServiceChargePlanRow]:
response = await self.call_hq(
"sdk_GetServiceChargePlan",
ExecuteStoredProcedure.SqlParameters([]),
)
if serialize_for_cli:
columns = ["ServiceChargePlanID", "PlanName", "Description"]
response = self.serialize_for_cli(response, columns)
return response