def METHOD_NAME(self, country, mode, **kw):
    return dict(
        fields=country.get_address_fields(),
        states=[
            (st.id, st.name, st.code)
            for st in country.get_website_sale_states(mode=mode)
        ],
        phone_code=country.phone_code,
    )