PowerOn/service-preprocessing
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: sanitize SQL queries by removing trailing semicolons

Browse source
This commit is contained in:
Christopher Gondek 2025-11-05 11:23:54 +01:00
parent dde61f447d
commit 3fbb41b980
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/dataquery/service.py
View file

@ -86,6 +86,9 @@ class DataQueryService:
""" """
max_limit = settings.SQL_ROW_LIMIT max_limit = settings.SQL_ROW_LIMIT
# Strip trailing semicolons and whitespace to prevent multi-statement errors
query = query.rstrip("; \t\n\r")
# Remove comments and normalize whitespace for parsing # Remove comments and normalize whitespace for parsing
cleaned_query = re.sub(r"--.*$", "", query, flags=re.MULTILINE) cleaned_query = re.sub(r"--.*$", "", query, flags=re.MULTILINE)
cleaned_query = re.sub(r"/\*.*?\*/", "", cleaned_query, flags=re.DOTALL) cleaned_query = re.sub(r"/\*.*?\*/", "", cleaned_query, flags=re.DOTALL)