prompt to log out/in when logged in/out

rtv/curses_helpers.py

@@ -52,12 +52,14 @@ def show_notification(stdscr, message):
     for index, line in enumerate(message, start=1):
         window.addstr(index, 1, line)
     window.refresh()
-    stdscr.getch()
+    ch = stdscr.getch()
 
     window.clear()
     window = None
     stdscr.refresh()
 
+    return ch
+
 
 def show_help(stdscr):
     """

rtv/page.py

@@ -229,29 +229,36 @@ class BasePage(object):
 
     @BaseController.register('u')
     def login(self):
-        username = self.prompt_input('Enter username: ')
-        password = self.prompt_input('Enter password: ', hide=True)
+        """
+        Prompt to log out if logged in.
+        Prompt to log in if looged out.
+        """
 
-        if username == '' or username is None:
+        if self.reddit.is_logged_in():
+            ch = show_notification(self.stdscr, ["Log out? (y/N)"])
+
+            if ch == 121:  # 'y'
                 self.reddit.clear_authentication()
                 show_notification(self.stdscr,
                                   ['Logged out'])
-            return
-        elif password == '' or password is None:
+            else:
+                curses.flash()
+
+        else:
+            username = self.prompt_input('Enter username: ')
+            password = self.prompt_input('Enter password: ', hide=True)
+
+            if (username == '' or username is None) \
+               or (password == '' or password is None):
                 curses.flash()
                 return
 
             try:
-            curses.endwin()
-            print('Connecting...')
-            _reddit = praw.Reddit(user_agent=AGENT)
-            _reddit.login(username, password)
-            curses.doupdate()
+                self.reddit.login(username, password)
             except praw.errors.InvalidUserPass:
                 show_notification(self.stdscr,
                                   ['Invalid password for username: {}'.format(username)])
             else:
-            self.reddit.login(username, password)
                 show_notification(self.stdscr,
                                   ['Successfully logged in as: {}'.format(username)])