def add_entity(self, entity):
"""
:param entity: Id for the entity being added to the dictionary
This method will add the
"""
if entity not in self.entities:
label = WikiData.get_label(entity)
self.entities[entity] = label
def add_prop(self, prop):
"""
:param prop: the id for the preposition to be added to the list
"""
if prop not in self.stored_labels:
label = WikiData.get_label(prop)
self.stored_labels[prop] = label
self.prepositions[prop] = label
else:
if prop not in self.prepositions:
self.prepositions[prop] = self.stored_labels[prop]
def get_sentence(title):
try:
params = { 'action' : 'query', 'format': 'json', 'prop': 'cirrusdoc', 'titles': title }
response = requests.get(WikiApi.url, params)
data = response.json()
pages = data["query"]["pages"]
first_page = next(iter(pages))
#print(pages[first_page])
content = pages[first_page]["cirrusdoc"][0]["source"]["text"]
#print(content)
sentences = sent_tokenize(content)
return sentences
except:
sentences = WikiData.get_sentence(entity)
return sentences
