Есть ли способ вернуть расстояние в QueryParser из whoosh - PullRequest
Новая
       27

Есть ли способ вернуть расстояние в QueryParser из whoosh

1 голос
/ 09 апреля 2019

Я использую пакет whoosh для нечеткого поиска в Python. Мне интересно, есть ли способ вернуть расстояние?

Мой код выглядит следующим образом: 

import codecs
import whoosh
import os, os.path



from whoosh.index import create_in
from whoosh.fields import *
from whoosh.query import FuzzyTerm


class MyFuzzyTerm(FuzzyTerm):
    def __init__(self, fieldname, text, boost=1.0, maxdist=5, prefixlength=1, constantscore=True):
        super(MyFuzzyTerm, self).__init__(fieldname, text, boost, maxdist, prefixlength, constantscore)


if not os.path.exists("indexdir"):
    os.mkdir("indexdir")

path = u"MMM2.txt"
content = open('MMM2.txt', 'r').read()

schema = Schema(name=TEXT(stored=True), content=TEXT)
ix = create_in("indexdir", schema)
writer = ix.writer()
writer.add_document(name=path, content= content)

writer.commit()

from whoosh.qparser import QueryParser, FuzzyTermPlugin, PhrasePlugin, SequencePlugin

with ix.searcher() as searcher:
    parser = QueryParser(u"content", ix.schema,termclass = MyFuzzyTerm)
    parser.add_plugin(FuzzyTermPlugin())
    parser.remove_plugin_class(PhrasePlugin)
    parser.add_plugin(SequencePlugin())
    query = parser.parse(u"\"Tennessee Riverkeep Inc\"~")
    results = searcher.search(query)
    print ("nb of results =", len(results))
    for r in results:
        print (r)

Предмет, который я ищу, - "Tennessee Riverkeep Inc". Пункт, показанный точно в документе, является "Теннесси Риверкипер Инк". Таким образом, расстояние составляет 2 в этом случае. Я установил максимальное расстояние 5. Есть ли способ вернуть число «2» в этом случае?

...