argparse

#!/usr/bin/env python3


import argparse


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='test....')
    parser.add_argument('-p', '--pdb', required=True, type=str,
                        help='the pdb id of protein')
    parser.add_argument('-c', '--chain', default='A', type=str,
                        help='the chain of protein, Default: %(default)s')
    parser.add_argument('--num', required=True, type=int,
                        help='the index of atom')
    args = parser.parse_args()


    print(f"pdb id is {args.pdb}")
    print(f"the chain of protein is {args.chain}")
    print(f"the index of atom is {args.num}")