__main__.py 文件源码

def main():
    ap = argparse.ArgumentParser(
        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    ap.add_argument("-u", "--udid", required=False, help="iPhone udid")

    subp = ap.add_subparsers()

    @contextmanager
    def add_parser(name):
        yield subp.add_parser(name, formatter_class=argparse.ArgumentDefaultsHelpFormatter)

    with add_parser('developer') as p:
        p.set_defaults(func=load_main('atx.cmds.iosdeveloper'))

    with add_parser('screencap') as p:
        p.add_argument('-o', '--output', default='screenshot.png', help='take iPhone screenshot')
        p.add_argument('-r', '--rotate', type=int, choices=[0, 90, 180, 270], default=0, help='screen rotation')
        p.set_defaults(func=_screencap)

    args = ap.parse_args()
    args.func(args)