GitHub/Anki
1
0
Fork 0
mirror of https://github.com/ankitects/anki.git synced 2025-09-23 00:12:25 -04:00
Code Issues Projects Releases Packages 1 Wiki Activity Actions

fids and template searches

Browse source
This commit is contained in:
Damien Elmes 2011-04-10 03:50:55 +09:00
parent a13c18cf4b
commit 94d4e319ae
2 changed files with 64 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
anki/find.py
View file

@ -9,7 +9,7 @@ SEARCH_TAG = 0
SEARCH_TYPE = 1
SEARCH_PHRASE = 2
SEARCH_FID = 3
SEARCH_CARD = 4
SEARCH_TEMPLATE = 4
SEARCH_DISTINCT = 5
SEARCH_FIELD = 6
SEARCH_FIELD_EXISTS = 7
@ -26,7 +26,9 @@ class Finder(object):
self.query = query
(q, args) = self.findCardsWhere()
#fidList = findCardsMatchingFilters(self.deck, filters)
query = "select id from cards where " + q
query = "select id from cards"
if q:
query += " where " + q
# if cmquery['pos'] or cmquery['neg']:
# if hasWhere is False:
# query += " where "
@ -61,31 +63,11 @@ class Finder(object):
if type == SEARCH_TAG:
self._findTag(token, isNeg, c)
elif type == SEARCH_TYPE:
self._findCardState(token, isNeg, c)
self._findCardState(token, isNeg)
elif type == SEARCH_FID:
if fidquery:
if isNeg:
fidquery += " except "
else:
fidquery += " intersect "
elif isNeg:
fidquery += "select id from cards except "
fidquery += "select id from cards where fid in (%s)" % token
elif type == SEARCH_CARD:
print "search_card broken"
token = token.replace("*", "%")
ids = deck.db.list("""
select id from tags where name like :tag escape '\\'""", tag=token)
if isNeg:
if cmquery['neg']:
cmquery['neg'] += " intersect "
cmquery['neg'] += """
select cardId from cardTags where src = 2 and cardTags.tagId in %s""" % ids2str(ids)
else:
if cmquery['pos']:
cmquery['pos'] += " intersect "
cmquery['pos'] += """
select cardId from cardTags where src = 2 and cardTags.tagId in %s""" % ids2str(ids)
self._findFids(token)
elif type == SEARCH_TEMPLATE:
self._findTemplate(token, isNeg)
elif type == SEARCH_FIELD or type == SEARCH_FIELD_EXISTS:
field = value = ''
if type == SEARCH_FIELD:
@ -152,7 +134,8 @@ class Finder(object):
elif val == "suspended":
self.lims['card'].append("queue = -1")
elif val == "due":
self.lims['card'].append("(queue = 2 and due <= %d)" % deck.sched.today)
self.lims['card'].append("(queue = 2 and due <= %d)" %
self.deck.sched.today)
def _findText(self, val, neg, c):
val = val.replace("*", "%")
@ -161,6 +144,24 @@ class Finder(object):
self.lims['fact'].append("flds %s like :_text_%d escape '\\'" % (
extra, c))
def _findFids(self, val):
self.lims['fact'].append("id in (%s)" % val)
def _findTemplate(self, val, isNeg):
lims = []
comp = "!=" if isNeg else "="
found = False
for m in self.deck.models().values():
for t in m.templates:
if t['name'].lower() == val.lower():
self.lims['card'].append((
"(fid in (select id from facts where mid = %d) "
"and ord %s %d)") % (m.id, comp, t['ord']))
found = True
if not found:
# no such templates exist; artificially limit query
self.lims['card'].append("ord = -1")
def findCardsWhere(self):
self._findLimits()
x = []
@ -271,7 +272,7 @@ class Finder(object):
type = SEARCH_FID
elif token['value'].startswith("card:"):
token['value'] = token['value'][5:]
type = SEARCH_CARD
type = SEARCH_TEMPLATE
elif token['value'].startswith("show:"):
token['value'] = token['value'][5:].lower()
type = SEARCH_DISTINCT

43
tests/test_find.py
View file

@ -18,23 +18,50 @@ def test_findCards():
f['Front'] = u'cat'
f['Back'] = u'sheep'
deck.addFact(f)
f = deck.newFact()
f['Front'] = u'template test'
f['Back'] = u'foo bar'
f.model().templates[1]['actv'] = True
deck.addFact(f)
# tag searches
assert not deck.findCards("tag:donkey")
assert len(deck.findCards("tag:sheep")) == 1
assert len(deck.findCards("tag:sheep tag:goat")) == 1
assert len(deck.findCards("tag:sheep tag:monkey")) == 0
assert len(deck.findCards("tag:monkey")) == 1
assert len(deck.findCards("tag:sheep -tag:monkey")) == 1
assert len(deck.findCards("-tag:sheep")) == 2
assert len(deck.findCards("-tag:sheep")) == 4
deck.addTags(deck.db.list("select id from cards"), "foo bar")
assert (len(deck.findCards("tag:foo")) ==
len(deck.findCards("tag:bar")) ==
5)
deck.delTags(deck.db.list("select id from cards"), "foo")
assert len(deck.findCards("tag:foo")) == 0
assert len(deck.findCards("tag:bar")) == 5
# text searches
assert len(deck.findCards("cat")) == 2
assert len(deck.findCards("cat -dog")) == 1
assert len(deck.findCards("cat -dog")) == 1
assert len(deck.findCards("are goats")) == 1
assert len(deck.findCards('"are goats"')) == 0
assert len(deck.findCards('"goats are"')) == 1
deck.addTags(deck.db.list("select id from cards"), "foo bar")
assert (len(deck.findCards("tag:foo")) ==
len(deck.findCards("tag:bar")) ==
3)
deck.delTags(deck.db.list("select id from cards"), "foo")
assert len(deck.findCards("tag:foo")) == 0
assert len(deck.findCards("tag:bar")) == 3
# card states
c = f.cards()[0]
c.type = 2
assert deck.findCards("is:rev") == []
c.flush()
assert deck.findCards("is:rev") == [c.id]
assert deck.findCards("is:due") == []
c.due = 0; c.queue = 2
c.flush()
assert deck.findCards("is:due") == [c.id]
c.queue = -1
c.flush()
assert deck.findCards("is:suspended") == [c.id]
# fids
assert deck.findCards("fid:54321") == []
assert len(deck.findCards("fid:%d"%f.id)) == 2
assert len(deck.findCards("fid:3,2")) == 2
# templates
assert len(deck.findCards("card:foo")) == 0
assert len(deck.findCards("card:forward")) == 4