ct.py

# -*- coding:utf-8 -*-

from argparse import ArgumentParser
from ComicThief.main import ComicThief

OPTIONS = [
    (('-s', '--search'), {'help': 'search'}),
    (('-xs', '--xsearch'), {'help': 'exact search to narrow more than one result'}),
    (('-e', '--episode'),  {'help': 'choose episode', 'type': str})
]


def add_arguments(parser):
    for option, kwargs in OPTIONS:
        parser.add_argument(*option, **kwargs)


def download_episode(episode):
    episode_url = result.get(episode)
    if episode_url:
        ct.download_episode(episode_url, episode)

if __name__ == '__main__':
    ct = ComicThief()
    parser = ArgumentParser()
    add_arguments(parser)
    args = parser.parse_args()
    if args.search:
        print('Searching..')
        results = ct.search(args.search)
        if results == 1 and args.episode:
            download_episode(args.episode)
    elif args.xsearch:
        result = ct.exact_search(args.xsearch)
        if args.episode:
            download_episode(args.episode)
    else:
        print('Use -s (normal search) or -xs (exact search).')