add fix for no whitespace in source or ref

datacoves · Apr 18, 2022 · e28812c · e28812c
1 parent cc144b9
commit e28812c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pre_commit_dbt/check_script_has_no_table_name.py b/pre_commit_dbt/check_script_has_no_table_name.py
@@ -41,12 +41,17 @@ def add_space_to_parenthesis(sql: str) -> str:
     return re.sub(REGEX_PARENTHESIS, r" \1 ", sql)
 
 
+def add_space_to_source_ref(sql: str) -> str:
+    return sql.replace("{{", "{{ ").replace("}}", " }}")
+
+
 def has_table_name(
     sql: str, filename: str, dotless: Optional[bool] = False
 ) -> Tuple[int, Set[str]]:
     status_code = 0
     sql_clean = replace_comments(sql)
     sql_clean = add_space_to_parenthesis(sql_clean)
+    sql_clean = add_space_to_source_ref(sql_clean)
     sql_split = re.split(REGEX_SPLIT, sql_clean)
     tables = set()
     cte = set()