147 lines
4.2 KiB
Python
147 lines
4.2 KiB
Python
import praw
|
|
import curses
|
|
import sys
|
|
|
|
from content_generators import SubmissionContent, SubredditContent
|
|
from utils import curses_session
|
|
from viewer import BaseViewer
|
|
|
|
class SubmissionViewer(BaseViewer):
|
|
|
|
def __init__(self, stdscr, content):
|
|
|
|
page_index, cursor_index = -1, 1
|
|
super(SubmissionViewer, self).__init__(
|
|
stdscr, content,
|
|
page_index=page_index, cursor_index=cursor_index)
|
|
|
|
def loop(self):
|
|
|
|
self.draw()
|
|
while True:
|
|
cmd = self.stdscr.getch()
|
|
|
|
if cmd == curses.KEY_UP:
|
|
self.move_cursor_up()
|
|
self.clear_input_queue()
|
|
|
|
elif cmd == curses.KEY_DOWN:
|
|
self.move_cursor_down()
|
|
self.clear_input_queue()
|
|
|
|
# Refresh page
|
|
elif cmd in (curses.KEY_F5, ord('r')):
|
|
self.refresh_content()
|
|
|
|
elif cmd == curses.KEY_RESIZE:
|
|
self.draw()
|
|
|
|
# Go back
|
|
elif cmd in (ord('b'), curses.KEY_LEFT):
|
|
break
|
|
|
|
# Quit
|
|
elif cmd == ord('q'):
|
|
sys.exit()
|
|
|
|
else:
|
|
curses.beep()
|
|
|
|
def draw(self):
|
|
|
|
n_rows, n_cols = self.stdscr.getmaxyx()
|
|
self._header_window = self.stdscr.derwin(1, n_cols, 0, 0)
|
|
self._content_window = self.stdscr.derwin(1, 0)
|
|
|
|
self.draw_header()
|
|
self.draw_content()
|
|
self.add_cursor()
|
|
|
|
def draw_item(self, win, data, inverted=False):
|
|
|
|
if data['type'] == 'MoreComments':
|
|
self.draw_more_comments(win, data)
|
|
|
|
elif data['type'] == 'Comment':
|
|
self.draw_comment(win, data, inverted=inverted)
|
|
|
|
else:
|
|
self.draw_submission(win, data)
|
|
|
|
@staticmethod
|
|
def draw_comment(win, data, inverted=False):
|
|
|
|
n_rows, n_cols = win.getmaxyx()
|
|
n_cols -= 2
|
|
|
|
# Handle the case where the window is not large enough to fit the data.
|
|
valid_rows = range(0, n_rows)
|
|
offset = 0 if not inverted else -(data['n_rows'] - n_rows)
|
|
|
|
row = offset
|
|
text = '{} {} {}'.format(data['author'], data['score'], data['created'])
|
|
if row in valid_rows:
|
|
win.addnstr(row, 1, text, n_cols)
|
|
|
|
n_body = len(data['split_body'])
|
|
for row, text in enumerate(data['split_body'], start=offset+1):
|
|
if row in valid_rows:
|
|
win.addnstr(row, 1, text, n_cols)
|
|
|
|
# Vertical line, unfortunately vline() doesn't support custom color so
|
|
# we have to build it one chr at a time.
|
|
for y in xrange(n_rows):
|
|
win.addch(y, 0, curses.ACS_VLINE)
|
|
|
|
@staticmethod
|
|
def draw_more_comments(win, data):
|
|
|
|
n_rows, n_cols = win.getmaxyx()
|
|
n_cols -= 2
|
|
win.addnstr(0, 1, data['body'], n_cols)
|
|
|
|
for y in xrange(n_rows):
|
|
win.addch(y, 0, curses.ACS_VLINE)
|
|
|
|
@staticmethod
|
|
def draw_submission(win, data):
|
|
|
|
n_rows, n_cols = win.getmaxyx()
|
|
n_cols -= 3 # one for each side of the border + one for offset
|
|
# Don't print at all if there is not enough room to fit the whole sub
|
|
if data['n_rows'] > n_rows:
|
|
return
|
|
|
|
for row, text in enumerate(data['split_title'], start=1):
|
|
win.addnstr(row, 1, text, n_cols)
|
|
|
|
text = '{} {} {}'.format(data['author'], data['created'], data['subreddit'])
|
|
row = len(data['split_title']) + 1
|
|
win.addnstr(row, 1, text, n_cols)
|
|
|
|
row = len(data['split_title']) + 2
|
|
win.addnstr(row, 1, data['url'], n_cols)
|
|
|
|
offset = len(data['split_title']) + 3
|
|
for row, text in enumerate(data['split_text'], start=offset):
|
|
win.addnstr(row, 1, text, n_cols)
|
|
|
|
text = '{} {}'.format(data['score'], data['comments'])
|
|
row = len(data['split_title']) + len(data['split_text']) + 3
|
|
win.addnstr(row, 1, text, n_cols)
|
|
|
|
win.border()
|
|
|
|
def main():
|
|
|
|
with curses_session() as stdscr:
|
|
r = praw.Reddit(user_agent='reddit terminal viewer (rtv) v0.0')
|
|
submission = SubredditContent(r).get(0)['object']
|
|
generator = SubmissionContent(submission)
|
|
|
|
viewer = SubmissionViewer(stdscr, generator)
|
|
viewer.loop()
|
|
|
|
if __name__ == '__main__':
|
|
|
|
main() |