"""Address util Document."""

from payee.connectors.secure_data.fields.text_field import TextField
from payee.utils.bank_details import BankDetailsDataHelper


class AddressUtilDocument:
    """Address util document."""

    fields = {
        'country_code': TextField('country_code', encrypt=True, required=True),
        'address_1': TextField('address_1', encrypt=True, required=True),
        'address_2': TextField('address_2', encrypt=True, required=True),
        'city': TextField('city', encrypt=True, required=True),
        'province': TextField('province', encrypt=True, required=False),
        'zip': TextField('zip', encrypt=True, required=False),
    }

    @staticmethod
    def build_response(secure_document_item, is_obscured: bool = False):
        """Format the response given the raw data."""

        values = secure_document_item.get_values(is_obscured)

        return BankDetailsDataHelper.build_payee_address(values)