Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Truncate ids in SQL IN clause #49

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Truncate ids in SQL IN clause #49

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
10 changes: 10 additions & 0 deletions devserver/modules/sql.py
View file
Open in desktop
Expand Up @@ -30,7 +30,17 @@ def format(text, *args, **kwargs):
import re import re
_sql_fields_re = re.compile(r'SELECT .*? FROM') _sql_fields_re = re.compile(r'SELECT .*? FROM')
_sql_aggregates_re = re.compile(r'SELECT .*?(COUNT|SUM|AVERAGE|MIN|MAX).*? FROM') _sql_aggregates_re = re.compile(r'SELECT .*?(COUNT|SUM|AVERAGE|MIN|MAX).*? FROM')
_sql_in_numbers_re = re.compile(r'IN\s*\(([\d\s,]+)\)')

def truncate_sql(sql, aggregates=True): def truncate_sql(sql, aggregates=True):
def truncate_number_list(match):
numbers = match.group(1).split(',')
result = [x.strip() for x in numbers[:5]]
if len(numbers) > len(result):
result.append('...')
return u'IN (%s)' % (u', '.join(result))
sql = _sql_in_numbers_re.sub(truncate_number_list, sql)

if not aggregates and _sql_aggregates_re.match(sql): if not aggregates and _sql_aggregates_re.match(sql):
return sql return sql
return _sql_fields_re.sub('SELECT ... FROM', sql) return _sql_fields_re.sub('SELECT ... FROM', sql)
Expand Down