gryf/tuir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Temporary fix.

Browse Source
This commit is contained in:
Michael Lazar
2015-04-16 18:15:59 -07:00
parent 2862b502a1
commit 5146244c26
2 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rtv/__main__.py
View File

@@ -19,7 +19,6 @@ from .__version__ import __version__
__all__ = [] __all__ = []
def load_config(): def load_config():
""" """
Search for a configuration file at the location ~/.rtv and attempt to load Search for a configuration file at the location ~/.rtv and attempt to load
@@ -100,6 +99,7 @@ def main():
if args.log: if args.log:
logging.basicConfig(level=logging.DEBUG, filename=args.log) logging.basicConfig(level=logging.DEBUG, filename=args.log)
logging.captureWarnings(True)
try: try:
print('Connecting...') print('Connecting...')

10
rtv/content.py
View File

@@ -1,4 +1,5 @@
import textwrap import textwrap
import logging
import praw import praw
import requests import requests
@@ -8,6 +9,7 @@ from .helpers import humanize_timestamp, wrap_text, strip_subreddit_url
__all__ = ['SubredditContent', 'SubmissionContent'] __all__ = ['SubredditContent', 'SubmissionContent']
_logger = logging.getLogger(__name__)
class BaseContent(object): class BaseContent(object):
@@ -41,9 +43,9 @@ class BaseContent(object):
retval = [] retval = []
while stack: while stack:
item = stack.pop(0) item = stack.pop(0)
if isinstance(item, praw.objects.MoreComments) and ( if not isinstance(item, praw.objects.MoreComments):
item.count == 0): if item._replies is None:
continue item._replies = [stack.pop(0)]
nested = getattr(item, 'replies', None) nested = getattr(item, 'replies', None)
if nested: if nested:
for n in nested: for n in nested:
@@ -210,7 +212,7 @@ class SubmissionContent(BaseContent):
elif data['type'] == 'MoreComments': elif data['type'] == 'MoreComments':
with self._loader(): with self._loader():
comments = data['object'].comments(update=False) comments = data['object'].comments(update=True)
comments = self.flatten_comments(comments, comments = self.flatten_comments(comments,
root_level=data['level']) root_level=data['level'])
comment_data = [self.strip_praw_comment(c) for c in comments] comment_data = [self.strip_praw_comment(c) for c in comments]