Source code for q2_cores.PCS.core

import logging
from typing import Optional

from q2_sdk.hq.models.hq_credentials import HqCredentials
from q2_sdk.models.cores.base_core import BaseCore
from q2_sdk.models.cores.models.core_user import CoreUser

from q2_cores.PCS.mappers import demographic_info
from q2_cores.PCS.queries import demographic_info_query




class Core(BaseCore):
    CONFIG_FILE_NAME = "PCS_Core"
    REQUIRED_CONFIGURATIONS = {"BANK_ID": "1"}

    def __init__(
        self,
        logger: logging.Logger,
        core_user: CoreUser,
        hq_credentials: Optional[HqCredentials] = None,
        **kwargs,
    ):
        super().__init__(logger, core_user, hq_credentials=hq_credentials, **kwargs)



    async def build_demographic_info(self):
        demographic_query = demographic_info_query.DemographicInfoQuery(
            self.logger, self.config.BANK_ID, self.configured_user.ssn
        )
        return demographic_info.DemographicInfoMapper(
            [demographic_query],
            hq_credentials=self.hq_credentials,
            zone_context=self.core_user,
        )