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