135 lines
4.1 KiB
Python
135 lines
4.1 KiB
Python
import os
|
|
import sys
|
|
import argparse
|
|
import locale
|
|
import logging
|
|
|
|
import requests
|
|
import praw
|
|
import praw.errors
|
|
from six.moves import configparser
|
|
|
|
from . import config
|
|
from .exceptions import SubmissionError, SubredditError, ProgramError
|
|
from .curses_helpers import curses_session
|
|
from .submission import SubmissionPage
|
|
from .subreddit import SubredditPage
|
|
from .docs import *
|
|
from .__version__ import __version__
|
|
|
|
__all__ = []
|
|
|
|
def load_config():
|
|
"""
|
|
Search for a configuration file at the location ~/.rtv and attempt to load
|
|
saved settings for things like the username and password.
|
|
"""
|
|
|
|
config = configparser.ConfigParser()
|
|
|
|
HOME = os.path.expanduser('~')
|
|
XDG_CONFIG_HOME = os.getenv('XDG_CONFIG_HOME', os.path.join(HOME, '.config'))
|
|
config_paths = [
|
|
os.path.join(XDG_CONFIG_HOME, 'rtv', 'rtv.cfg'),
|
|
os.path.join(HOME, '.rtv')
|
|
]
|
|
|
|
# read only the first existing config file
|
|
for config_path in config_paths:
|
|
if os.path.exists(config_path):
|
|
config.read(config_path)
|
|
break
|
|
|
|
defaults = {}
|
|
if config.has_section('rtv'):
|
|
defaults = dict(config.items('rtv'))
|
|
|
|
if 'unicode' in defaults:
|
|
defaults['unicode'] = config.getboolean('rtv', 'unicode')
|
|
|
|
return defaults
|
|
|
|
|
|
def command_line():
|
|
|
|
parser = argparse.ArgumentParser(
|
|
prog='rtv', description=SUMMARY,
|
|
epilog=CONTROLS + HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter)
|
|
|
|
parser.add_argument('-s', dest='subreddit', help='subreddit name')
|
|
parser.add_argument('-l', dest='link', help='full link to a submission')
|
|
parser.add_argument('--unicode', help='enable unicode (experimental)')
|
|
parser.add_argument('--log', metavar='FILE', action='store',
|
|
help='Log HTTP requests')
|
|
|
|
group = parser.add_argument_group('authentication (optional)', AUTH)
|
|
group.add_argument('-u', dest='username', help='reddit username')
|
|
group.add_argument('-p', dest='password', help='reddit password')
|
|
|
|
args = parser.parse_args()
|
|
|
|
return args
|
|
|
|
|
|
def main():
|
|
"Main entry point"
|
|
|
|
# logging.basicConfig(level=logging.DEBUG, filename='rtv.log')
|
|
locale.setlocale(locale.LC_ALL, '')
|
|
|
|
args = command_line()
|
|
local_config = load_config()
|
|
|
|
# set the terminal title
|
|
title = 'rtv {0}'.format(__version__)
|
|
if os.name == 'nt':
|
|
os.system('title {0}'.format(title))
|
|
else:
|
|
sys.stdout.write("\x1b]2;{0}\x07".format(title))
|
|
|
|
# Fill in empty arguments with config file values. Paramaters explicitly
|
|
# typed on the command line will take priority over config file params.
|
|
for key, val in local_config.items():
|
|
if getattr(args, key, None) is None:
|
|
setattr(args, key, val)
|
|
|
|
config.unicode = False if args.unicode is None else args.unicode
|
|
|
|
# Squelch SSL warnings for Ubuntu
|
|
logging.captureWarnings(True)
|
|
if args.log:
|
|
logging.basicConfig(level=logging.DEBUG, filename=args.log)
|
|
|
|
try:
|
|
print('Connecting...')
|
|
reddit = praw.Reddit(user_agent=AGENT)
|
|
reddit.config.decode_html_entities = False
|
|
if args.username:
|
|
# PRAW will prompt for password if it is None
|
|
reddit.login(args.username, args.password)
|
|
with curses_session() as stdscr:
|
|
if args.link:
|
|
page = SubmissionPage(stdscr, reddit, url=args.link)
|
|
page.loop()
|
|
subreddit = args.subreddit or 'front'
|
|
page = SubredditPage(stdscr, reddit, subreddit)
|
|
page.loop()
|
|
except praw.errors.InvalidUserPass:
|
|
print('Invalid password for username: {}'.format(args.username))
|
|
except requests.ConnectionError:
|
|
print('Connection timeout')
|
|
except requests.HTTPError:
|
|
print('HTTP Error: 404 Not Found')
|
|
except SubmissionError as e:
|
|
print('Could not reach submission URL: {}'.format(e.url))
|
|
except SubredditError as e:
|
|
print('Could not reach subreddit: {}'.format(e.name))
|
|
except ProgramError as e:
|
|
print('Error: could not open file with program "{}", '
|
|
'try setting RTV_EDITOR'.format(e.name))
|
|
except KeyboardInterrupt:
|
|
return
|
|
|
|
sys.exit(main())
|