import urwid import yaml from concurrent.futures import ThreadPoolExecutor from API import Fetcher class LeftPane(urwid.ListBox): def __init__(self, categories): super().__init__(self) items = [urwid.AttrMap(urwid.Columns([(5, urwid.Text(str(category["count"]))), urwid.Text(category["name"])]), (category["id"], category["count"]), "reveal focus") for category in categories] walker = urwid.SimpleListWalker(items) self.body = walker self.categoryPosition = 0 self.isCategoryView = True self.currentCategory = "" def getArticlesFromCategory(self, category, number): tui.fetcher.articlesFromCategory(category[13:], str(number)) articles = tui.fetcher.articles[category[13:]] return articles def getArticlesFromFeed(self, feed): return tui.fetcher.articlesFromFeed(feed, self.currentCategory[13:]) def getArticles(self, attrMap): if self.isCategoryView: itemId = attrMap[0] number = attrMap[1] tui.fetcher.articlesFromCategory(itemId[13:], str(number)) return tui.fetcher.articles[itemId[13:]] else: itemId = attrMap return tui.fetcher.articlesFromFeed(itemId, self.currentCategory[13:]) def keypress(self, size, key): if key in ("j", "down"): item_size = len(self.body) focus_widget, idx = self.get_focus() if idx < item_size - 1: idx = idx + 1 self.set_focus(idx) focus_widget, idx = self.get_focus() articles = self.getArticles(focus_widget.attr_map[None]) tui.articles.fill(articles) return elif key in ("k", "up"): focus_widget, idx = self.get_focus() if idx > 0: idx = idx - 1 self.set_focus(idx) focus_widget, idx = self.get_focus() articles = self.getArticles(focus_widget.attr_map[None]) tui.articles.fill(articles) return elif key in ("l", "right"): if self.isCategoryView: self.isCategoryView = False focus_widget, idx = self.get_focus() self.categoryPosition = idx categoryId = focus_widget.attr_map[None][0] self.currentCategory = categoryId feeds = tui.fetcher.feedsFromCategory(categoryId) feedItems = [urwid.AttrMap(urwid.Columns([(5, urwid.Text(str(tui.fetcher.unreadCounts[feed["id"]][0]))), urwid.Text(feed["title"])]), feed["id"], "reveal focus") for feed in feeds] walker = urwid.SimpleListWalker(feedItems) self.body = walker tui.leftBox.set_title(categoryId[13:]) focus_widget, idx = self.get_focus() articles = self.getArticles(focus_widget.attr_map[None]) tui.articles.fill(articles) return elif key in ("h", "left"): if not self.isCategoryView: self.isCategoryView = True items = [urwid.AttrMap(urwid.Columns([(5, urwid.Text(str(category["count"]))), urwid.Text(category["name"])]), (category["id"], category["count"]), "reveal focus") for category in tui.fetcher.categories] walker = urwid.SimpleFocusListWalker(items) self.body = walker tui.leftBox.set_title("Categories") self.set_focus(self.categoryPosition) focus_widget, idx = self.get_focus() articles = self.getArticles(focus_widget.attr_map[None]) tui.articles.fill(articles) return return super().keypress(size, key) class RightPane(urwid.ListBox): def __init__(self, items): super().__init__(self) walker = urwid.SimpleListWalker([]) self.body = walker def fill(self, articles): items = [urwid.AttrMap(urwid.Columns([(2, urwid.Text("N")), urwid.Text(article["title"])]), article["id"], "reveal focus") for article in articles] walker = urwid.SimpleListWalker(items) self.body = walker def keypress(self, size, key): if key in ("j", "down"): item_size = len(self.body) focus_widget, idx = self.get_focus() if idx < item_size - 1: idx = idx + 1 self.set_focus(idx) return elif key in ("k", "up"): focus_widget, idx = self.get_focus() if idx > 0: idx = idx - 1 self.set_focus(idx) return elif key in ("l", "right"): focus_widget, idx = self.get_focus() value = next(focus_widget.render((100,), focus=False).content())[1][2].decode().strip() feeds = tui.leftPaneItems[value]["feeds"] feedItems = [urwid.Columns([(4, urwid.Text(item["count"])), urwid.Text(item["name"])]) for item in feeds] walker = urwid.SimpleListWalker([urwid.AttrMap(item, None, 'reveal focus') for item in feedItems]) self.body = walker tui.leftBox.set_title(value) return elif key in ("h", "left"): feedItems = tui.feedItems walker = urwid.SimpleListWalker([urwid.AttrMap(item, None, 'reveal focus') for item in feedItems]) self.body = walker tui.leftBox.set_title("Categories") return return super().keypress(size, key) class TUI(urwid.Frame): @classmethod def create(cls): tui = cls() palette = [('header', 'white', 'black'), ('reveal focus', 'black', 'dark cyan', 'standout')] loop = urwid.MainLoop( tui, palette, event_loop=urwid.AsyncioEventLoop(), unhandled_input=tui.unhandled_input, ) tui.loop = loop return tui def __init__(self): with open('config.yml', 'r') as file: config = yaml.safe_load(file) URL = config["server"]["URL"] token = config["server"]["token"] self.fetcher = Fetcher(URL, token) self.leftPaneItems = {} self.activePane = False self.executor = ThreadPoolExecutor(max_workers=1) self.loop = None self.articleItems = [urwid.Text("Article# " + str(i)) for i in range(0, 10)] self.feeds = LeftPane(self.fetcher.categories) self.articles = RightPane([]) self.leftBox = urwid.LineBox(self.feeds, title="Categories") self.container = urwid.Columns( [(40, self.leftBox), urwid.LineBox(self.articles, title="Articles")]) self.body = self.container super().__init__(self.body) def run(self): focus_widget, idx = self.feeds.get_focus() itemId = focus_widget.attr_map[None][0] number = focus_widget.attr_map[None][1] tui.fetcher.articlesFromCategory(itemId[13:], str(number)) articles = tui.fetcher.articles[itemId[13:]] self.articles.fill(articles) self.loop.run() self.executor.shutdown(wait=False) def unhandled_input(self, key): if key == "tab": self.activePane = not self.activePane self.container.set_focus(int(self.activePane)) elif key == "q": raise urwid.ExitMainLoop() tui = TUI.create() tui.run()