More work on CS TB

This commit is contained in:
Kovid Goyal 2016-02-04 12:13:35 +05:30
parent c1cc8332f2
commit 2b7178a08f
2 changed files with 67 additions and 7 deletions

View File

@ -65,7 +65,7 @@ _include_fields = frozenset(Tag.__slots__) - frozenset({
def category_as_json(items, category, display_name, count, tooltip=None, parent=None,
is_editable=True, is_gst=False, is_hierarchical=False, is_searchable=True,
is_user_category=False):
is_user_category=False, is_first_letter=False):
ans = {'category': category, 'name': display_name, 'is_category':True, 'count':count}
if tooltip:
ans['tooltip'] = tooltip
@ -81,6 +81,8 @@ def category_as_json(items, category, display_name, count, tooltip=None, parent=
ans['is_searchable'] = True
if is_user_category:
ans['is_user_category'] = True
if is_first_letter:
ans['is_first_letter'] = True
item_id = 'c' + str(len(items))
items[item_id] = ans
return item_id
@ -93,11 +95,13 @@ def category_item_as_json(x, clear_rating=False):
if k == 'original_categories':
val = tuple(val)
ans[k] = val.copy() if isinstance(val, set) else val
s = ans.get('sort')
if x.use_sort_as_name:
ans['name'] = ans['sort']
ans['name'] = s
if x.original_name != ans['name']:
ans['original_name'] = x.original_name
ans.pop('sort', None)
if x.use_sort_as_name or not s or s == ans['name']:
ans.pop('sort', None)
if clear_rating:
del ans['avg_rating']
return ans
@ -300,7 +304,7 @@ def collapse_first_letter(collapse_nodes, items, category_node, cl_list, idx, is
node_id = category_as_json(
items, items[category_node['id']]['category'], collapse_letter, 0,
parent=category_node['id'], is_editable=False, is_gst=is_gst,
is_hierarchical=category_is_hierarchical)
is_hierarchical=category_is_hierarchical, is_first_letter=True)
node_parent = {'id':node_id, 'children':[]}
category_node['children'].append(node_parent)
collapse_nodes.append(node_parent)

View File

@ -164,8 +164,63 @@ class SearchPanel:
div.firstChild.addEventListener('click', click_handler(self.node_clicked, i))
container.appendChild(div)
def search_expression_for_item(self, node):
pass
def search_expression_for_item(self, node, state):
item = node.data
if item.is_searchable is False or not state or state == 'clear':
return ''
search_state = {'plus':'true', 'plusplus':'.true', 'minus':'false', 'minusminus':'.false'}[state]
if item.is_category:
category = item.category
if item.is_first_letter:
letters_seen = {}
for child in node.children:
if child.data.sort:
letters_seen[child.data.sort[0]] = True
letters_seen = Object.keys(letters_seen)
if letters_seen.length:
charclass = letters_seen.join('')
if category == 'authors':
expr = str.format(r'author_sort:"~(^[{0}])|(&\s*[{0}])"', charclass)
elif category == 'series':
expr = str.format(r'series_sort:"~^[{0}]"', charclass)
else:
expr = str.format(r'{0}:"~^[{1}]"', category, charclass)
else:
expr = str.format('{}:false', category)
elif category == 'news':
expr = str.format('tags:"={}"', item.name)
else:
return str.format('{}:{}', category, search_state)
if 'false' in search_state:
expr = '(not ' + expr + ')'
return expr
category = 'tags' if item.category == 'news' else item.category
if item.name and item.name[0] == '★':
# Assume ratings
expr = str.format('{}:{}', category, item.name.length)
else:
fm = self.interface_data.field_metadata[item.category]
suffix = ':' if fm and fm.is_csp else ''
name = item.name or item.sort
if not name:
return ''
name = str.replace(name, '"', r'\"')
if name[0] == '.':
name = '.' + name
if search_state == 'plusplus' or search_state == 'minusminus':
name = '.' + name
expr = str.format('{}:"={}{}"', category, name, suffix)
if 'false' in search_state:
expr = '(not ' + expr + ')'
return expr
def node_clicked(self, i):
node = self.node_for_path().children[i]
@ -173,7 +228,8 @@ class SearchPanel:
self.tag_path.append(i)
self.render_tag_browser()
else:
self.execute_search(self.search_expression_for_item(node))
expr = self.search_expression_for_item(node, 'plus')
self.execute_search(expr)
@property
def container(self):