Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách thêm đối số dòng lệnh trong Python


Module argparse giúp phân tích đối số dòng lệnh (command line arguments).

Cơ bản

  • Định nghĩa argument bằng parser.add_argument().
  • Parse: args = parser.parse_args().
  • Sử dụng: args.param.

Ví dụ 1: 1 đối số positional (string)

import argparse
def get_args():
    parser = argparse.ArgumentParser(description='Example', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('player', metavar='player', type=str, help='Tennis Player')
    return parser.parse_args()

def main(player):
    print(f"The {player} had won 20 grandslam titles.")

if __name__ == '__main__':
    args = get_args()
    main(args.player)

Chạy: python script.py federer → The federer had won 20 grandslam titles.

Ví dụ 2: 2 đối số (string, int)

parser.add_argument('player', type=str, help='Tennis Player')
parser.add_argument('titles', type=int, help='Grandslam Titles')
# Chạy: python script.py federer 20

Kết luận

argparse tự sinh help, validate type, bắt buộc/không bắt buộc.