anonymization dbt macro

anonymize.sql

{%  macro anonymize(column_name, function, add_alias=True) %}
   {% if execute %}
       {% call statement('salt', fetch_result=True) %}
           select salt from sources.private.salt limit 1;
       {% endcall %}
       {% set salt = load_result('salt')['data'][0][0] %}

       {% if function == 'full_hash' %}
           SHA256(concat({{ column_name }},'{{ salt }}'))
       {% endif %}

       {% if function == 'hash_email' %}
           SHA256(concat(split({{ column_name }},'@')[0],'{{ salt }}')) || '@' || split({{ column_name }},'@')[1]
       {% endif %}

       {% if add_alias %}
           as {{ column_name }}
       {% endif %}
   {% endif %}
{% endmacro %}