test

def get_normalize_zip_code_for_create(model, values, zip_field, country_field=False):
    zip_code = values.get(zip_field, False)
    if zip_code:
        if isinstance(country_field, str):
            country_value = values.get(country_field, False)
            country_id = model.env['res.country'].browse(country_value)
        else:
            country_id = country_field
        country_code = country_id and country_id.code or 'CA'
        values[zip_field] = normalize_zip_code(model, zip_code, country_code)
    return values


def get_normalize_zip_code_for_write(model, record, values, zip_field, country_field=False):
    # zip_value = values.get(zip_field, False)
    # country_value = values.get(country_field, False)
    # if zip_value or country_value:
    #     country_id = model.env['res.country'].browse(country_value) if country_value else getattr(record, country_field)
    #     country_code = country_id and country_id.code or 'CA'
    #     zip_code = zip_value or getattr(record, zip_field)
    #     values[zip_field] = normalize_zip_code(model, zip_code, country_code)

    country_code = 'CA'
    if country_field:
        if isinstance(country_field, str):
            country_value = values.get(country_field, False)
            country_id = model.env['res.country'].browse(country_value) if country_value else getattr(record,
                                                                                                      country_field)
        else:
            country_id = country_field
        country_code = country_id and country_id.code or 'CA'
    zip_code = values.get(country_field, False) or getattr(record, zip_field)
    values[zip_field] = normalize_zip_code(model, zip_code, country_code)

    return values