Source code for q2_sdk.hq.db.wire_domestic

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

from q2_sdk.core.dynamic_imports import (
    api_ExecuteStoredProcedure as ExecuteStoredProcedure,
)
from q2_sdk.core.exceptions import DatabaseDataError
from q2_sdk.hq.table_row import TableRow

from .db_object import DbObject
from .generated_transactions import GeneratedTransactions


[docs] class WireDomesticRow(TableRow): WireDomesticID: int WireTransferID: int TransactionID: int ABA: str RecipientFiName: str RecipientFiAddress1: str RecipientFiAddress2: str RecipientFiCity: str RecipientFiState: str RecipientFiPostalCode: str RecipientFiCountryID: int CustomerID: int UserID: int CreateDate: str TransactionStatusID: int TransactionTypeID: int ProcessDate: str ProcessedDate: str OriginatingAccountID: int TransactionAmount: float ToAccount: str ToAccountType: str PayerName: str PayerCountryID: int
[docs] class WireDomestic(DbObject): # GET_BY_NAME_KEY = "column in the db response" NAME = "WireDomestic" REPRESENTATION_ROW_CLASS = WireDomesticRow
[docs] def add_arguments(self, parser: _SubParsersAction): subparser = parser.add_parser("get_wire_domestic") subparser.set_defaults(parser="get") subparser.set_defaults(func=partial(self.get, serialize_for_cli=True)) subparser.add_argument( "transaction_id", help="Q2_GeneratedTransactions.TransactionID" )
[docs] async def get( self, transaction_id: int, serialize_for_cli=False ) -> List[WireDomesticRow]: transaction = GeneratedTransactions( self.logger, self.hq_credentials, ret_table_obj=True ) transaction_row = await transaction.get_by_id(transaction_id=transaction_id) if not transaction_row: raise DatabaseDataError( f"GeneratedTransactions with ID {transaction_id} does not exist" ) response = await self.call_hq( "sdk_GetWireDomestic", ExecuteStoredProcedure.SqlParameters([ ExecuteStoredProcedure.SqlParam( ExecuteStoredProcedure.DataType.Int, "transaction_id", transaction_id, ) ]), ) if serialize_for_cli: columns = [ "WireDomesticID", "WireTransferID", "TransactionID", "RecipientFiName", "RecipientFiCountryID", "CustomerID", "UserID", "TransactionStatusID", "TransactionTypeID", "OriginatingAccountID", "TransactionAmount", "ToAccount", ] response = self.serialize_for_cli(response, columns) return response