Accords-Library-Archive
/
gallery.accords-library.com
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

server/search: fix searching for invalid numbers

This commit is contained in:
rr- 2016-06-02 20:48:42 +02:00
parent 1034362b84
commit a72f6aa585
2 changed files with 35 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
server/szurubooru/search/base_search_config.py
View File

@ -38,22 +38,26 @@ class BaseSearchConfig(object):
''' '''
Decorate SQLAlchemy filter on given column using supplied criterion. Decorate SQLAlchemy filter on given column using supplied criterion.
''' '''
if isinstance(criterion, criteria.PlainSearchCriterion): try:
expr = column == int(criterion.value) if isinstance(criterion, criteria.PlainSearchCriterion):
elif isinstance(criterion, criteria.ArraySearchCriterion): expr = column == int(criterion.value)
expr = column.in_(int(value) for value in criterion.values) elif isinstance(criterion, criteria.ArraySearchCriterion):
elif isinstance(criterion, criteria.RangedSearchCriterion): expr = column.in_(int(value) for value in criterion.values)
assert criterion.min_value != '' \ elif isinstance(criterion, criteria.RangedSearchCriterion):
or criterion.max_value != '' assert criterion.min_value != '' \
if criterion.min_value != '' and criterion.max_value != '': or criterion.max_value != ''
expr = column.between( if criterion.min_value != '' and criterion.max_value != '':
int(criterion.min_value), int(criterion.max_value)) expr = column.between(
elif criterion.min_value != '': int(criterion.min_value), int(criterion.max_value))
expr = column >= int(criterion.min_value) elif criterion.min_value != '':
elif criterion.max_value != '': expr = column >= int(criterion.min_value)
expr = column <= int(criterion.max_value) elif criterion.max_value != '':
else: expr = column <= int(criterion.max_value)
assert False else:
assert False
except ValueError as e:
raise errors.SearchError(
'Criterion value %r must be a number.' % (criterion,))
if criterion.negated: if criterion.negated:
expr = ~expr expr = ~expr
return expr return expr

15
server/szurubooru/tests/search/test_tag_search_config.py
View File

@ -134,7 +134,9 @@ def test_filter_by_edit_time(
('post-count:2', ['t1']), ('post-count:2', ['t1']),
('post-count:1', ['t2']), ('post-count:1', ['t2']),
('post-count:1..', ['t1', 't2']), ('post-count:1..', ['t1', 't2']),
('post-count-min:1', ['t1', 't2']),
('post-count:..1', ['t2']), ('post-count:..1', ['t2']),
('post-count-max:1', ['t2']),
('usage-count:2', ['t1']), ('usage-count:2', ['t1']),
('usage-count:1', ['t2']), ('usage-count:1', ['t2']),
('usages:2', ['t1']), ('usages:2', ['t1']),
@ -153,6 +155,19 @@ def test_filter_by_post_count(
post2.tags.append(tag1) post2.tags.append(tag1)
verify_unpaged(input, expected_tag_names) verify_unpaged(input, expected_tag_names)
@pytest.mark.parametrize('input', [
'post-count:..',
'post-count:asd',
'post-count:asd,1',
'post-count:1,asd',
'post-count:asd..1',
'post-count:1..asd',
])
def test_filter_by_invalid_input(executor, input):
with pytest.raises(errors.SearchError):
actual_count, actual_posts = executor.execute(
input, page=1, page_size=100)
@pytest.mark.parametrize('input,expected_tag_names', [ @pytest.mark.parametrize('input,expected_tag_names', [
('suggestion-count:2', ['t1']), ('suggestion-count:2', ['t1']),
('suggestion-count:1', ['t2']), ('suggestion-count:1', ['t2']),