mirror of
https://github.com/ankitects/anki.git
synced 2025-09-19 06:22:22 -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.
161 lines
4.1 KiB
Python
161 lines
4.1 KiB
Python
# coding: utf-8
|
|
|
|
import os
|
|
import tempfile
|
|
|
|
from nose2.tools.decorators import with_setup
|
|
|
|
from anki import Collection as aopen
|
|
from anki.exporting import *
|
|
from anki.importing import Anki2Importer
|
|
from tests.shared import getEmptyCol as getEmptyColOrig
|
|
|
|
|
|
def getEmptyCol():
|
|
col = getEmptyColOrig()
|
|
col.changeSchedulerVer(2)
|
|
return col
|
|
|
|
|
|
deck = None
|
|
ds = None
|
|
testDir = os.path.dirname(__file__)
|
|
|
|
|
|
def setup1():
|
|
global deck
|
|
deck = getEmptyCol()
|
|
f = deck.newNote()
|
|
f["Front"] = "foo"
|
|
f["Back"] = "bar<br>"
|
|
f.tags = ["tag", "tag2"]
|
|
deck.addNote(f)
|
|
# with a different deck
|
|
f = deck.newNote()
|
|
f["Front"] = "baz"
|
|
f["Back"] = "qux"
|
|
f.model()["did"] = deck.decks.id("new deck")
|
|
deck.addNote(f)
|
|
|
|
|
|
##########################################################################
|
|
|
|
|
|
@with_setup(setup1)
|
|
def test_export_anki():
|
|
# create a new deck with its own conf to test conf copying
|
|
did = deck.decks.id("test")
|
|
dobj = deck.decks.get(did)
|
|
confId = deck.decks.confId("newconf")
|
|
conf = deck.decks.getConf(confId)
|
|
conf["new"]["perDay"] = 5
|
|
deck.decks.save(conf)
|
|
deck.decks.setConf(dobj, confId)
|
|
# export
|
|
e = AnkiExporter(deck)
|
|
fd, newname = tempfile.mkstemp(prefix="ankitest", suffix=".anki2")
|
|
newname = str(newname)
|
|
os.close(fd)
|
|
os.unlink(newname)
|
|
e.exportInto(newname)
|
|
# exporting should not have changed conf for original deck
|
|
conf = deck.decks.confForDid(did)
|
|
assert conf["id"] != 1
|
|
# connect to new deck
|
|
d2 = aopen(newname)
|
|
assert d2.cardCount() == 2
|
|
# as scheduling was reset, should also revert decks to default conf
|
|
did = d2.decks.id("test", create=False)
|
|
assert did
|
|
conf2 = d2.decks.confForDid(did)
|
|
assert conf2["new"]["perDay"] == 20
|
|
dobj = d2.decks.get(did)
|
|
# conf should be 1
|
|
assert dobj["conf"] == 1
|
|
# try again, limited to a deck
|
|
fd, newname = tempfile.mkstemp(prefix="ankitest", suffix=".anki2")
|
|
newname = str(newname)
|
|
os.close(fd)
|
|
os.unlink(newname)
|
|
e.did = 1
|
|
e.exportInto(newname)
|
|
d2 = aopen(newname)
|
|
assert d2.cardCount() == 1
|
|
|
|
|
|
@with_setup(setup1)
|
|
def test_export_ankipkg():
|
|
# add a test file to the media folder
|
|
with open(os.path.join(deck.media.dir(), "今日.mp3"), "w") as f:
|
|
f.write("test")
|
|
n = deck.newNote()
|
|
n["Front"] = "[sound:今日.mp3]"
|
|
deck.addNote(n)
|
|
e = AnkiPackageExporter(deck)
|
|
fd, newname = tempfile.mkstemp(prefix="ankitest", suffix=".apkg")
|
|
newname = str(newname)
|
|
os.close(fd)
|
|
os.unlink(newname)
|
|
e.exportInto(newname)
|
|
|
|
|
|
@with_setup(setup1)
|
|
def test_export_anki_due():
|
|
deck = getEmptyCol()
|
|
f = deck.newNote()
|
|
f["Front"] = "foo"
|
|
deck.addNote(f)
|
|
deck.crt -= 86400 * 10
|
|
deck.sched.reset()
|
|
c = deck.sched.getCard()
|
|
deck.sched.answerCard(c, 3)
|
|
deck.sched.answerCard(c, 3)
|
|
# should have ivl of 1, due on day 11
|
|
assert c.ivl == 1
|
|
assert c.due == 11
|
|
assert deck.sched.today == 10
|
|
assert c.due - deck.sched.today == 1
|
|
# export
|
|
e = AnkiExporter(deck)
|
|
e.includeSched = True
|
|
fd, newname = tempfile.mkstemp(prefix="ankitest", suffix=".anki2")
|
|
newname = str(newname)
|
|
os.close(fd)
|
|
os.unlink(newname)
|
|
e.exportInto(newname)
|
|
# importing into a new deck, the due date should be equivalent
|
|
deck2 = getEmptyCol()
|
|
imp = Anki2Importer(deck2, newname)
|
|
imp.run()
|
|
c = deck2.getCard(c.id)
|
|
deck2.sched.reset()
|
|
assert c.due - deck2.sched.today == 1
|
|
|
|
|
|
# @with_setup(setup1)
|
|
# def test_export_textcard():
|
|
# e = TextCardExporter(deck)
|
|
# f = unicode(tempfile.mkstemp(prefix="ankitest")[1])
|
|
# os.unlink(f)
|
|
# e.exportInto(f)
|
|
# e.includeTags = True
|
|
# e.exportInto(f)
|
|
|
|
|
|
@with_setup(setup1)
|
|
def test_export_textnote():
|
|
e = TextNoteExporter(deck)
|
|
fd, f = tempfile.mkstemp(prefix="ankitest")
|
|
f = str(f)
|
|
os.close(fd)
|
|
os.unlink(f)
|
|
e.exportInto(f)
|
|
assert open(f).readline() == "foo\tbar<br>\ttag tag2\n"
|
|
e.includeTags = False
|
|
e.includeHTML = False
|
|
e.exportInto(f)
|
|
assert open(f).readline() == "foo\tbar\n"
|
|
|
|
|
|
def test_exporters():
|
|
assert "*.apkg" in str(exporters())
|