ce243c2cae
The existing code was really difficult to reason about: - The default notetype depended on the selected deck, and vice versa, and this logic was buried in the deck and notetype choosing screens, and models.py. - Changes to the notetype were not passed back directly, but were fired via a hook, which changed any screen in the app that had a notetype selector. It also wasn't great for performance, as the most recent deck and tags were embedded in the notetype, which can be expensive to save and sync for large notetypes. To address these points: - The current deck for a notetype, and notetype for a deck, are now stored in separate config variables, instead of directly in the deck or notetype. These are cheap to read and write, and we'll be able to sync them individually in the future once config syncing is updated in the future. I seem to recall some users not wanting the tag saving behaviour, so I've dropped that for now, but if people end up missing it, it would be simple to add as an extra auxiliary config variable. - The logic for getting the starting deck and notetype has been moved into the backend. It should be the same as the older Python code, with one exception: when "change deck depending on notetype" is enabled in the preferences, it will start with the current notetype ("curModel"), instead of first trying to get a deck-specific notetype. - ModelChooser has been duplicated into notetypechooser.py, and it has been updated to solely be concerned with keeping track of a selected notetype - it no longer alters global state.
172 lines
4.5 KiB
Python
172 lines
4.5 KiB
Python
# coding: utf-8
|
|
|
|
import os
|
|
import tempfile
|
|
|
|
from anki import Collection as aopen
|
|
from anki.dbproxy import emulate_named_args
|
|
from anki.lang import TR, without_unicode_isolation
|
|
from anki.stdmodels import addBasicModel, get_stock_notetypes
|
|
from anki.utils import isWin
|
|
from tests.shared import assertException, getEmptyCol
|
|
|
|
|
|
def test_create_open():
|
|
(fd, path) = tempfile.mkstemp(suffix=".anki2", prefix="test_attachNew")
|
|
try:
|
|
os.close(fd)
|
|
os.unlink(path)
|
|
except OSError:
|
|
pass
|
|
col = aopen(path)
|
|
# for open()
|
|
newPath = col.path
|
|
newMod = col.mod
|
|
col.close()
|
|
del col
|
|
|
|
# reopen
|
|
col = aopen(newPath)
|
|
assert col.mod == newMod
|
|
col.close()
|
|
|
|
# non-writeable dir
|
|
if isWin:
|
|
dir = "c:\root.anki2"
|
|
else:
|
|
dir = "/attachroot.anki2"
|
|
assertException(Exception, lambda: aopen(dir))
|
|
# reuse tmp file from before, test non-writeable file
|
|
os.chmod(newPath, 0)
|
|
assertException(Exception, lambda: aopen(newPath))
|
|
os.chmod(newPath, 0o666)
|
|
os.unlink(newPath)
|
|
|
|
|
|
def test_noteAddDelete():
|
|
col = getEmptyCol()
|
|
# add a note
|
|
note = col.newNote()
|
|
note["Front"] = "one"
|
|
note["Back"] = "two"
|
|
n = col.addNote(note)
|
|
assert n == 1
|
|
# test multiple cards - add another template
|
|
m = col.models.current()
|
|
mm = col.models
|
|
t = mm.newTemplate("Reverse")
|
|
t["qfmt"] = "{{Back}}"
|
|
t["afmt"] = "{{Front}}"
|
|
mm.addTemplate(m, t)
|
|
mm.save(m)
|
|
assert col.cardCount() == 2
|
|
# creating new notes should use both cards
|
|
note = col.newNote()
|
|
note["Front"] = "three"
|
|
note["Back"] = "four"
|
|
n = col.addNote(note)
|
|
assert n == 2
|
|
assert col.cardCount() == 4
|
|
# check q/a generation
|
|
c0 = note.cards()[0]
|
|
assert "three" in c0.q()
|
|
# it should not be a duplicate
|
|
assert not note.duplicate_or_empty()
|
|
# now let's make a duplicate
|
|
note2 = col.newNote()
|
|
note2["Front"] = "one"
|
|
note2["Back"] = ""
|
|
assert note2.duplicate_or_empty()
|
|
# empty first field should not be permitted either
|
|
note2["Front"] = " "
|
|
assert note2.duplicate_or_empty()
|
|
|
|
|
|
def test_fieldChecksum():
|
|
col = getEmptyCol()
|
|
note = col.newNote()
|
|
note["Front"] = "new"
|
|
note["Back"] = "new2"
|
|
col.addNote(note)
|
|
assert col.db.scalar("select csum from notes") == int("c2a6b03f", 16)
|
|
# changing the val should change the checksum
|
|
note["Front"] = "newx"
|
|
note.flush()
|
|
assert col.db.scalar("select csum from notes") == int("302811ae", 16)
|
|
|
|
|
|
def test_addDelTags():
|
|
col = getEmptyCol()
|
|
note = col.newNote()
|
|
note["Front"] = "1"
|
|
col.addNote(note)
|
|
note2 = col.newNote()
|
|
note2["Front"] = "2"
|
|
col.addNote(note2)
|
|
# adding for a given id
|
|
col.tags.bulk_add([note.id], "foo")
|
|
note.load()
|
|
note2.load()
|
|
assert "foo" in note.tags
|
|
assert "foo" not in note2.tags
|
|
# should be canonified
|
|
col.tags.bulk_add([note.id], "foo aaa")
|
|
note.load()
|
|
assert note.tags[0] == "aaa"
|
|
assert len(note.tags) == 2
|
|
|
|
|
|
def test_timestamps():
|
|
col = getEmptyCol()
|
|
assert len(col.models.all_names_and_ids()) == len(get_stock_notetypes(col))
|
|
for i in range(100):
|
|
addBasicModel(col)
|
|
assert len(col.models.all_names_and_ids()) == 100 + len(get_stock_notetypes(col))
|
|
|
|
|
|
def test_furigana():
|
|
col = getEmptyCol()
|
|
mm = col.models
|
|
m = mm.current()
|
|
# filter should work
|
|
m["tmpls"][0]["qfmt"] = "{{kana:Front}}"
|
|
mm.save(m)
|
|
n = col.newNote()
|
|
n["Front"] = "foo[abc]"
|
|
col.addNote(n)
|
|
c = n.cards()[0]
|
|
assert c.q().endswith("abc")
|
|
# and should avoid sound
|
|
n["Front"] = "foo[sound:abc.mp3]"
|
|
n.flush()
|
|
assert "anki:play" in c.q(reload=True)
|
|
# it shouldn't throw an error while people are editing
|
|
m["tmpls"][0]["qfmt"] = "{{kana:}}"
|
|
mm.save(m)
|
|
c.q(reload=True)
|
|
|
|
|
|
def test_translate():
|
|
col = getEmptyCol()
|
|
no_uni = without_unicode_isolation
|
|
|
|
assert (
|
|
col.tr(TR.CARD_TEMPLATE_RENDERING_FRONT_SIDE_PROBLEM)
|
|
== "Front template has a problem:"
|
|
)
|
|
assert no_uni(col.tr(TR.STATISTICS_REVIEWS, reviews=1)) == "1 review"
|
|
assert no_uni(col.tr(TR.STATISTICS_REVIEWS, reviews=2)) == "2 reviews"
|
|
|
|
|
|
def test_db_named_args(capsys):
|
|
sql = "select a, 2+:test5 from b where arg =:foo and x = :test5"
|
|
args = []
|
|
kwargs = dict(test5=5, foo="blah")
|
|
|
|
s, a = emulate_named_args(sql, args, kwargs)
|
|
assert s == "select a, 2+?1 from b where arg =?2 and x = ?1"
|
|
assert a == [5, "blah"]
|
|
|
|
# swallow the warning
|
|
_ = capsys.readouterr()
|