Added username to banner if logged in.

rtv/page.py

@@ -99,9 +99,10 @@ class BasePage(object):
     MIN_HEIGHT = 10
     MIN_WIDTH = 20
 
-    def __init__(self, stdscr, content, **kwargs):
+    def __init__(self, stdscr, reddit, content, **kwargs):
 
         self.stdscr = stdscr
+        self.reddit = reddit
         self.content = content
         self.nav = Navigator(self.content.get, **kwargs)
 
@@ -155,6 +156,14 @@ class BasePage(object):
         attr = curses.A_REVERSE | curses.A_BOLD | Color.CYAN
         self._header_window.bkgd(' ', attr)
         self._header_window.addnstr(0, 0, self.content.name, n_cols-1)
+
+        if self.reddit.user is not None:
+            username = self.reddit.user.name
+            s_col = (n_cols - len(username) - 1)
+            # Only print the username if it fits in the empty space on the right
+            if (s_col - 1) >= len(self.content.name):
+                self._header_window.addnstr(0, s_col, username, (n_cols-s_col-1))
+
         self._header_window.refresh()
 
     def _draw_content(self):

rtv/submission.py

@@ -21,7 +21,7 @@ class SubmissionPage(BasePage):
         else:
             raise ValueError('Must specify url or submission')
 
-        super(SubmissionPage, self).__init__(stdscr, content, page_index=-1)
+        super(SubmissionPage, self).__init__(stdscr, reddit, content, page_index=-1)
 
     def loop(self):
 

rtv/subreddit.py

@@ -17,11 +17,10 @@ class SubredditPage(BasePage):
 
     def __init__(self, stdscr, reddit, name):
 
-        self.reddit = reddit
         self.loader = LoadScreen(stdscr)
 
         content = SubredditContent.from_name(reddit, name, self.loader)
-        super(SubredditPage, self).__init__(stdscr, content)
+        super(SubredditPage, self).__init__(stdscr, reddit, content)
 
     def loop(self):