# Copyright: Ankitects Pty Ltd and contributors
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
# coding: utf-8
import os
import tempfile
from anki import Collection as aopen
from anki.exporting import *
from anki.importing import Anki2Importer
from tests.shared import errorsAfterMidnight
from tests.shared import getEmptyCol as getEmptyColOrig
def getEmptyCol():
col = getEmptyColOrig()
col.upgrade_to_v2_scheduler()
return col
col = None
ds = None
testDir = os.path.dirname(__file__)
def setup1():
global col
col = getEmptyCol()
note = col.newNote()
note["Front"] = "foo"
note["Back"] = "bar
"
note.tags = ["tag", "tag2"]
col.addNote(note)
# with a different col
note = col.newNote()
note["Front"] = "baz"
note["Back"] = "qux"
note.note_type()["did"] = col.decks.id("new col")
col.addNote(note)
##########################################################################
def test_export_anki():
setup1()
# create a new col with its own conf to test conf copying
did = col.decks.id("test")
dobj = col.decks.get(did)
confId = col.decks.add_config_returning_id("newconf")
conf = col.decks.get_config(confId)
conf["new"]["perDay"] = 5
col.decks.save(conf)
col.decks.set_config_id_for_deck_dict(dobj, confId)
# export
e = AnkiExporter(col)
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 = col.decks.config_dict_for_deck_id(did)
assert conf["id"] != 1
# connect to new deck
col2 = aopen(newname)
assert col2.card_count() == 2
# as scheduling was reset, should also revert decks to default conf
did = col2.decks.id("test", create=False)
assert did
conf2 = col2.decks.config_dict_for_deck_id(did)
assert conf2["new"]["perDay"] == 20
dobj = col2.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)
col2 = aopen(newname)
assert col2.card_count() == 1
def test_export_ankipkg():
setup1()
# add a test file to the media folder
with open(os.path.join(col.media.dir(), "今日.mp3"), "w") as note:
note.write("test")
n = col.newNote()
n["Front"] = "[sound:今日.mp3]"
col.addNote(n)
e = AnkiPackageExporter(col)
fd, newname = tempfile.mkstemp(prefix="ankitest", suffix=".apkg")
newname = str(newname)
os.close(fd)
os.unlink(newname)
e.exportInto(newname)
@errorsAfterMidnight
def test_export_anki_due():
setup1()
col = getEmptyCol()
note = col.newNote()
note["Front"] = "foo"
col.addNote(note)
col.crt -= 86400 * 10
col.sched.reset()
c = col.sched.getCard()
col.sched.answerCard(c, 3)
col.sched.answerCard(c, 3)
# should have ivl of 1, due on day 11
assert c.ivl == 1
assert c.due == 11
assert col.sched.today == 10
assert c.due - col.sched.today == 1
# export
e = AnkiExporter(col)
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
col2 = getEmptyCol()
imp = Anki2Importer(col2, newname)
imp.run()
c = col2.getCard(c.id)
col2.sched.reset()
assert c.due - col2.sched.today == 1
# def test_export_textcard():
# setup1()
# e = TextCardExporter(col)
# note = unicode(tempfile.mkstemp(prefix="ankitest")[1])
# os.unlink(note)
# e.exportInto(note)
# e.includeTags = True
# e.exportInto(note)
def test_export_textnote():
setup1()
e = TextNoteExporter(col)
fd, note = tempfile.mkstemp(prefix="ankitest")
note = str(note)
os.close(fd)
os.unlink(note)
e.exportInto(note)
with open(note) as file:
assert file.readline() == "foo\tbar
\ttag tag2\n"
e.includeTags = False
e.includeHTML = False
e.exportInto(note)
with open(note) as file:
assert file.readline() == "foo\tbar\n"
def test_exporters():
assert "*.apkg" in str(exporters(getEmptyCol()))