python 命令行解析

import argparse
from .client import client_main
from .server import server_main


def main():
    parser = argparse.ArgumentParser(prog="ok-tier", description="a connect app")
    subparsers = parser.add_subparsers(dest="command", required=True)
    # client
    client_parser = subparsers.add_parser("client", help="Run as client")
    client_parser.add_argument(
        "--server",
        type=str,
        default="tcp://127.0.0.1:3000",
        help="server address",
        dest="server",
    )
    # server
    server_parser = subparsers.add_parser("server", help="Run as server")
    server_parser.add_argument(
        "--listen",
        type=str,
        default="tcp://127.0.0.1:3000",
        help="listen address",
        dest="listen",
    )
    args = parser.parse_args()
    if args.command == "client":
        client_main(args.server)
    elif args.command == "server":
        server_main(args.listen)