feat(sanitize_column): improve check

Signed-off-by: Akhil Narang <me@akhilnarang.dev>
This commit is contained in:
Akhil Narang 2024-12-20 18:20:28 +05:30
parent 21a6d2a717
commit b5bad56cdd
No known key found for this signature in database
GPG key ID: 9DCC61E211BF645F

View file

@ -2142,6 +2142,11 @@ def sanitize_column(column_name: str) -> None:
elif any(keyword in column_name.split() for keyword in blacklisted_keywords):
_raise_exception()
elif any(
re.search(rf"\b{keyword}\b", column_name, re.IGNORECASE) for keyword in blacklisted_keywords
):
_raise_exception()
elif regex.match(column_name):
_raise_exception()