Inomnibus/App.py

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()