Esempio n. 1
0
File: Main.py Progetto: jcnm/HotSpot
def generate (n):
    if n>0:
        title = wikipedia.random(pages=1)
        try:
            page = wikipedia.WikipediaPage(title)
            summary = page.summary
            content = tb(page.content)
            tff = {word: tf.tf(word,content) for word in content.words}
            res = {}
            for i in tff:
                if tff[i] == 0.0:
                    continue
                else:
                    res[i] = tff[i]
            article = art.Article(title, summary, kw=res, tfidf={})
            mod.insert(article)
            generate(n-1)
        except wikipedia.exceptions.DisambiguationError : 
            generate(n)
        except bson.errors.InvalidDocument :
            generate(n)