Source code for q2_cores.SOA.queries.address_query

import logging
from lxml import etree
from lxml.builder import E

from q2_sdk.models.cores.queries.base_query import BaseQuery
from q2_cores.SOA.queries.mock_responses import mock_address_response




class AddressQuery(BaseQuery):
    """Builds the xml payload for the initial search call for SOA"""

    def __init__(
        self, logger: logging.Logger, portfolio: str, group_name: str, inst_number: str
    ):
        self.portfolio = portfolio
        self.group_name = group_name
        self.inst_number = inst_number
        super().__init__(logger)



    def build(self):
        xml = etree.tostring(
            E.Request(
                E.Portfolio(self.portfolio),
                E.Sequence("1"),
                E.GroupName(self.group_name),
                E.InstNumber(self.inst_number),
                E.Type("Addresses"),
                Echo="true",
                TypeOfRequest="GetPortfolio",
            ),
            encoding="unicode",
        )
        return xml




    def mock_response(self):
        return mock_address_response()