python argparse模块使用

import argparse
def main():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--data_dir',
        type=str,
        help='Directory for storing input data')
    parser.add_argument('integer', type = int, help = 'display an integer')
    FLAGS, unparsed = parser.parse_known_args()
    #args = parser.parse_args()
    if FLAGS.data_dir:
        print(FLAGS.data_dir)
    if FLAGS.integer:
        print(FLAGS.integer)
    if unparsed:
        print(unparsed)
if __name__ == '__main__':
    main()