Switched to relative imports for python3.

main.py

@@ -2,10 +2,10 @@ import argparse
 import praw
 from requests.exceptions import ConnectionError
 
-from errors import SubmissionURLError, SubredditNameError
-from utils import curses_session
-from subreddit import SubredditPage
-from submission import SubmissionPage
+from rtv.errors import SubmissionURLError, SubredditNameError
+from rtv.utils import curses_session
+from rtv.subreddit import SubredditPage
+from rtv.submission import SubmissionPage
 
 parser = argparse.ArgumentParser(description='Reddit Terminal Viewer')
 parser.add_argument('-s', dest='subreddit', default='front', help='subreddit name')

rtv/__init__.py

@@ -1 +0,0 @@
-from main import main

rtv/content.py

@@ -5,7 +5,7 @@ from contextlib import contextmanager
 import praw
 import six
 
-from errors import SubmissionURLError, SubredditNameError
+from .errors import SubmissionURLError, SubredditNameError
 
 def clean(unicode_string):
     """
@@ -69,7 +69,7 @@ class BaseContent(object):
 
         while True:
             try:
-                yield self.get(index, n_cols)
+                yield self.get(index, n_cols=n_cols)
             except IndexError:
                 break
             index += step
@@ -231,7 +231,7 @@ class SubmissionContent(BaseContent):
 
         return data
 
-    def toggle(self, index):
+    def toggle(self, index, n_cols=70):
         """
         Toggle the state of the object at the given index.
 
@@ -244,7 +244,7 @@ class SubmissionContent(BaseContent):
         if data['type'] == 'Comment':
             cache = [data]
             count = 1
-            for d in self.iterate(index+1, 1):
+            for d in self.iterate(index+1, 1, n_cols):
                 if d['level'] <= data['level']:
                     break
 
@@ -256,7 +256,7 @@ class SubmissionContent(BaseContent):
             comment['cache'] = cache
             comment['count'] = count
             comment['level'] = data['level']
-            comment['body'] = 'Hidden [{}]'.format(count)
+            comment['body'] = 'Hidden'.format(count)
             self._comment_data[index:index+len(cache)] = [comment]
 
         elif data['type'] == 'HiddenComment':

rtv/page.py

@@ -1,6 +1,6 @@
 import curses
 
-from utils import Color
+from .utils import Color
 
 class Navigator(object):
     """

rtv/submission.py

@@ -3,9 +3,9 @@ import sys
 
 import six
 
-from content import SubmissionContent
-from page import BasePage
-from utils import LoadScreen, Color
+from .content import SubmissionContent
+from .page import BasePage
+from .utils import LoadScreen, Color
 
 class SubmissionPage(BasePage):
 

rtv/subreddit.py

@@ -1,11 +1,11 @@
 import curses
 import sys
 
-from errors import SubredditNameError
-from page import BasePage
-from submission import SubmissionPage
-from content import SubredditContent
-from utils import LoadScreen, text_input, display_message, Color
+from .errors import SubredditNameError
+from .page import BasePage
+from .submission import SubmissionPage
+from .content import SubredditContent
+from .utils import LoadScreen, text_input, display_message, Color
 
 class SubredditPage(BasePage):
 

rtv/utils.py

@@ -5,7 +5,7 @@ import threading
 from curses import textpad
 from contextlib import contextmanager
 
-from errors import EscapePressed
+from .errors import EscapePressed
 
 class Color(object):
 

setup.py

@@ -13,5 +13,5 @@ setup(
     keywords='reddit terminal praw',
     packages=['rtv'],
     install_requires=['praw', 'six'],
-    entry_points={'console_scripts': ['rtv=rtv:main']}
+    entry_points={'console_scripts': ['rtv=main:main']}
 )