diff options
Diffstat (limited to '')
| -rwxr-xr-x | pyblog | 141 | ||||
| -rw-r--r-- | utils/utils.py | 141 |
2 files changed, 142 insertions, 140 deletions
@@ -53,147 +53,8 @@ from generators import generators from cli import cli -class PostSelector: - - def __init__(self, term, posts): - self._term = term - self.posts_per_page = term.height - 2 - self.pages = [posts[i:i+self.posts_per_page] - for i in range(0, len(posts), self.posts_per_page)] - self.num_pages = len(self.pages) - self.pagepos = 0 - self.postpos = 0 - self.inserting = False # True if in the middle of inserting a post #, False otherwise - term.enter_fullscreen() - print(term.clear(), end="") - sys.stdout.flush() - self.selection = "" - self.quit = False - - self.display_page() - - def _clear_to_eol(self): - term = self._term - print(term.clear_eol, end="") - sys.stdout.flush() - - def _print_line(self, line, linenum, highlight=False): - term = self._term - width = term.width - with term.location(0, linenum): - if highlight: - print(term.reverse(line[:width]), end="") - else: - print(line[:width], end="") - self._clear_to_eol() - - def _print_post(self, page, pos, highlight=False): - if pos >= len(page): - # if position out of range, just clear the line - self._print_line("", pos + 1, highlight) - else: - date, title, path = page[pos] - line = "%3d: %s %s" % (pos, date.strftime("%m/%d/%y"), title) - self._print_line(line, pos + 1, highlight) - - def display_page(self): - term = self._term - page = self.pages[self.pagepos] - - with term.hidden_cursor(): - topline = " PAGE %d/%d POST %d" % (self.pagepos + 1, self.num_pages, self.postpos) - if self.inserting: - topline += term.blink("_") - self._print_line(topline, 0, highlight=True) - - for i in range(self.posts_per_page): - self._print_post(page, i) - # highlight selected post - self._print_post(page, self.postpos, highlight=True) - - bottomline = " Press h for help." - self._print_line(bottomline, term.height - 1, highlight=True) - - def dispatch(self, key): - term = self._term - if key in string.digits: - # insert - if self.inserting: - newpostpos = 10 * self.postpos + int(key) - if newpostpos < len(self.pages[self.pagepos]): - self.postpos = newpostpos - else: - self.postpos = int(key) - self.inserting = True - elif key.name == "KEY_DELETE": - self.postpos //= 10 - self.inserting = True - else: - self.inserting = False - if key.name == "KEY_ENTER": - self.selection = self.pages[self.pagepos][self.postpos][2] - if key in {"q", "Q"}: - self.quit = True - elif key.name == "KEY_DOWN" or key in {"n", "N"}: - if self.postpos + 1 < len(self.pages[self.pagepos]): - self.postpos += 1 - elif key.name == "KEY_UP" or key in {"p", "P"}: - if self.postpos > 0: - self.postpos -= 1 - elif key.name == "KEY_RIGHT" or key in {".", ">"}: - if self.pagepos + 1 < self.num_pages: - self.pagepos += 1 - self.postpos = 0 - elif key.name == "KEY_LEFT" or key in {",", "<"}: - if self.pagepos > 0: - self.pagepos -= 1 - self.postpos = 0 - elif key in {"h", "H"}: - print(term.clear_eol, end="") - sys.stdout.flush() - help_text_lines = [ - "Next post: n or <down>", - "Previous post: p or <up>", - "Next page: . or > or <right>", - "Previous page: , or < or <left>", - "Select post: <enter> or <return>", - "Select by number: type number as shown (delete or backspace to edit)", - "Get help: h", - "Quit program: q", - ] - for i in range(term.height - 1): - self._print_line(help_text_lines[i] if i < len(help_text_lines) else "", i) - bottomline = " Press any key to continue." - self._print_line(bottomline, term.height - 1, highlight=True) - - with term.raw(): - term.inkey() - - def restore(self): - term = self._term - term.exit_fullscreen() - print(term.clear(), end="") - sys.stdout.flush() - - def select(self): - term = self._term - try: - while True: - with term.raw(): - self.dispatch(term.inkey()) - if self.selection or self.quit: - break - self.display_page() - except Exception: - raise - finally: - self.restore() - - return self.selection - - def edit_existing_post(args): - selector = PostSelector(blessed.Terminal(), utils.list_posts()) + selector = utils.PostSelector(blessed.Terminal(), utils.list_posts()) selection = selector.select() if selection: print(selection) diff --git a/utils/utils.py b/utils/utils.py index f8a756c5..11d806f5 100644 --- a/utils/utils.py +++ b/utils/utils.py @@ -19,6 +19,8 @@ import multiprocessing import os import sys +import string + from config.config import * from rss import * @@ -298,3 +300,142 @@ def list_posts(): posts.sort(key=lambda post: post[0], reverse=True) return posts + + +class PostSelector: + + def __init__(self, term, posts): + self._term = term + self.posts_per_page = term.height - 2 + self.pages = [posts[i:i+self.posts_per_page] + for i in range(0, len(posts), self.posts_per_page)] + self.num_pages = len(self.pages) + self.pagepos = 0 + self.postpos = 0 + self.inserting = False # True if in the middle of inserting a post #, False otherwise + term.enter_fullscreen() + print(term.clear(), end="") + sys.stdout.flush() + self.selection = "" + self.quit = False + + self.display_page() + + def _clear_to_eol(self): + term = self._term + print(term.clear_eol, end="") + sys.stdout.flush() + + def _print_line(self, line, linenum, highlight=False): + term = self._term + width = term.width + with term.location(0, linenum): + if highlight: + print(term.reverse(line[:width]), end="") + else: + print(line[:width], end="") + self._clear_to_eol() + + def _print_post(self, page, pos, highlight=False): + if pos >= len(page): + # if position out of range, just clear the line + self._print_line("", pos + 1, highlight) + else: + date, title, path = page[pos] + line = "%3d: %s %s" % (pos, date.strftime("%m/%d/%y"), title) + self._print_line(line, pos + 1, highlight) + + def display_page(self): + term = self._term + page = self.pages[self.pagepos] + + with term.hidden_cursor(): + topline = " PAGE %d/%d POST %d" % (self.pagepos + 1, self.num_pages, self.postpos) + if self.inserting: + topline += term.blink("_") + self._print_line(topline, 0, highlight=True) + + for i in range(self.posts_per_page): + self._print_post(page, i) + # highlight selected post + self._print_post(page, self.postpos, highlight=True) + + bottomline = " Press h for help." + self._print_line(bottomline, term.height - 1, highlight=True) + + def dispatch(self, key): + term = self._term + if key in string.digits: + # insert + if self.inserting: + newpostpos = 10 * self.postpos + int(key) + if newpostpos < len(self.pages[self.pagepos]): + self.postpos = newpostpos + else: + self.postpos = int(key) + self.inserting = True + elif key.name == "KEY_DELETE": + self.postpos //= 10 + self.inserting = True + else: + self.inserting = False + if key.name == "KEY_ENTER": + self.selection = self.pages[self.pagepos][self.postpos][2] + if key in {"q", "Q"}: + self.quit = True + elif key.name == "KEY_DOWN" or key in {"n", "N"}: + if self.postpos + 1 < len(self.pages[self.pagepos]): + self.postpos += 1 + elif key.name == "KEY_UP" or key in {"p", "P"}: + if self.postpos > 0: + self.postpos -= 1 + elif key.name == "KEY_RIGHT" or key in {".", ">"}: + if self.pagepos + 1 < self.num_pages: + self.pagepos += 1 + self.postpos = 0 + elif key.name == "KEY_LEFT" or key in {",", "<"}: + if self.pagepos > 0: + self.pagepos -= 1 + self.postpos = 0 + elif key in {"h", "H"}: + print(term.clear_eol, end="") + sys.stdout.flush() + help_text_lines = [ + "Next post: n or <down>", + "Previous post: p or <up>", + "Next page: . or > or <right>", + "Previous page: , or < or <left>", + "Select post: <enter> or <return>", + "Select by number: type number as shown (delete or backspace to edit)", + "Get help: h", + "Quit program: q", + ] + for i in range(term.height - 1): + self._print_line(help_text_lines[i] if i < len(help_text_lines) else "", i) + bottomline = " Press any key to continue." + self._print_line(bottomline, term.height - 1, highlight=True) + + with term.raw(): + term.inkey() + + def restore(self): + term = self._term + term.exit_fullscreen() + print(term.clear(), end="") + sys.stdout.flush() + + def select(self): + term = self._term + try: + while True: + with term.raw(): + self.dispatch(term.inkey()) + if self.selection or self.quit: + break + self.display_page() + except Exception: + raise + finally: + self.restore() + + return self.selection |