Source code for q2_cores.Symitar.queries.demographic_info_query

import logging

from typing import Optional
from q2_cores.Symitar.queries.base_query import SymitarBaseQuery
from q2_cores.Symitar.queries import mock_responses




class DemographicInfoQuery(SymitarBaseQuery):
    """Returns demographic info for a Symitar Entity"""

    def __init__(
        self,
        logger: logging.Logger,
        customer_id,
        unit_number,
        device_type,
        card_prefix,
        guid: Optional[str] = None,
        use_symxchange=False,
        set_member_number_as_demo_cif=False,
        strip_zeroes_from_cif=False,
    ):
        super().__init__(
            logger,
            customer_id,
            unit_number,
            device_type,
            card_prefix,
            "Q2",
            "GETALLPROFILES",
            rg_params={"JRGUSERNUM1": 0},
            guid=guid,
            use_symxchange=use_symxchange,
        )
        self.set_member_number_as_demo_cif = set_member_number_as_demo_cif
        self.strip_zeroes_from_cif = strip_zeroes_from_cif



    def mock_response(self):
        return mock_responses.mock_demo_info_response()