Source code for q2_sdk.hq.db.wire_international

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 WireInternationalRow(TableRow): WireInternationalID: int WireTransferID: int TransactionID: int SwiftCodeOrBicMasked: str IBAN: str IntermedSwiftOrBicMasked: str IntermedFiIbanMasked: str RecipientFiName: 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 WireInternational(DbObject): # GET_BY_NAME_KEY = "column in the db response" NAME = "WireInternational" REPRESENTATION_ROW_CLASS = WireInternationalRow
[docs] def add_arguments(self, parser: _SubParsersAction): subparser = parser.add_parser("get_wire_international") 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[WireInternationalRow]: 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_GetWireInternational", ExecuteStoredProcedure.SqlParameters([ ExecuteStoredProcedure.SqlParam( ExecuteStoredProcedure.DataType.Int, "transaction_id", transaction_id, ) ]), ) if serialize_for_cli: columns = [ "WireInternationalID", "WireTransferID", "TransactionID", "RecipientFiName", "RecipientFiCountryID", "CustomerID", "UserID", "TransactionStatusID", "TransactionTypeID", "OriginatingAccountID", "TransactionAmount", "ToAccount", ] response = self.serialize_for_cli(response, columns) return response