from argparse import _SubParsersAction
from functools import partial
from typing import List
from lxml.objectify import IntElement, StringElement
from q2_sdk.core.dynamic_imports import (
api_ExecuteStoredProcedure as ExecuteStoredProcedure,
)
from .db_object import DbObject
from .representation_row_base import RepresentationRowBase
[docs]
class SubsidiaryRow(RepresentationRowBase):
UserID: IntElement = "UserID"
SubsidiaryID: IntElement = "SubsidiaryID"
CustomerID: IntElement = "CustomerID"
DisplayName: StringElement = "DisplayName"
AchName: StringElement = "AchName"
AchTaxId: IntElement = "AchTaxId"
WireName: StringElement = "WireName"
AddressID: IntElement = "AddressID"
CreatedAt: StringElement = "CreatedAt"
CreatedByUserLogonID: IntElement = "CreatedByUserLogonID"
CreatedbyAdminUserLogonID: IntElement = "CreatedbyAdminUserLogonID"
[docs]
class Subsidiary(DbObject):
GET_BY_NAME_KEY = "DisplayName"
NAME = "Subsidiary"
REPRESENTATION_ROW_CLASS = SubsidiaryRow
[docs]
def add_arguments(self, parser: _SubParsersAction):
subparser = parser.add_parser("get_subsidiaries")
subparser.set_defaults(parser="get")
subparser.set_defaults(func=partial(self.get, serialize_for_cli=True))
subparser.add_argument("customer_id", help="Q2_Customer.CustomerID")
[docs]
async def get(
self, customer_id: int, serialize_for_cli=False
) -> List[SubsidiaryRow]:
response = await self.call_hq(
"sdk_GetSubsidiary",
ExecuteStoredProcedure.SqlParameters([
ExecuteStoredProcedure.SqlParam(
ExecuteStoredProcedure.DataType.Int, "customer_id", customer_id
)
]),
)
if serialize_for_cli:
columns = ["SubsidiaryID", "CustomerID", "DisplayName"]
response = self.serialize_for_cli(response, columns)
return response