BUG-59, changes to update README on how to add model level hooks (#64)

README.md

@@ -192,6 +192,12 @@ vars:
       - "{{ dbt_snow_mask.apply_masking_policy('snapshots') }}"
   ```
 
+- If you prefer to add this directly to a dbt model, see below command for an correct example
+
+  ```yaml
+  {{ config(post_hook = "{{ dbt_snow_mask.apply_masking_policy('models') }}") }}
+  ```
+
 # How to remove masking policy ?
 
 - Remove the masking policy applied by this package by running below commands  

integration_tests/dbt_project.yml

@@ -35,7 +35,7 @@ models:
   pre-hook:
     - "{{ dbt_snow_mask.create_masking_policy('models')}}"
   post-hook: 
-    - "{{ dbt_snow_mask.apply_masking_policy('models') }}"
+#    - "{{ dbt_snow_mask.apply_masking_policy('models') }}"
 #    - "{{ dbt_snow_mask.unapply_masking_policy('models') }}"
 
 snapshots:

integration_tests/macros/snow-mask-ddl/create_masking_policy_mp_ip.sql

@@ -0,0 +1,10 @@
+{% macro create_masking_policy_mp_ip(node_database,node_schema) %}
+
+CREATE MASKING POLICY IF NOT EXISTS {{node_database}}.{{node_schema}}.mp_ip AS (val string) 
+  RETURNS string ->
+      CASE WHEN CURRENT_ROLE() IN ('SYSADMIN') THEN val 
+           WHEN CURRENT_ROLE() IN ('DEVELOPER') THEN SPLIT(val,'.')[3]::VARCHAR
+      ELSE '**********'
+      END
+
+{% endmacro %}

integration_tests/models/staging/pii/stg_contacts.sql

@@ -1,3 +1,5 @@
+{{ config(post_hook = "{{ dbt_snow_mask.apply_masking_policy('models') }}") }}
+
 WITH contacts AS (
     SELECT
         contact_id,

integration_tests/models/staging/pii/stg_contacts.yml

@@ -12,6 +12,7 @@ models:
           masking_policy: mp_conditional_contacts_pii
           mp_conditional_columns: 
             - last_name 
+
       - name: last_name
         description: ""
       - name: email
@@ -23,6 +24,8 @@ models:
 
       - name: ip_address
         description: ""
+        meta:
+          masking_policy: mp_ip
 
       - name: ssn
         description: ""