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

Add max title length.

Browse Source
This commit is contained in:
Michael Lazar
2016-07-31 22:36:00 -07:00
parent 0a03083e6b
commit 9cee493dff
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
rtv/content.py
View File

@@ -88,10 +88,7 @@ class Content(object):
data = {} data = {}
data['object'] = comment data['object'] = comment
# Saved comments do not have a nested_level # Saved comments do not have a nested_level
if hasattr(comment, 'nested_level'): data['level'] = getattr(comment, 'nested_level', None)
data['level'] = comment.nested_level
else:
data['level'] = None
if isinstance(comment, praw.objects.MoreComments): if isinstance(comment, praw.objects.MoreComments):
data['type'] = 'MoreComments' data['type'] = 'MoreComments'
@@ -384,10 +381,11 @@ class SubredditContent(Content):
list for repeat access. list for repeat access.
""" """
def __init__(self, name, submissions, loader, order=None): def __init__(self, name, submissions, loader, order=None, max_title_rows=4):
self.name = name self.name = name
self.order = order self.order = order
self.max_title_rows = max_title_rows
self._loader = loader self._loader = loader
self._submissions = submissions self._submissions = submissions
self._submission_data = [] self._submission_data = []
@@ -568,7 +566,7 @@ class SubredditContent(Content):
except StopIteration: except StopIteration:
raise IndexError raise IndexError
else: else:
if hasattr(submission,'title'): if hasattr(submission, 'title'):
data = self.strip_praw_submission(submission) data = self.strip_praw_submission(submission)
else: else:
# when submission is a saved commment # when submission is a saved commment
@@ -583,6 +581,9 @@ class SubredditContent(Content):
# Modifies the original dict, faster than copying # Modifies the original dict, faster than copying
data = self._submission_data[index] data = self._submission_data[index]
data['split_title'] = self.wrap_text(data['title'], width=n_cols) data['split_title'] = self.wrap_text(data['title'], width=n_cols)
if len(data['split_title']) > self.max_title_rows:
data['split_title'] = data['split_title'][:self.max_title_rows-1]
data['split_title'].append('(Not enough space to display)')
data['n_rows'] = len(data['split_title']) + 3 data['n_rows'] = len(data['split_title']) + 3
data['offset'] = 0 data['offset'] = 0