added the ability to check for unread messages with the 'i' key

This commit is contained in:
Lawrence Vanderpool
2015-06-10 11:04:01 -04:00
parent 46ec642e60
commit abef7b4045

View File

@@ -412,8 +412,23 @@ class BasePage(object):
s.catch = False s.catch = False
self.refresh_content() self.refresh_content()
@BaseController.register('i')
def get_inbox(self):
"""
Checks the inbox for unread messages and displays a notification.
"""
try:
if len(list(self.reddit.get_unread(limit=1))) > 0:
show_notification(self.stdscr, ['New Messages'])
elif len(list(self.reddit.get_unread(limit=1))) == 0:
show_notification(self.stdscr, ['No New Messages'])
except praw.errors.LoginOrScopeRequired:
show_notification(self.stdscr, ['Not Logged In'])
def clear_input_queue(self): def clear_input_queue(self):
"Clear excessive input caused by the scroll wheel or holding down a key" """
Clear excessive input caused by the scroll wheel or holding down a key
"""
self.stdscr.nodelay(1) self.stdscr.nodelay(1)
while self.stdscr.getch() != -1: while self.stdscr.getch() != -1:
@@ -421,7 +436,9 @@ class BasePage(object):
self.stdscr.nodelay(0) self.stdscr.nodelay(0)
def logout(self): def logout(self):
"Prompt to log out of the user's account." """
Prompt to log out of the user's account.
"""
ch = prompt_input(self.stdscr, "Log out? (y/n):") ch = prompt_input(self.stdscr, "Log out? (y/n):")
if ch == 'y': if ch == 'y':