cheat.sh/lib/standalone.py

"""
Standalone wrapper for the cheat.sh server.
"""

import sys
import textwrap

import cheat_wrapper

def show_usage():
    """
    Show how to use the program in the standalone mode
    """

    print textwrap.dedent("""
        Usage:

            lib/standalone.py [OPTIONS] QUERY

        For OPTIONS see :help
    """)[1:-1]

def parse_cmdline(args):
    """
    Parses command line arguments and returns
    query and request_options
    """

    if not args:
        show_usage()
        sys.exit(0)

    request_options = {}
    query = args[0]
    query = query.lstrip("/")
    if not query:
        query = ":firstpage"
    return query, request_options


def main(args):
    """
    standalone wrapper for cheat_wrapper()
    """

    query, request_options = parse_cmdline(args)
    answer, _ = cheat_wrapper.cheat_wrapper(query, request_options=request_options)
    sys.stdout.write(answer.encode("utf-8"))

if __name__ == '__main__':
    main(sys.argv[1:])