""" The interface to WordNet.
This module contains:
- WordNet: An interface to the WordNet online English lexical database.
"""
from pkg_resources import resource_stream
import pysumo
from . import parser
[docs]class WordNet:
""" An interface to locate and search for items in WordNet. Searches
WordNet for information about terms in the Ontology, to find colloquial
variations of term names and to find terms that are synonyms of a word.
Methods:
- locate_term: Locates a term in WordNet.
- find_synonym: Finds possibly synonyms for a word.
"""
def __init__(self):
try:
self.mapping = parser.wparse([(open('%s/wordnet/sdata.%s' % (pysumo.CONFIG_PATH, pos.name), 'r+b'), pos) for pos in parser.Pos])
except FileNotFoundError:
self.mapping = parser.wparse([(resource_stream('pysumo', '/'.join(['data', 'wordnet', 'sdata.%s' % pos.name])), pos) for pos in parser.Pos])
[docs] def locate_term(self, term):
""" Use the mapping from SUMO to WordNet to retrieve information about a term.
Kwargs:
- term: the term to locate
Returns:
- (name : String, type : SSType, gloss : String)[]
"""
return [(x.synset[0][0], x.ss_type, x.gloss) for x in self.mapping[term]]
[docs] def find_synonym(self, word):
""" Searches WordNet for possible synonyms for word.
Returns:
- String[]
"""
ret = set()
w = word.replace(' ', '_')
for key, value in self.mapping.items():
for wn_item in value:
if w in [x[0] for x in wn_item.synset]:
ret.add(key)
return ret