_test_v0_6.py

import typed_args as ta
from icecream import ic
import logging
import pickle


@ta.dataclass
class Group1:
    x2: str = ta.add_argument('--x2')
    x3: str = ta.add_argument('--x3')


@ta.dataclass
class Group2:
    x4: str = ta.add_argument('--x4')
    """
    x4 in group2
    """

    x5: str = ta.add_argument('--x5')
    """x5 in group2"""


@ta.dataclass
class Cmd1:
    x6: str = ta.add_argument('--x6')
    x7: str = ta.add_argument('--x7')


@ta.dataclass
class Cmd2:
    x6: str = ta.add_argument('--x6')
    x7: str = ta.add_argument('--x7')
    x8: str = ta.add_argument('--x8')


class SubCommands(ta.SubcommandEnum):
    CMD1: Cmd1 = ta.add_parser('cmd1')
    CMD2: Cmd2 = ta.add_parser('cmd2')


@ta.argument_parser(prog='dddd')
class Args(ta.TypedArgs):
    x1: str = ta.add_argument()
    group1: Group1 = ta.add_argument_group()
    group2: Group2 = ta.add_argument_group()
    sub_cmd: SubCommands = ta.add_subparsers()
    x9: int = ta.add_argument('--x9', type=int)


def test():
    logging.basicConfig(level=logging.DEBUG)

    args = Args.parse_args()

    ic(args)
    args.group1.x2

    pickle.dumps(args)


if __name__ == '__main__':
    test()