# -*- coding: utf-8 -*-

import sys

import os

import tempfile

import shutil

import traceback

FONAME = "test.apkg"

# Looks like anki libs change working directory to media directory of current deck

# Therefore absolute path should be stored before creating temporary deck

FONAME = os.path.abspath(FONAME)

FBASENAME, _ = os.path.splitext(os.path.basename(FONAME))

CWD = os.getcwd()

TMPDIR = tempfile.mkdtemp(dir = os.path.dirname(FONAME))

def cleanup():

    os.chdir(CWD)

    shutil.rmtree(TMPDIR)

def exception_handler(etype, ex, tb):

    cleanup()

    traceback.print_exception(etype, ex, tb)

sys.excepthook = exception_handler

sys.path.append('/usr/share/anki')

import anki

from anki.exporting import AnkiPackageExporter

collection = anki.Collection(os.path.join(TMPDIR, 'collection.anki2'))

deck_id = collection.decks.id(FBASENAME + "_deck")

deck = collection.decks.get(deck_id)

# deck = collection.decks.confForDid(deck_id)

# collection.decks.update(deck)

# print(dir(deck))

# print(type(deck))

# print(deck)

model = collection.models.new(FBASENAME + "_model")

model['tags'].append(FBASENAME + "_tag")

model['did'] = deck_id

model['css'] = """

.card {

  font-family: arial;

  font-size: 20px;

  text-align: center;

  color: black;

  background-color: white;

}

.from {

  font-style: italic;

}

"""

collection.models.addField(model, collection.models.newField('en'))

collection.models.addField(model, collection.models.newField('ru'))

tmpl = collection.models.newTemplate('en -> ru')

tmpl['qfmt'] = '<div class="from">{{en}}</div>'

tmpl['afmt'] = '{{FrontSide}}<hr id=answer>{{ru}}'

collection.models.addTemplate(model, tmpl)

tmpl = collection.models.newTemplate('ru -> en')

tmpl['qfmt'] = '{{ru}}'

tmpl['afmt'] = '{{FrontSide}}<hr id=answer><div class="from">{{en}}</div>'

collection.models.addTemplate(model, tmpl)

# print(dir(model))

# print(type(model))

# print(model)

# Equivalent of:

# collection.models.add(model)

# without setting auto-generated ID:

model['id'] = 12345678  # essential for upgrade detection

collection.models.update(model)

collection.models.setCurrent(model)

collection.models.save(model)

# collection.decks.select(deck_id)

note = anki.notes.Note(collection, model)

# print(dir(note))

# print(type(note))

# print(note._fmap)

# print(note)

note['en'] = "hello"

note['ru'] = u"[heləʊ] int. привет"

note.guid = "enru_1"

collection.addNote(note)

# print(dir(note))

note = collection.newNote()

note['en'] = "bye"

note['ru'] = u"[baɪ] int. пока"

note.guid = "enru_2"

collection.addNote(note)

# model.add(deck)

# model.save()

################################################################

# Add another model and new types of notes into same storage!

model_irrverb = collection.models.new(FBASENAME + "_model_irrverb")

model_irrverb['tags'].append(FBASENAME + "_tag_irrverb")

model_irrverb['did'] = deck_id

model_irrverb['css'] = """

.card {

  font-family: arial;

  font-size: 20px;

  text-align: center;

  color: black;

  background-color: white;

}

.v1 { color: red; }

.v2 { color: green; }

.v3 { color: blue; }

"""

collection.models.addField(model_irrverb, collection.models.newField('v1'))

collection.models.addField(model_irrverb, collection.models.newField('v2'))

collection.models.addField(model_irrverb, collection.models.newField('v3'))

ANSWER_MSG = '{{FrontSide}}<hr id=answer>{{v1}}<br>{{v2}}<br>{{v3}}'

tmpl = collection.models.newTemplate('v1')

tmpl['qfmt'] = '<div class="v1">{{v1}}</div>'

tmpl['afmt'] = ANSWER_MSG

collection.models.addTemplate(model_irrverb, tmpl)

tmpl = collection.models.newTemplate('v2')

tmpl['qfmt'] = '<div class="v2">{{v2}}</div>'

tmpl['afmt'] = ANSWER_MSG

collection.models.addTemplate(model_irrverb, tmpl)

tmpl = collection.models.newTemplate('v3')

tmpl['qfmt'] = '<div class="v3">{{v3}}</div>'

tmpl['afmt'] = ANSWER_MSG

collection.models.addTemplate(model_irrverb, tmpl)

# We mustn't define 'id' before adding tmpl because addTemplate() fail because

# it tries to update DB but we have no DB entry yet...

model_irrverb['id'] = 12345678+1  # essential for upgrade detection

collection.models.update(model_irrverb)

collection.models.setCurrent(model_irrverb)

collection.models.save(model_irrverb)

note = anki.notes.Note(collection, model_irrverb)

note['v1'] = "go"

note['v2'] = "went"

note['v3'] = "gone"

note.guid = "irrverb_1"

collection.addNote(note)

note = anki.notes.Note(collection, model_irrverb)

note['v1'] = "drive"

note['v2'] = "drove"

note['v3'] = "driven"

note.guid = "irrverb_2"

collection.addNote(note)

export = AnkiPackageExporter(collection)

try:

    os.remove(FONAME)

except OSError as ex:

    if ex.errno not in [2]:

        raise

export.exportInto(FONAME)

cleanup()