def classes_of_article(article, limit=sys.maxint):
"""
Given a case-sensitive article title, return the wikipedia classes of
the article (as a list of strings).
"""
result = (WikiClass.select().join(ArticleClass).join(Article).where(
Article.title == article).limit(limit))
return [x.class_name for x in result]
def classes_of_hypernym(hypernym, limit=sys.maxint):
"""
Given a hypernym from DBpedia (string), return a list of
wikipedia classes of that hypernym. Note: use 'thing'
instead of 'owl:Thing'.
"""
hypernym = to_dbpedia_class(hypernym)
result = (WikiClass.select().join(Hypernym).join(DbpediaClass).where(
DbpediaClass.dbpedia_class == hypernym).limit(limit))
return [x.class_name for x in result]
