#! /usr/bin/env python3
"""Script to get a key from the keyboard"""

from __future__ import print_function
import sys


from pysyte.oss import getch
from pysyte.cli.main import run


__version__ = '0.1.1'


class ScriptError(NotImplementedError):
    pass


def add_args(parser):
    p = parser
    p.boolean('-c', '--codes', help='Show raw codes')
    p.boolean('-p', '--prompt', default='',
       help='Prompt to show before getting keys ' '(default: none)')
    p.boolean('-s', '--string', help='Show string')


def script(args):
    if args.codes:
        print(', '.join(getch.get_codes()))
    elif args.string:
        print(getch.get_string())
    else:
        print(getch.get_key())
    return True


run(script, add_args)
