Coverage for modules/org/openteacher/logic/loaders/anki2/anki2 : 85%

#! /usr/bin/env python3 

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

#       Copyright 2011, Milan Boers 

#       Copyright 2011-2013, Marten de Vries 

# 

#       This file is part of OpenTeacher. 

# 

#       OpenTeacher is free software: you can redistribute it and/or modify 

#       it under the terms of the GNU General Public License as published by 

#       the Free Software Foundation, either version 3 of the License, or 

#       (at your option) any later version. 

# 

#       OpenTeacher is distributed in the hope that it will be useful, 

#       but WITHOUT ANY WARRANTY; without even the implied warranty of 

#       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 

#       GNU General Public License for more details. 

# 

#       You should have received a copy of the GNU General Public License 

#       along with OpenTeacher.  If not, see <http://www.gnu.org/licenses/>. 

import sqlite3 

import re 

class Anki2LoaderModule: 

        """A pretty basic .anki2 importer. For now it imports everything 

           like it is a words file, which might not always be the best way 

           of dealing with anki files. Also, it does nothing with the results 

           in the database. But, in the end, it might work fine for people 

           who want to switch. :) 

        """ 

        def __init__(self, moduleManager, *args, **kwargs): 

                super().__init__(*args, **kwargs) 

                self.type = "load" 

                self.loads = {"anki2": ["words"]} 

                self.priorities = { 

                        "default": 432, 

                } 

                self._mm = moduleManager 

                self.uses = ( 

                        self._mm.mods(type="translator"), 

                ) 

                self.requires = ( 

                        self._mm.mods(type="wordsStringParser"), 

                ) 

                self.filesWithTranslations = ("anki2.py",) 

        @property 

        def _parse(self): 

                return self._modules.default("active", type="wordsStringParser").parse 

        def _stripTags(self, html): 

                """Thanks mmmdreg! See: http://stackoverflow.com/questions/753052/strip-html-from-strings-in-python""" 

                return re.sub("<[^<]+?>", "", html) 

        def _retranslate(self): 

                try: 

                        translator = self._modules.default("active", type="translator") 

                except IndexError: 

                        _, ngettext = str, lambda a, b, n: a if n == 1 else b 

                else: 

                        _, ngettext = translator.gettextFunctions( 

                                self._mm.resourcePath("translations") 

                        ) 

                #TRANSLATORS: This is one of the file formats OpenTeacher 

                #TRANSLATORS: can read. It's named after the program that uses 

                #TRANSLATORS: it. See http://ankisrs.net/ for more info on it. 

                self.name = _("Anki 2.0") 

        def enable(self): 

                self.mimetype = "application/x-anki2" 

                self._modules = set(self._mm.mods(type="modules")).pop() 

                try: 

                        translator = self._modules.default("active", type="translator") 

                except IndexError: 

                        pass 

                else: 

                        translator.languageChanged.handle(self._retranslate) 

                self._retranslate() 

                self.active = True 

        def disable(self): 

                self.active = False 

                del self.name 

                del self.mimetype 

                del self._modules 

        def getFileTypeOf(self, path): 

exit                if path.endswith(".anki2"): 

                        return "words" 

        def load(self, path): 

                items = [] 

                with sqlite3.connect(path) as connection: 

                        cursor = connection.cursor() 

                        cursor.execute("SELECT flds FROM notes") 

                        for i, (fields,) in enumerate(cursor.fetchall()): 

                                #split on the 'Unit Separator'. The max split 

                                #shouldn't be necessary, nor catching the 

                                #ValueError, but just in case... 

                                try: 

                                        front, back = fields.split(chr(0x1F), 1) 

                                except ValueError: 

                                        continue 

                                items.append({ 

                                        "id": i, 

                                        "questions": self._parse(self._stripTags(front)), 

                                        "answers": self._parse(self._stripTags(back)) 

                                }) 

                        cursor.close() 

                return { 

                        "resources": {}, 

                        "list": { 

                                "items": items, 

                                "results": [], 

                        } 

                } 

def init(moduleManager): 

        return Anki2LoaderModule(moduleManager)