Refactored vote arrow logic.

This commit is contained in:
Michael Lazar
2015-05-02 16:37:25 -07:00
parent 02e9e5e399
commit f0ec9ba712
3 changed files with 22 additions and 26 deletions

View File

@@ -8,8 +8,8 @@ import praw.errors
from .content import SubmissionContent
from .page import BasePage, Navigator, BaseController
from .helpers import open_browser, open_editor
from .curses_helpers import (BULLET, UARROW, DARROW, GOLD, Color, LoadScreen,
show_notification, text_input, add_line)
from .curses_helpers import (GOLD, Color, LoadScreen, get_arrow, add_line,
show_notification, text_input)
from .docs import COMMENT_FILE
__all__ = ['SubmissionController', 'SubmissionPage']
@@ -166,14 +166,8 @@ class SubmissionPage(BasePage):
attr = curses.A_BOLD | Color.YELLOW
add_line(win, u'{flair} '.format(**data), attr=attr)
if data['likes'] is None:
text, attr = BULLET, curses.A_BOLD
elif data['likes']:
text, attr = UARROW, (curses.A_BOLD | Color.GREEN)
else:
text, attr = DARROW, (curses.A_BOLD | Color.RED)
text, attr = get_arrow(likes)
add_line(win, text, attr=attr)
add_line(win, u' {score} {created} '.format(**data))
if data['gold']:
@@ -247,13 +241,7 @@ class SubmissionPage(BasePage):
row = len(data['split_title']) + len(split_text) + 3
add_line(win, u'{score} '.format(**data), row, 1)
if data['likes'] is None:
text, attr = BULLET, curses.A_BOLD
elif data['likes']:
text, attr = UARROW, curses.A_BOLD | Color.GREEN
else:
text, attr = DARROW, curses.A_BOLD | Color.RED
text, attr = get_arrow(data['likes'])
add_line(win, text, attr=attr)
add_line(win, u' {comments} '.format(**data))