Experimental curses app for viewing the timeline - toot - Unnamed repository; edit this file 'description' to name the repository.
(DIR) Log
(DIR) Files
(DIR) Refs
(DIR) LICENSE
---
(DIR) commit a3b207647b36b877399a0151f4faeb4de2503945
(DIR) parent 1c22eaa44fe49379caaf8b8dac1cfe6c66b3d255
(HTM) Author: Ivan Habunek <ivan@habunek.com>
Date: Fri, 21 Apr 2017 20:23:48 +0200
Experimental curses app for viewing the timeline
Diffstat:
toot/commands.py | 6 ++++++
toot/console.py | 6 ++++++
toot/curses.py | 251 +++++++++++++++++++++++++++++++
3 files changed, 263 insertions(+), 0 deletions(-)
---
(DIR) diff --git a/toot/commands.py b/toot/commands.py
@@ -15,6 +15,7 @@ from textwrap import TextWrapper, wrap
from toot import api, config, DEFAULT_INSTANCE, User, App, ConsoleError
from toot.output import green, yellow, print_error
+from toot.curses import TimelineApp
def register_app(instance):
@@ -166,6 +167,11 @@ def timeline(app, user, args):
print("─" * 31 + "┼" + "─" * 88)
+def curses(app, user, args):
+ generator = api.timeline_generator(app, user)
+ TimelineApp(generator).run()
+
+
def post(app, user, args):
if args.media:
media = _do_upload(app, user, args.media)
(DIR) diff --git a/toot/console.py b/toot/console.py
@@ -138,6 +138,12 @@ COMMANDS = [
arguments=[],
require_auth=True,
),
+ Command(
+ name="curses",
+ description="An experimental timeline app.",
+ arguments=[],
+ require_auth=True,
+ ),
]
(DIR) diff --git a/toot/curses.py b/toot/curses.py
@@ -0,0 +1,251 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+from __future__ import print_function
+
+import curses
+import re
+import webbrowser
+
+from bs4 import BeautifulSoup
+from textwrap import wrap
+
+
+class Color:
+ @staticmethod
+ def setup_palette():
+ curses.init_pair(1, curses.COLOR_BLUE, curses.COLOR_BLACK)
+ curses.init_pair(2, curses.COLOR_GREEN, curses.COLOR_BLACK)
+ curses.init_pair(3, curses.COLOR_YELLOW, curses.COLOR_BLACK)
+
+ @staticmethod
+ def blue():
+ return curses.color_pair(1)
+
+ @staticmethod
+ def green():
+ return curses.color_pair(2)
+
+ @staticmethod
+ def yellow():
+ return curses.color_pair(3)
+
+
+class TimelineApp:
+ def __init__(self, status_generator):
+ self.status_generator = status_generator
+ self.statuses = []
+ self.selected = None
+
+ def run(self):
+ curses.wrapper(self._wrapped_run)
+
+ def _wrapped_run(self, stdscr):
+ self.left_width = 60
+ self.right_width = curses.COLS - self.left_width
+
+ # Setup windows
+ self.top = curses.newwin(2, curses.COLS, 0, 0)
+ self.left = curses.newpad(curses.LINES * 2, self.left_width)
+ self.right = curses.newwin(curses.LINES - 4, self.right_width, 2, self.left_width)
+ self.bottom = curses.newwin(2, curses.COLS, curses.LINES - 2, 0)
+
+ Color.setup_palette()
+
+ # Load some data and redraw
+ self.fetch_next()
+ self.selected = 0
+ self.full_redraw()
+
+ self.loop()
+
+ def loop(self):
+ while True:
+ key = self.left.getkey()
+
+ if key.lower() == 'q':
+ return
+
+ elif key.lower() == 'v':
+ status = self.get_selected_status()
+ if status:
+ webbrowser.open(status['url'])
+
+ elif key.lower() == 'j' or key == curses.KEY_DOWN:
+ self.select_next()
+
+ elif key.lower() == 'k' or key == curses.KEY_UP:
+ self.select_previous()
+
+ def select_previous(self):
+ """Move to the previous status in the timeline."""
+ if self.selected == 0:
+ return
+
+ old_index = self.selected
+ new_index = self.selected - 1
+
+ self.selected = new_index
+ self.redraw_after_selection_change(old_index, new_index)
+
+ def select_next(self):
+ """Move to the next status in the timeline."""
+ if self.selected + 1 >= len(self.statuses):
+ return
+
+ old_index = self.selected
+ new_index = self.selected + 1
+
+ self.selected = new_index
+ self.redraw_after_selection_change(old_index, new_index)
+
+ def redraw_after_selection_change(self, old_index, new_index):
+ old_status = self.statuses[old_index]
+ new_status = self.statuses[new_index]
+
+ # Perform a partial redraw
+ self.draw_status_row(self.left, old_status, 3 * old_index - 1, False)
+ self.draw_status_row(self.left, new_status, 3 * new_index - 1, True)
+ self.draw_status_details(self.right, new_status)
+
+ def fetch_next(self):
+ try:
+ statuses = self.status_generator.__next__()
+ except StopIteration:
+ return None
+
+ for status in statuses:
+ self.statuses.append(parse_status(status))
+
+ return len(statuses)
+
+ def full_redraw(self):
+ """Perform a full redraw of the UI."""
+ self.left.clear()
+ self.right.clear()
+ self.top.clear()
+ self.bottom.clear()
+
+ self.left.box()
+ self.right.box()
+
+ self.top.addstr(" toot - your Mastodon command line interface\n", Color.yellow())
+ self.top.addstr(" https://github.com/ihabunek/toot")
+
+ self.draw_statuses(self.left)
+ self.draw_status_details(self.right, self.get_selected_status())
+ self.draw_usage(self.bottom)
+
+ self.left.refresh(0, 0, 2, 0, curses.LINES - 4, self.left_width)
+
+ self.right.refresh()
+ self.top.refresh()
+ self.bottom.refresh()
+
+ def draw_usage(self, window):
+ # Show usage on the bottom
+ window.addstr("Usage: | ")
+ window.addch("j", Color.green())
+ window.addstr(" next | ")
+ window.addch("k", Color.green())
+ window.addstr(" previous | ")
+ window.addch("v", Color.green())
+ window.addstr(" open in browser | ")
+ window.addch("q", Color.green())
+ window.addstr(" quit")
+
+ window.refresh()
+
+ def get_selected_status(self):
+ if len(self.statuses) > self.selected:
+ return self.statuses[self.selected]
+
+ def draw_status_row(self, window, status, offset, highlight=False):
+ width = window.getmaxyx()[1]
+ color = Color.blue() if highlight else 0
+
+ date, time = status['created_at']
+ window.addstr(offset + 2, 2, date, color)
+ window.addstr(offset + 3, 2, time, color)
+
+ window.addstr(offset + 2, 15, status['author']['acct'], color)
+ window.addstr(offset + 3, 15, status['author']['display_name'], color)
+
+ window.addstr(offset + 4, 1, '─' * (width - 2))
+
+ window.refresh(0, 0, 2, 0, curses.LINES - 4, self.left_width)
+
+
+ def draw_statuses(self, window):
+ for index, status in enumerate(self.statuses):
+ offset = 3 * index - 1
+ highlight = self.selected == index
+ self.draw_status_row(window, status, offset, highlight)
+
+ def draw_status_details(self, window, status):
+ window.erase()
+ window.box()
+
+ acct = status['author']['acct']
+ name = status['author']['display_name']
+
+ window.addstr(1, 2, "@" + acct, Color.green())
+ window.addstr(2, 2, name, Color.yellow())
+
+ text_width = self.right_width - 4
+
+ y = 4
+ for line in status['lines']:
+ for wrapped in wrap(line, text_width):
+ window.addstr(y, 2, wrapped.ljust(text_width))
+ y += 1
+ y += 1
+
+ window.addstr(y, 2, '─' * text_width)
+ y += 1
+
+ window.addstr(y, 2, status['url'])
+ y += 1
+
+ if status['boosted_by']:
+ acct = status['boosted_by']['acct']
+ window.addstr(y, 2, "Boosted by ")
+ window.addstr("@", Color.green())
+ window.addstr(acct, Color.green())
+ y += 1
+
+ window.refresh()
+
+
+def parse_status(status):
+ content = status['reblog']['content'] if status['reblog'] else status['content']
+ account = parse_account(status['reblog']['account'] if status['reblog'] else status['account'])
+ boosted_by = parse_account(status['account']) if status['reblog'] else None
+
+ lines = parse_html(content)
+
+ created_at = status['created_at'][:19].split('T')
+
+ return {
+ 'author': account,
+ 'boosted_by': boosted_by,
+ 'lines': lines,
+ 'url': status['url'],
+ 'created_at': created_at,
+ }
+
+
+def parse_account(account):
+ return {
+ 'id': account['id'],
+ 'acct': account['acct'],
+ 'display_name': account['display_name'],
+ }
+
+
+def parse_html(html):
+ """Attempt to convert html to plain text while keeping line breaks"""
+ return [
+ BeautifulSoup(l, "html.parser").get_text().replace(''', "'")
+ for l in re.split("</?p[^>]*>", html)
+ if l
+ ]