from argparse import ArgumentParser

from .robots_loader import ROBOTS, load

ROBOTS = sorted(ROBOTS.keys())

parser = ArgumentParser(description=load.__doc__)
parser.add_argument("robot", nargs="?", default=ROBOTS[0], choices=ROBOTS)
parser.add_argument("--no-display", action="store_true")

args = parser.parse_args()

load(args.robot, display=not args.no_display)
