Updated code for new data structure. #6

1 files changed, 4 insertions, 4 deletions

diff --git a/utils.py b/utils.py
index 576c9ea6..24d92c35 100755
--- a/utils.py
+++ b/utils.py
@@ -155,15 +155,15 @@ def normalize_filename(name):
     file_name = strip_accents(file_name, "utf-8")
     return os.path.normpath(file_name)
 
-def top_words(dic_articles, n=10, size=5):
+def top_words(feeds, n=10, size=5):
     """
     Return the n most frequent words in a list.
     """
     words = Counter()
     wordre = re.compile(r'\b\w{%s,}\b' % size, re.I)
-    for rss_feed_id in dic_articles.keys():
-        for article in dic_articles[rss_feed_id]:
-            for word in wordre.findall(clear_string(article[4].encode('utf-8'))):
+    for feed in feeds.values():
+        for article in feed.articles.values():
+            for word in wordre.findall(clear_string(article.article_description.encode('utf-8'))):
                 words[word.lower()] += 1
     return words.most_common(n)