From 2358eed46ec09b59429c761bc105335e030991fb Mon Sep 17 00:00:00 2001
From: Gustavo Zambonin <stenny_gus@hotmail.com>
Date: Wed, 14 Oct 2015 00:39:47 -0300
Subject: [PATCH] page.py: exit prompt added logout prompt now works as
 intended

---
 rtv/oauth.py |  2 +-
 rtv/page.py  | 32 ++++++++++++++++----------------
 2 files changed, 17 insertions(+), 17 deletions(-)

diff --git a/rtv/oauth.py b/rtv/oauth.py
index 62aee86..6d73cc4 100644
--- a/rtv/oauth.py
+++ b/rtv/oauth.py
@@ -106,7 +106,7 @@ class OAuthTool(object):
         except (praw.errors.OAuthAppRequired, praw.errors.OAuthInvalidToken):
             show_notification(self.stdscr, ['Invalid OAuth data'])
         else:
-            message = ['Welcome {}!'.format(self.reddit.user.name)]
+            message = ['Welcome, {}!'.format(self.reddit.user.name)]
             show_notification(self.stdscr, message)
 
     def clear_oauth_data(self):
diff --git a/rtv/page.py b/rtv/page.py
index b4a4dde..396d5f7 100644
--- a/rtv/page.py
+++ b/rtv/page.py
@@ -265,7 +265,15 @@ class BasePage(object):
 
     @BaseController.register('q')
     def exit(self):
-        sys.exit()
+        """
+        Prompt to exit the application.
+        """
+
+        ch = prompt_input(self.stdscr, "Do you really want to quit? (y/n): ")
+        if ch == 'y':
+            sys.exit()
+        elif ch != 'n':
+            curses.flash()
 
     @BaseController.register('?')
     def help(self):
@@ -350,8 +358,12 @@ class BasePage(object):
         """
 
         if self.reddit.is_oauth_session():
-            self.oauth.clear_oauth_data()
-            show_notification(self.stdscr, ['Logged out'])
+            ch = prompt_input(self.stdscr, "Log out? (y/n): ")
+            if ch == 'y':
+                self.reddit.clear_authentication()
+                show_notification(self.stdscr, ['Logged out'])
+            elif ch != 'n':
+                curses.flash()
         else:
             self.oauth.authorize()
 
@@ -370,7 +382,7 @@ class BasePage(object):
             curses.flash()
             return
 
-        prompt = 'Are you sure you want to delete this? (y/n):'
+        prompt = 'Are you sure you want to delete this? (y/n): '
         char = prompt_input(self.stdscr, prompt)
         if char != 'y':
             show_notification(self.stdscr, ['Aborted'])
@@ -446,18 +458,6 @@ class BasePage(object):
             continue
         self.stdscr.nodelay(0)
 
-    def logout(self):
-        """
-        Prompt to log out of the user's account.
-        """
-
-        ch = prompt_input(self.stdscr, "Log out? (y/n):")
-        if ch == 'y':
-            self.reddit.clear_authentication()
-            show_notification(self.stdscr, ['Logged out'])
-        elif ch != 'n':
-            curses.flash()
-
     @property
     def safe_call(self):
         """