Cleaning up old scripts.
This commit is contained in:
Michael Lazar
@@ -1,248 +0,0 @@
|
|||||||
import textwrap
|
|
||||||
import curses
|
|
||||||
|
|
||||||
import praw
|
|
||||||
|
|
||||||
import utils
|
|
||||||
|
|
||||||
class OOBError(Exception):
|
|
||||||
pass
|
|
||||||
|
|
||||||
class SubmissionDisplay(object):
|
|
||||||
|
|
||||||
DEFAULT_REPLY_COLORS = [
|
|
||||||
curses.COLOR_MAGENTA,
|
|
||||||
curses.COLOR_GREEN,
|
|
||||||
curses.COLOR_CYAN,
|
|
||||||
curses.COLOR_YELLOW,
|
|
||||||
]
|
|
||||||
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
stdscr,
|
|
||||||
max_indent_level=5,
|
|
||||||
indent_size=1,
|
|
||||||
reply_colors=None,
|
|
||||||
):
|
|
||||||
|
|
||||||
self.stdscr = stdscr
|
|
||||||
self.line = 0
|
|
||||||
|
|
||||||
self._max_indent_level = max_indent_level
|
|
||||||
self._indent_size = indent_size
|
|
||||||
self._reply_colors = (reply_colors if reply_colors is not None
|
|
||||||
else self.DEFAULT_REPLY_COLORS)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def clean(unicode_string):
|
|
||||||
"Convert unicode string into ascii-safe characters."
|
|
||||||
return unicode_string.encode('ascii', 'replace').replace('\\', '')
|
|
||||||
|
|
||||||
def _get_reply_color(self, nested_level):
|
|
||||||
return self._reply_colors[nested_level % len(self._reply_colors)]
|
|
||||||
|
|
||||||
def _draw_post(self, submission):
|
|
||||||
"Draw the sumbission author's post"
|
|
||||||
|
|
||||||
n_rows, n_cols = self.stdscr.getmaxyx()
|
|
||||||
n_cols -= 1
|
|
||||||
|
|
||||||
title = textwrap.wrap(self.clean(submission.title), n_cols-1)
|
|
||||||
indents = {'initial_indent':' ', 'subsequent_indent':' '}
|
|
||||||
text = textwrap.wrap(self.clean(submission.selftext), n_cols-1, **indents)
|
|
||||||
url = ([self.clean(submission.url)] if
|
|
||||||
getattr(submission, 'url') else [])
|
|
||||||
|
|
||||||
required_rows = 4 + len(text) + len(url) + len(title)
|
|
||||||
if self.line + required_rows > n_rows:
|
|
||||||
raise OOBError()
|
|
||||||
|
|
||||||
win = self.stdscr.derwin(required_rows, n_cols, self.line, 0)
|
|
||||||
submission.window = win
|
|
||||||
win_y = 0
|
|
||||||
self.line += required_rows
|
|
||||||
|
|
||||||
win.hline(win_y, 0, curses.ACS_HLINE, n_cols)
|
|
||||||
win_y += 1
|
|
||||||
|
|
||||||
# Submission title
|
|
||||||
color_attr = curses.color_pair(curses.COLOR_CYAN)
|
|
||||||
win.addstr(win_y, 0, '\n'.join(title), color_attr|curses.A_BOLD)
|
|
||||||
win_y += len(title)
|
|
||||||
|
|
||||||
# Author / Date / Subreddit
|
|
||||||
author = (self.clean(submission.author.name) if
|
|
||||||
getattr(submission, 'author') else '[deleted]')
|
|
||||||
date = utils.humanize_timestamp(submission.created_utc)
|
|
||||||
subreddit = self.clean(submission.subreddit.url)
|
|
||||||
color_attr = curses.color_pair(curses.COLOR_GREEN)
|
|
||||||
win.addstr(win_y, 0, author, curses.A_UNDERLINE|color_attr)
|
|
||||||
win.addstr(' {} {}'.format(date, subreddit), curses.A_BOLD)
|
|
||||||
win_y += 1
|
|
||||||
|
|
||||||
if url:
|
|
||||||
color_attr = curses.color_pair(curses.COLOR_MAGENTA)
|
|
||||||
win.addstr(win_y, 0, url[0], curses.A_BOLD|color_attr)
|
|
||||||
win_y += len(url)
|
|
||||||
|
|
||||||
if text:
|
|
||||||
win.addstr(win_y + len(url), 0, '\n'.join(text))
|
|
||||||
win_y += len(text)
|
|
||||||
|
|
||||||
# Score / Comments
|
|
||||||
score = submission.score
|
|
||||||
num_comments = submission.num_comments
|
|
||||||
info = '{} points {} comments'.format(score, num_comments)
|
|
||||||
win.addstr(win_y, 0, info, curses.A_BOLD)
|
|
||||||
win_y += 1
|
|
||||||
|
|
||||||
win.hline(win_y, 0, curses.ACS_HLINE, n_cols)
|
|
||||||
|
|
||||||
|
|
||||||
def _draw_more_comments(self, comment):
|
|
||||||
"Indicate that more comments can be loaded"
|
|
||||||
|
|
||||||
n_rows, n_cols = self.stdscr.getmaxyx()
|
|
||||||
n_cols -= 1
|
|
||||||
|
|
||||||
required_rows = 2
|
|
||||||
if self.line + required_rows > n_rows:
|
|
||||||
raise OOBError()
|
|
||||||
|
|
||||||
# Determine the indent level of the comment
|
|
||||||
indent_level = min(self._max_indent_level, comment.nested_level)
|
|
||||||
indent = indent_level * self._indent_size
|
|
||||||
n_cols -= indent
|
|
||||||
|
|
||||||
win = self.stdscr.derwin(required_rows, n_cols, self.line, indent)
|
|
||||||
comment.window = win
|
|
||||||
self.line += required_rows
|
|
||||||
win.addnstr(0, indent, '[+] More comments', curses.A_BOLD)
|
|
||||||
|
|
||||||
|
|
||||||
def _draw_comment(self, comment):
|
|
||||||
"Draw a single comment"
|
|
||||||
|
|
||||||
n_rows, n_cols = self.stdscr.getmaxyx()
|
|
||||||
n_cols -= 1
|
|
||||||
|
|
||||||
# Determine the indent level of the comment
|
|
||||||
indent_level = min(self._max_indent_level, comment.nested_level)
|
|
||||||
indent = indent_level * self._indent_size
|
|
||||||
n_cols -= indent
|
|
||||||
|
|
||||||
indents = {'initial_indent':' ', 'subsequent_indent':' '}
|
|
||||||
text = textwrap.wrap(self.clean(comment.body), n_cols-1, **indents)
|
|
||||||
|
|
||||||
required_rows = 2 + len(text)
|
|
||||||
if self.line + required_rows > n_rows:
|
|
||||||
raise OOBError()
|
|
||||||
|
|
||||||
win = self.stdscr.derwin(required_rows, n_cols, self.line, indent)
|
|
||||||
comment.window = win
|
|
||||||
self.line += required_rows
|
|
||||||
|
|
||||||
# Author / Score / Date
|
|
||||||
author = (self.clean(comment.author.name) if
|
|
||||||
getattr(comment, 'author') else '[deleted]')
|
|
||||||
date = utils.humanize_timestamp(comment.created_utc)
|
|
||||||
score = submission.score
|
|
||||||
color_attr = (curses.color_pair(curses.COLOR_GREEN) if comment.is_author
|
|
||||||
else curses.color_pair(curses.COLOR_BLUE))
|
|
||||||
win.addstr(0, 1, author, curses.A_UNDERLINE|color_attr)
|
|
||||||
win.addstr(' {} points {}'.format(score, date), curses.A_BOLD)
|
|
||||||
|
|
||||||
# Body
|
|
||||||
win.addstr(1, 0, '\n'.join(text))
|
|
||||||
|
|
||||||
# Vertical line, unfortunately vline() doesn't support custom color so
|
|
||||||
# we have to build it one chr at a time.
|
|
||||||
reply_color = self._get_reply_color(comment.nested_level)
|
|
||||||
color_attr = curses.color_pair(reply_color)
|
|
||||||
for y in xrange(required_rows-1):
|
|
||||||
win.addch(y, 0, curses.ACS_VLINE, color_attr)
|
|
||||||
|
|
||||||
def _draw_url(self, submission):
|
|
||||||
"Draw the submission url"
|
|
||||||
|
|
||||||
n_rows, n_cols = self.stdscr.getmaxyx()
|
|
||||||
|
|
||||||
color_attr = curses.color_pair(curses.COLOR_RED)
|
|
||||||
url = self.clean(submission.permalink)
|
|
||||||
self.stdscr.addnstr(self.line, 0, url, n_cols-1, color_attr|curses.A_STANDOUT)
|
|
||||||
|
|
||||||
self.line += 1
|
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
def draw_page(self, submission, comments, index=-1):
|
|
||||||
"""
|
|
||||||
Draw the comments page starting at the given index.
|
|
||||||
"""
|
|
||||||
|
|
||||||
# Initialize screen
|
|
||||||
self.stdscr.erase()
|
|
||||||
self.line = 0
|
|
||||||
|
|
||||||
# URL is always drawn
|
|
||||||
self._draw_url(submission)
|
|
||||||
|
|
||||||
if index == -1:
|
|
||||||
self._draw_post(submission)
|
|
||||||
index += 1
|
|
||||||
|
|
||||||
for comment in comments[index:]:
|
|
||||||
try:
|
|
||||||
if isinstance(comment, praw.objects.MoreComments):
|
|
||||||
self._draw_more_comments(comment)
|
|
||||||
else:
|
|
||||||
comment.is_author = (comment.author == submission.author)
|
|
||||||
self._draw_comment(comment)
|
|
||||||
|
|
||||||
except OOBError:
|
|
||||||
break
|
|
||||||
|
|
||||||
self.stdscr.refresh()
|
|
||||||
|
|
||||||
class SubmissionController(object):
|
|
||||||
|
|
||||||
def __init__(self, display):
|
|
||||||
|
|
||||||
self.display = display
|
|
||||||
|
|
||||||
self._index = -1
|
|
||||||
self._cursor = 0
|
|
||||||
|
|
||||||
def loop(self, submission, comments):
|
|
||||||
|
|
||||||
self.display.draw_page(submission, comments, self._index)
|
|
||||||
|
|
||||||
while True:
|
|
||||||
|
|
||||||
key = self.display.stdscr.getch()
|
|
||||||
if key == curses.KEY_DOWN:
|
|
||||||
self._index += 1
|
|
||||||
elif key == curses.KEY_UP and self._index > -1:
|
|
||||||
self._index -= 1
|
|
||||||
elif key == curses.KEY_RESIZE:
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
continue
|
|
||||||
|
|
||||||
self.display.draw_page(submission, comments, self._index)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
|
|
||||||
r = praw.Reddit(user_agent='reddit terminal viewer (rtv) v0.0')
|
|
||||||
r.config.decode_html_entities = True
|
|
||||||
submissions = r.get_subreddit('all').get_hot(limit=5)
|
|
||||||
submission = submissions.next()
|
|
||||||
comments = utils.flatten_tree(submission.comments)
|
|
||||||
|
|
||||||
|
|
||||||
with utils.curses_session() as stdscr:
|
|
||||||
|
|
||||||
display = SubmissionDisplay(stdscr)
|
|
||||||
controller = SubmissionController(display)
|
|
||||||
controller.loop(submission, comments)
|
|
||||||
Reference in New Issue
Block a user