def book_title_search(query, author_ids = None, max_length=5):
"""Searches 'query' in book.title field.
Filter by list of author id.
Returns list of SphinxSearch."""
result_list = []
# simple full-text search
query_set = Book.title_search.query(query)
# search_engine = SphinxSearchEngine()
# result = search_engine.book_search(max_length=max_length, title=query)
# filter by authors id
if author_ids:
query_set = query_set.filter(author_id=author_ids[0])
if len(author_ids) > 1:
# select book written by first author, using sphinx tool
authors = set( Author.objects.filter(id__in=author_ids[1:]) )
result_list = []
# if amount target author_ids there are all authors in database
if len(authors) == len(author_ids) - 1:
for book in query_set:
if authors.issubset(book.author.all()):
result_list.append(book)
# if we found enough books, stop filtering
if len(result_list) == max_length:
break
else:
result_list = query_set[0:max_length]
else:
result_list = query_set[0:max_length]
# calculating distance between words
for book in result_list:
book.rel = name_distance(query, book.title)
# sorting list by element relevant index
result_list.sort( lambda x, y: cmp(x.rel, y.rel) )
return result_list
def book_title_search(query, author_ids=None, max_length=5):
"""Searches 'query' in book.title field.
Filter by list of author id.
Returns list of SphinxSearch."""
result_list = []
# simple full-text search
query_set = Book.title_search.query(query)
# search_engine = SphinxSearchEngine()
# result = search_engine.book_search(max_length=max_length, title=query)
# filter by authors id
if author_ids:
query_set = query_set.filter(author_id=author_ids[0])
if len(author_ids) > 1:
# select book written by first author, using sphinx tool
authors = set(Author.objects.filter(id__in=author_ids[1:]))
result_list = []
# if amount target author_ids there are all authors in database
if len(authors) == len(author_ids) - 1:
for book in query_set:
if authors.issubset(book.author.all()):
result_list.append(book)
# if we found enough books, stop filtering
if len(result_list) == max_length:
break
else:
result_list = query_set[0:max_length]
else:
result_list = query_set[0:max_length]
# calculating distance between words
for book in result_list:
book.rel = name_distance(query, book.title)
# sorting list by element relevant index
result_list.sort(lambda x, y: cmp(x.rel, y.rel))
return result_list
def author_search(query, max_length=5):
"""Searches 'query' in author field.
Returns list of SphinxSearch."""
# # TODO search by author name and alias
#
# result_list = []
#
# if not query:
# return []
#
# # simple full-text search
# query_set = Author.simple_search.query(query)
# result_list = query_set[0:max_length]
#
# if query_set.count() < max_length:
# # if number of the results is less then max_length
# # than try to search using soundex
# soundex_query_set = \
# Author.soundex_search.query(query)[0:max_length]
#
# # remove found in simple search results from soundex search
# id_set = set( [a.id for a in result_list] )
# soundex_query_set = rm_items(soundex_query_set, id_set, id_field)
# result_list.extend(soundex_query_set)
search_engine = SphinxSearchEngine()
result = search_engine.author_search(max_length=max_length, author=query)
# calculating distance between words
for item in result:
item.rel = name_distance(query, item.name)
# sorting list by relevant index
result.sort( lambda x, y: cmp(x.rel, y.rel) )
return result[:max_length]
def author_search(query, max_length=5):
"""Searches 'query' in author field.
Returns list of SphinxSearch."""
# # TODO search by author name and alias
#
# result_list = []
#
# if not query:
# return []
#
# # simple full-text search
# query_set = Author.simple_search.query(query)
# result_list = query_set[0:max_length]
#
# if query_set.count() < max_length:
# # if number of the results is less then max_length
# # than try to search using soundex
# soundex_query_set = \
# Author.soundex_search.query(query)[0:max_length]
#
# # remove found in simple search results from soundex search
# id_set = set( [a.id for a in result_list] )
# soundex_query_set = rm_items(soundex_query_set, id_set, id_field)
# result_list.extend(soundex_query_set)
search_engine = SphinxSearchEngine()
result = search_engine.author_search(max_length=max_length, author=query)
# calculating distance between words
for item in result:
item.rel = name_distance(query, item.name)
# sorting list by relevant index
result.sort(lambda x, y: cmp(x.rel, y.rel))
return result[:max_length]
