1

Adding visual character search to dictionary

Former-commit-id: 6d98f5b79a3702391046aa069a656b6d9fdce374
This commit is contained in:
Alex Yatskov 2013-11-08 13:46:30 -08:00
parent 1a9dcc69ff
commit 988916820a

View File

@ -16,6 +16,7 @@
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import operator
import re
import sqlite3
@ -39,13 +40,44 @@ class Dictionary:
def findCharacter(self, character):
cursor = self.db.cursor()
if not self.hasIndex('KanjiIndex'):
cursor.execute('CREATE INDEX KanjiIndex ON Kanji(character)')
self.db.commit()
cursor.execute('SELECT * FROM Kanji WHERE character=?', character)
return cursor.fetchall()
return cursor.fetchone()
def findCharacterVisually(self, characters):
radicals = dict()
for character in characters:
for radical in self.findRadicalsByCharacter(character):
radicals[radical] = radicals.get(radical, 0) + 1
results = dict()
for radical, count in radicals.items():
for character in self.findCharactersByRadical(radical):
results[character] = results.get(character, 0) + count
return sorted(results.items(), key=operator.itemgetter(1), reverse=True)
def findRadicalsByCharacter(self, character):
cursor = self.db.cursor()
cursor.execute('SELECT radicals FROM Radicals WHERE character=?', character)
columns = cursor.fetchone()
if columns is None:
return None
return columns[0].split()
def findCharactersByRadical(self, radical):
cursor = self.db.cursor()
cursor.execute('SELECT character FROM Radicals WHERE radicals LIKE ?', (u'%{0}%'.format(radical),))
columns = cursor.fetchall()
if columns is None:
return None
return map(operator.itemgetter(0), columns)
def hasIndex(self, name):