SubredditViewer refactor, cursor almost passable.

This commit is contained in:
Michael Lazar
2015-01-21 01:20:23 -08:00
parent cf204f23ff
commit 8da813833f
2 changed files with 104 additions and 62 deletions

View File

@@ -1,5 +1,5 @@
from datetime import datetime, timedelta
import textwrap
def clean(unicode_string):
"""
@@ -84,8 +84,10 @@ class SubredditGenerator(object):
if subreddit is None:
self._submissions = self.r.get_front_page(limit=None)
self.display_name = 'Front Page'
else:
self._submissions = self.r.get_subreddit(subreddit, limit=None)
self.display_name = self._submissions.display_name
self._submission_data = []
@@ -118,7 +120,7 @@ class SubredditGenerator(object):
assert(index >= 0)
while index >= len(self._submission_data):
data = self._strip_praw_submission(self._submissions.next())
data = self.strip_praw_submission(self._submissions.next())
self._submission_data.append(data)
# Modifies the original dict, faster than copying