diff --git a/leech.py b/leech.py index 25deab9..91b14b8 100755 --- a/leech.py +++ b/leech.py @@ -17,6 +17,7 @@ USER_AGENT = 'Leech/%s +http://davidlynch.org' % __version__ logger = logging.getLogger(__name__) + def configure_logging(verbose): if verbose: logging.basicConfig( @@ -29,6 +30,7 @@ def configure_logging(verbose): format="[%(name)s] %(message)s" ) + def create_session(cache): if cache: session = requests_cache.CachedSession('leech', expire_after=4 * 3600) @@ -46,6 +48,7 @@ def create_session(cache): }) return session + def open_story(url, session, site_options): site, url = sites.get(url) @@ -84,6 +87,7 @@ def open_story(url, session, site_options): raise Exception("Couldn't extract story") return story + @click.group(cls=DefaultGroup, default='download', default_if_no_args=True) def cli(): """Top level click group. Uses click-default-group to preserve most behavior from leech v1."""