Moved main back out of package. Still looking for a better solution.

This commit is contained in:
Michael Lazar
2015-02-11 19:29:37 -08:00
parent a980e6b4c3
commit c2dc8df29f
2 changed files with 6 additions and 6 deletions

View File

@@ -1,51 +0,0 @@
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
def main():
parser = argparse.ArgumentParser(description='Reddit Terminal Viewer')
parser.add_argument('-s', dest='subreddit', default='front', help='subreddit name')
parser.add_argument('-l', dest='link', help='full link to a submission')
group = parser.add_argument_group('authentication (optional)')
group.add_argument('-u', dest='username', help='reddit username')
group.add_argument('-p', dest='password', help='reddit password')
args = parser.parse_args()
try:
reddit = praw.Reddit(user_agent='reddit terminal viewer v0.0')
reddit.config.decode_html_entities = True
if args.username and args.password:
reddit.login(args.username, args.password)
with curses_session() as stdscr:
if args.link:
page = SubmissionPage(stdscr, reddit, url=args.link)
page.loop()
page = SubredditPage(stdscr, reddit, args.subreddit)
page.loop()
except KeyboardInterrupt:
return
except ConnectionError:
print('Timeout: Could not connect to website')
except SubmissionURLError as e:
print('Could not reach submission URL: {}'.format(e.url))
except SubredditNameError as e:
print('Could not reach subreddit: {}'.format(e.name))
if __name__ == '__main__':
main()