from argparse import _SubParsersAction
from functools import partial
from lxml.etree import tostring
from lxml.objectify import E
from q2_sdk.core.dynamic_imports import (
api_ExecuteStoredProcedure as ExecuteStoredProcedure,
)
from q2_sdk.hq.models.hq_params.stored_procedure import ParamsBuilder
from q2_sdk.hq.table_row import TableRow
from .db_object import DbObject
D_TYPES = ExecuteStoredProcedure.DataType
[docs]
class SubsidiaryRow(TableRow):
UserID: int
SubsidiaryID: int
CustomerID: int
DisplayName: str
AchName: str
AchTaxId: int
WireName: str
AddressID: int
CreatedAt: str
CreatedByUserLogonID: int
CreatedbyAdminUserLogonID: int
[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
[docs]
async def get_by_customer_ids(
self,
customer_ids: list[int],
) -> list[SubsidiaryRow]:
"""
Get subsidiaries for the specified customer IDs.
:param customer_ids: list of Q2_Customer.CustomerID to query
"""
customer_nodes = [E.customer(id=str(cid)) for cid in customer_ids]
request_xml = E.root(*customer_nodes)
request = tostring(request_xml, encoding="utf-8")
params = (
ParamsBuilder().add_param(D_TYPES.Xml, "request", request.decode()).build()
)
return await self.call_hq(
"sdk_GetSubsidiariesByCustomerIds",
ExecuteStoredProcedure.SqlParameters(params),
)