mirror of
https://github.com/ankitects/anki.git
synced 2025-09-18 22:12:21 -04:00
b23b6fbe35
Earlier today I pushed a change that split this code up into multiple repos, but that has proved to complicate things too much. So we're back to a single repo, except the individual submodules are better separated than they were before. The README files need updating again; I will push them out soon. Aside from splitting out the different modules, the sound code has moved from from anki to aqt.
220 lines
6.9 KiB
Python
220 lines
6.9 KiB
Python
# coding: utf-8
|
|
|
|
import os
|
|
|
|
from anki.importing import (
|
|
Anki2Importer,
|
|
AnkiPackageImporter,
|
|
MnemosyneImporter,
|
|
SupermemoXmlImporter,
|
|
TextImporter,
|
|
)
|
|
from anki.utils import ids2str
|
|
from tests.shared import getEmptyCol, getUpgradeDeckPath
|
|
|
|
testDir = os.path.dirname(__file__)
|
|
|
|
srcNotes = None
|
|
srcCards = None
|
|
|
|
|
|
def test_anki2_mediadupes():
|
|
tmp = getEmptyCol()
|
|
# add a note that references a sound
|
|
n = tmp.newNote()
|
|
n["Front"] = "[sound:foo.mp3]"
|
|
mid = n.model()["id"]
|
|
tmp.addNote(n)
|
|
# add that sound to media folder
|
|
with open(os.path.join(tmp.media.dir(), "foo.mp3"), "w") as f:
|
|
f.write("foo")
|
|
tmp.close()
|
|
# it should be imported correctly into an empty deck
|
|
empty = getEmptyCol()
|
|
imp = Anki2Importer(empty, tmp.path)
|
|
imp.run()
|
|
assert os.listdir(empty.media.dir()) == ["foo.mp3"]
|
|
# and importing again will not duplicate, as the file content matches
|
|
empty.remCards(empty.db.list("select id from cards"))
|
|
imp = Anki2Importer(empty, tmp.path)
|
|
imp.run()
|
|
assert os.listdir(empty.media.dir()) == ["foo.mp3"]
|
|
n = empty.getNote(empty.db.scalar("select id from notes"))
|
|
assert "foo.mp3" in n.fields[0]
|
|
# if the local file content is different, and import should trigger a
|
|
# rename
|
|
empty.remCards(empty.db.list("select id from cards"))
|
|
with open(os.path.join(empty.media.dir(), "foo.mp3"), "w") as f:
|
|
f.write("bar")
|
|
imp = Anki2Importer(empty, tmp.path)
|
|
imp.run()
|
|
assert sorted(os.listdir(empty.media.dir())) == ["foo.mp3", "foo_%s.mp3" % mid]
|
|
n = empty.getNote(empty.db.scalar("select id from notes"))
|
|
assert "_" in n.fields[0]
|
|
# if the localized media file already exists, we rewrite the note and
|
|
# media
|
|
empty.remCards(empty.db.list("select id from cards"))
|
|
with open(os.path.join(empty.media.dir(), "foo.mp3"), "w") as f:
|
|
f.write("bar")
|
|
imp = Anki2Importer(empty, tmp.path)
|
|
imp.run()
|
|
assert sorted(os.listdir(empty.media.dir())) == ["foo.mp3", "foo_%s.mp3" % mid]
|
|
assert sorted(os.listdir(empty.media.dir())) == ["foo.mp3", "foo_%s.mp3" % mid]
|
|
n = empty.getNote(empty.db.scalar("select id from notes"))
|
|
assert "_" in n.fields[0]
|
|
|
|
|
|
def test_apkg():
|
|
tmp = getEmptyCol()
|
|
apkg = str(os.path.join(testDir, "support/media.apkg"))
|
|
imp = AnkiPackageImporter(tmp, apkg)
|
|
assert os.listdir(tmp.media.dir()) == []
|
|
imp.run()
|
|
assert os.listdir(tmp.media.dir()) == ["foo.wav"]
|
|
# importing again should be idempotent in terms of media
|
|
tmp.remCards(tmp.db.list("select id from cards"))
|
|
imp = AnkiPackageImporter(tmp, apkg)
|
|
imp.run()
|
|
assert os.listdir(tmp.media.dir()) == ["foo.wav"]
|
|
# but if the local file has different data, it will rename
|
|
tmp.remCards(tmp.db.list("select id from cards"))
|
|
with open(os.path.join(tmp.media.dir(), "foo.wav"), "w") as f:
|
|
f.write("xyz")
|
|
imp = AnkiPackageImporter(tmp, apkg)
|
|
imp.run()
|
|
assert len(os.listdir(tmp.media.dir())) == 2
|
|
|
|
|
|
def test_anki2_diffmodel_templates():
|
|
# different from the above as this one tests only the template text being
|
|
# changed, not the number of cards/fields
|
|
dst = getEmptyCol()
|
|
# import the first version of the model
|
|
tmp = getUpgradeDeckPath("diffmodeltemplates-1.apkg")
|
|
imp = AnkiPackageImporter(dst, tmp)
|
|
imp.dupeOnSchemaChange = True
|
|
imp.run()
|
|
# then the version with updated template
|
|
tmp = getUpgradeDeckPath("diffmodeltemplates-2.apkg")
|
|
imp = AnkiPackageImporter(dst, tmp)
|
|
imp.dupeOnSchemaChange = True
|
|
imp.run()
|
|
# collection should contain the note we imported
|
|
assert dst.noteCount() == 1
|
|
# the front template should contain the text added in the 2nd package
|
|
tcid = dst.findCards("")[0] # only 1 note in collection
|
|
tnote = dst.getCard(tcid).note()
|
|
assert "Changed Front Template" in dst.findTemplates(tnote)[0]["qfmt"]
|
|
|
|
|
|
def test_anki2_updates():
|
|
# create a new empty deck
|
|
dst = getEmptyCol()
|
|
tmp = getUpgradeDeckPath("update1.apkg")
|
|
imp = AnkiPackageImporter(dst, tmp)
|
|
imp.run()
|
|
assert imp.dupes == 0
|
|
assert imp.added == 1
|
|
assert imp.updated == 0
|
|
# importing again should be idempotent
|
|
imp = AnkiPackageImporter(dst, tmp)
|
|
imp.run()
|
|
assert imp.dupes == 1
|
|
assert imp.added == 0
|
|
assert imp.updated == 0
|
|
# importing a newer note should update
|
|
assert dst.noteCount() == 1
|
|
assert dst.db.scalar("select flds from notes").startswith("hello")
|
|
tmp = getUpgradeDeckPath("update2.apkg")
|
|
imp = AnkiPackageImporter(dst, tmp)
|
|
imp.run()
|
|
assert imp.dupes == 0
|
|
assert imp.added == 0
|
|
assert imp.updated == 1
|
|
assert dst.noteCount() == 1
|
|
assert dst.db.scalar("select flds from notes").startswith("goodbye")
|
|
|
|
|
|
def test_csv():
|
|
deck = getEmptyCol()
|
|
file = str(os.path.join(testDir, "support/text-2fields.txt"))
|
|
i = TextImporter(deck, file)
|
|
i.initMapping()
|
|
i.run()
|
|
# four problems - too many & too few fields, a missing front, and a
|
|
# duplicate entry
|
|
assert len(i.log) == 5
|
|
assert i.total == 5
|
|
# if we run the import again, it should update instead
|
|
i.run()
|
|
assert len(i.log) == 10
|
|
assert i.total == 5
|
|
# but importing should not clobber tags if they're unmapped
|
|
n = deck.getNote(deck.db.scalar("select id from notes"))
|
|
n.addTag("test")
|
|
n.flush()
|
|
i.run()
|
|
n.load()
|
|
assert n.tags == ["test"]
|
|
# if add-only mode, count will be 0
|
|
i.importMode = 1
|
|
i.run()
|
|
assert i.total == 0
|
|
# and if dupes mode, will reimport everything
|
|
assert deck.cardCount() == 5
|
|
i.importMode = 2
|
|
i.run()
|
|
# includes repeated field
|
|
assert i.total == 6
|
|
assert deck.cardCount() == 11
|
|
deck.close()
|
|
|
|
|
|
def test_csv2():
|
|
deck = getEmptyCol()
|
|
mm = deck.models
|
|
m = mm.current()
|
|
f = mm.newField("Three")
|
|
mm.addField(m, f)
|
|
mm.save(m)
|
|
n = deck.newNote()
|
|
n["Front"] = "1"
|
|
n["Back"] = "2"
|
|
n["Three"] = "3"
|
|
deck.addNote(n)
|
|
# an update with unmapped fields should not clobber those fields
|
|
file = str(os.path.join(testDir, "support/text-update.txt"))
|
|
i = TextImporter(deck, file)
|
|
i.initMapping()
|
|
i.run()
|
|
n.load()
|
|
assert n["Front"] == "1"
|
|
assert n["Back"] == "x"
|
|
assert n["Three"] == "3"
|
|
deck.close()
|
|
|
|
|
|
def test_supermemo_xml_01_unicode():
|
|
deck = getEmptyCol()
|
|
file = str(os.path.join(testDir, "support/supermemo1.xml"))
|
|
i = SupermemoXmlImporter(deck, file)
|
|
# i.META.logToStdOutput = True
|
|
i.run()
|
|
assert i.total == 1
|
|
cid = deck.db.scalar("select id from cards")
|
|
c = deck.getCard(cid)
|
|
# Applies A Factor-to-E Factor conversion
|
|
assert c.factor == 2879
|
|
assert c.reps == 7
|
|
deck.close()
|
|
|
|
|
|
def test_mnemo():
|
|
deck = getEmptyCol()
|
|
file = str(os.path.join(testDir, "support/mnemo.db"))
|
|
i = MnemosyneImporter(deck, file)
|
|
i.run()
|
|
assert deck.cardCount() == 7
|
|
assert "a_longer_tag" in deck.tags.all()
|
|
assert deck.db.scalar("select count() from cards where type = 0") == 1
|
|
deck.close()
|