mirror of
https://github.com/ankitects/anki.git
synced 2025-09-20 15:02:21 -04:00
7afe6a9a7d
Rather than use a combination of id lookups on the groups table and a group configuration cache in the scheduler, I've moved the groups and group config into json objects on the deck table. This results in a net saving of code and saves one or more DB lookups on each card answer, in exchange for a small increase in deck load/save work. I did a quick survey of AnkiWeb, and the vast majority of decks use less than 100 tags, and it's safe to assume groups will follow a similar pattern. All groups and group configs except the default one will use integer timestamps now, to simplify merging when syncing and importing. defaultGroup() has been removed in favour of keeping the models up to date (not yet done).
197 lines
5.6 KiB
Python
197 lines
5.6 KiB
Python
# coding: utf-8
|
|
|
|
import os, re, datetime
|
|
from tests.shared import assertException, getEmptyDeck, testDir
|
|
from anki.stdmodels import BasicModel
|
|
|
|
from anki import Deck
|
|
|
|
newPath = None
|
|
newMod = None
|
|
|
|
def test_create():
|
|
global newPath, newMod
|
|
path = "/tmp/test_attachNew.anki"
|
|
try:
|
|
os.unlink(path)
|
|
except OSError:
|
|
pass
|
|
deck = Deck(path)
|
|
# for open()
|
|
newPath = deck.path
|
|
deck.save()
|
|
newMod = deck.mod
|
|
deck.close()
|
|
del deck
|
|
|
|
def test_open():
|
|
deck = Deck(newPath)
|
|
assert deck.mod == newMod
|
|
deck.close()
|
|
|
|
def test_openReadOnly():
|
|
# non-writeable dir
|
|
assertException(Exception,
|
|
lambda: Deck("/attachroot"))
|
|
# reuse tmp file from before, test non-writeable file
|
|
os.chmod(newPath, 0)
|
|
assertException(Exception,
|
|
lambda: Deck(newPath))
|
|
os.chmod(newPath, 0666)
|
|
os.unlink(newPath)
|
|
|
|
def test_factAddDelete():
|
|
deck = getEmptyDeck()
|
|
# add a fact
|
|
f = deck.newFact()
|
|
f['Front'] = u"one"; f['Back'] = u"two"
|
|
n = deck.addFact(f)
|
|
assert n == 1
|
|
deck.rollback()
|
|
assert deck.cardCount() == 0
|
|
# try with two cards
|
|
f = deck.newFact()
|
|
f['Front'] = u"one"; f['Back'] = u"two"
|
|
m = f.model()
|
|
m.templates[1]['actv'] = True
|
|
m.flush()
|
|
n = deck.addFact(f)
|
|
assert n == 2
|
|
# check q/a generation
|
|
c0 = f.cards()[0]
|
|
assert re.sub("</?.+?>", "", c0.q()) == u"one"
|
|
# it should not be a duplicate
|
|
for p in f.problems():
|
|
assert not p
|
|
# now let's make a duplicate and test uniqueness
|
|
f2 = deck.newFact()
|
|
f2.model().fields[1]['req'] = True
|
|
f2['Front'] = u"one"; f2['Back'] = u""
|
|
p = f2.problems()
|
|
assert p[0] == "unique"
|
|
assert p[1] == "required"
|
|
# try delete the first card
|
|
cards = f.cards()
|
|
id1 = cards[0].id; id2 = cards[1].id
|
|
assert deck.cardCount() == 2
|
|
assert deck.factCount() == 1
|
|
deck.delCards([id1])
|
|
assert deck.cardCount() == 1
|
|
assert deck.factCount() == 1
|
|
# and the second should clear the fact
|
|
deck.delCards([id2])
|
|
assert deck.cardCount() == 0
|
|
assert deck.factCount() == 0
|
|
|
|
def test_fieldChecksum():
|
|
deck = getEmptyDeck()
|
|
f = deck.newFact()
|
|
f['Front'] = u"new"; f['Back'] = u"new2"
|
|
deck.addFact(f)
|
|
assert deck.db.scalar(
|
|
"select csum from fsums") == int("22af645d", 16)
|
|
# empty field should have no checksum
|
|
f['Front'] = u""
|
|
f.flush()
|
|
assert deck.db.scalar(
|
|
"select count() from fsums") == 0
|
|
# changing the val should change the checksum
|
|
f['Front'] = u"newx"
|
|
f.flush()
|
|
assert deck.db.scalar(
|
|
"select csum from fsums") == int("4b0e5a4c", 16)
|
|
# turning off unique and modifying the fact should delete the sum
|
|
m = f.model()
|
|
m.fields[0]['uniq'] = False
|
|
m.flush()
|
|
f.flush()
|
|
assert deck.db.scalar(
|
|
"select count() from fsums") == 0
|
|
# and turning on both should ensure two checksums generated
|
|
m.fields[0]['uniq'] = True
|
|
m.fields[1]['uniq'] = True
|
|
m.flush()
|
|
f.flush()
|
|
assert deck.db.scalar(
|
|
"select count() from fsums") == 2
|
|
|
|
def test_upgrade():
|
|
import tempfile, shutil
|
|
src = os.path.join(testDir, "support", "anki12.anki")
|
|
(fd, dst) = tempfile.mkstemp(suffix=".anki")
|
|
print "upgrade to", dst
|
|
shutil.copy(src, dst)
|
|
deck = Deck(dst)
|
|
# creation time should have been adjusted
|
|
d = datetime.datetime.fromtimestamp(deck.crt)
|
|
assert d.hour == 4 and d.minute == 0
|
|
# 3 new, 2 failed, 1 due
|
|
assert deck.sched.counts() == (3,2,1)
|
|
# now's a good time to test the integrity check too
|
|
deck.fixIntegrity()
|
|
|
|
def test_groups():
|
|
deck = getEmptyDeck()
|
|
# we start with a standard group
|
|
assert len(deck.groups) == 1
|
|
# it should have an id of 1
|
|
assert deck.groups['1']
|
|
# create a new group
|
|
ts = deck.groupID("new group")
|
|
assert ts
|
|
assert len(deck.groups) == 2
|
|
# should get the same id
|
|
assert deck.groupID("new group") == ts
|
|
# by default, everything should be shown
|
|
assert not deck.qconf['groups']
|
|
|
|
def test_selective():
|
|
deck = getEmptyDeck()
|
|
f = deck.newFact()
|
|
f['Front'] = u"1"; f.tags = ["one", "three"]
|
|
deck.addFact(f)
|
|
f = deck.newFact()
|
|
f['Front'] = u"2"; f.tags = ["two", "three", "four"]
|
|
deck.addFact(f)
|
|
f = deck.newFact()
|
|
f['Front'] = u"3"; f.tags = ["one", "two", "three", "four"]
|
|
deck.addFact(f)
|
|
assert len(deck.selTagFids(["one"], [])) == 2
|
|
assert len(deck.selTagFids(["three"], [])) == 3
|
|
assert len(deck.selTagFids([], ["three"])) == 0
|
|
assert len(deck.selTagFids(["one"], ["three"])) == 0
|
|
assert len(deck.selTagFids(["one"], ["two"])) == 1
|
|
assert len(deck.selTagFids(["two", "three"], [])) == 3
|
|
assert len(deck.selTagFids(["two", "three"], ["one"])) == 1
|
|
assert len(deck.selTagFids(["one", "three"], ["two", "four"])) == 1
|
|
deck.setGroupForTags(["three"], [], 3)
|
|
assert deck.db.scalar("select count() from cards where gid = 3") == 3
|
|
deck.setGroupForTags(["one"], [], 2)
|
|
assert deck.db.scalar("select count() from cards where gid = 2") == 2
|
|
|
|
def test_addDelTags():
|
|
deck = getEmptyDeck()
|
|
f = deck.newFact()
|
|
f['Front'] = u"1"
|
|
deck.addFact(f)
|
|
f2 = deck.newFact()
|
|
f2['Front'] = u"2"
|
|
deck.addFact(f2)
|
|
# adding for a given id
|
|
deck.addTags([f.id], "foo")
|
|
f.load(); f2.load()
|
|
assert "foo" in f.tags
|
|
assert "foo" not in f2.tags
|
|
# should be canonified
|
|
deck.addTags([f.id], "foo aaa")
|
|
f.load()
|
|
assert f.tags[0] == "aaa"
|
|
assert len(f.tags) == 2
|
|
|
|
def test_timestamps():
|
|
deck = getEmptyDeck()
|
|
assert len(deck.models()) == 2
|
|
for i in range(100):
|
|
deck.addModel(BasicModel(deck))
|
|
assert len(deck.models()) == 102
|
|
|