You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
232 lines
8.9 KiB
232 lines
8.9 KiB
import urwid
|
|
import yaml
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
from API import Fetcher
|
|
from Render import Article
|
|
|
|
|
|
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 = ""
|
|
focus_widget, idx = self.get_focus()
|
|
if self.isCategoryView:
|
|
self.currentCategory = focus_widget.attr_map[None][0]
|
|
|
|
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()
|
|
if self.isCategoryView:
|
|
self.currentCategory = focus_widget.attr_map[None][0]
|
|
tui.articles = self.getArticles(focus_widget.attr_map[None])
|
|
tui.articleView.fill(tui.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()
|
|
if self.isCategoryView:
|
|
self.currentCategory = focus_widget.attr_map[None][0]
|
|
tui.articles = self.getArticles(focus_widget.attr_map[None])
|
|
tui.articleView.fill(tui.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]
|
|
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()
|
|
if self.isCategoryView:
|
|
self.currentCategory = focus_widget.attr_map[None][0]
|
|
tui.articles = self.getArticles(focus_widget.attr_map[None])
|
|
tui.articleView.fill(tui.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()
|
|
tui.articles = self.getArticles(focus_widget.attr_map[None])
|
|
tui.articleView.fill(tui.articles)
|
|
return
|
|
|
|
return super().keypress(size, key)
|
|
|
|
|
|
class RightPane(urwid.ListBox):
|
|
def __init__(self, items):
|
|
super().__init__(self)
|
|
walker = urwid.SimpleListWalker([])
|
|
self.body = walker
|
|
self.isList = True
|
|
self.articlePosition = 0
|
|
self.article = None
|
|
self.chunkNumber = 0
|
|
|
|
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 getArticle(self, articleId):
|
|
article = [item for item in tui.fetcher.articles[tui.feedView.currentCategory[13:]] if item["id"] == articleId][0]
|
|
return article
|
|
|
|
def keypress(self, size, key):
|
|
if key in ("j", "down"):
|
|
if not self.isList:
|
|
if self.chunkNumber >= len(self.article.chunks) - 1:
|
|
pass
|
|
else:
|
|
self.chunkNumber += 1
|
|
walker = urwid.SimpleListWalker([urwid.Text(self.article.chunks[self.chunkNumber])])
|
|
self.body = walker
|
|
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"):
|
|
if not self.isList:
|
|
if self.chunkNumber == 0:
|
|
pass
|
|
else:
|
|
self.chunkNumber -= 1
|
|
walker = urwid.SimpleListWalker([urwid.Text(self.article.chunks[self.chunkNumber])])
|
|
self.body = walker
|
|
focus_widget, idx = self.get_focus()
|
|
if idx > 0:
|
|
idx = idx - 1
|
|
self.set_focus(idx)
|
|
return
|
|
elif key in ("l", "right"):
|
|
if self.isList is True:
|
|
self.isList = False
|
|
self.chunkNumber = 0
|
|
focus_widget, idx = self.get_focus()
|
|
self.articlePosition = idx
|
|
articleId = focus_widget.attr_map[None]
|
|
self.article = Article(self.getArticle(articleId))
|
|
walker = urwid.SimpleListWalker([urwid.Text(self.article.chunks[self.chunkNumber])])
|
|
self.body = walker
|
|
tui.rightBox.set_title(self.article.title)
|
|
return
|
|
elif key in ("h", "left"):
|
|
if self.isList is False:
|
|
self.isList = True
|
|
self.fill(tui.articles)
|
|
tui.rightBox.set_title("Articles")
|
|
self.set_focus(self.articlePosition)
|
|
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.feedView = LeftPane(self.fetcher.categories)
|
|
self.articleView = RightPane([])
|
|
self.leftBox = urwid.LineBox(self.feedView, title="Categories")
|
|
self.rightBox = urwid.LineBox(self.articleView, title="Articles")
|
|
self.container = urwid.Columns(
|
|
[(40, self.leftBox), self.rightBox])
|
|
self.body = self.container
|
|
self.articles = []
|
|
|
|
super().__init__(self.body)
|
|
|
|
def run(self):
|
|
focus_widget, idx = self.feedView.get_focus()
|
|
itemId = focus_widget.attr_map[None][0]
|
|
number = focus_widget.attr_map[None][1]
|
|
tui.fetcher.articlesFromCategory(itemId[13:], str(number))
|
|
self.articles = tui.fetcher.articles[itemId[13:]]
|
|
self.articleView.fill(self.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()
|