mirror of
https://github.com/ankitects/anki.git
synced 2025-09-19 14:32:22 -04:00
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright: Damien Elmes <anki@ichi2.net>
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
import zipfile, os
|
|
from anki.utils import tmpfile, json
|
|
from anki.importing.anki2 import Anki2Importer
|
|
|
|
class AnkiPackageImporter(Anki2Importer):
|
|
|
|
def run(self):
|
|
# extract the deck from the zip file
|
|
self.zip = z = zipfile.ZipFile(self.file)
|
|
col = z.read("collection.anki2")
|
|
colpath = tmpfile(suffix=".anki2")
|
|
open(colpath, "wb").write(col)
|
|
self.file = colpath
|
|
# we need the media dict in advance, and we'll need a map of fname ->
|
|
# number to use during the import
|
|
self.nameToNum = {}
|
|
for k, v in json.loads(z.read("media")).items():
|
|
self.nameToNum[v] = k
|
|
# run anki2 importer
|
|
Anki2Importer.run(self)
|
|
# import static media
|
|
for file, c in self.nameToNum.items():
|
|
if not file.startswith("_"):
|
|
continue
|
|
path = os.path.join(self.col.media.dir(), file)
|
|
if not os.path.exists(path):
|
|
open(path, "wb").write(z.read(c))
|
|
|
|
def _srcMediaData(self, fname):
|
|
if fname in self.nameToNum:
|
|
return self.zip.read(self.nameToNum[fname])
|
|
return None
|