close

Sử dụng argparse trong Python: Tham số dòng lệnh

Module argparse trong Python là một công cụ mạnh mẽ để xử lý và phân tích các tham số được truyền từ dòng lệnh khi chạy một chương trình. Nó giúp bạn dễ dàng định nghĩa các tham số và tùy chọn cần thiết cho chương trình của mình và cung cấp các cơ chế linh hoạt để đọc và sử dụng chúng.

Dưới đây là các bước để sử dụng module argparse:

  1. Nhập module argparse: Bắt đầu chương trình của bạn bằng việc nhập module argparse.

  2. Định nghĩa đối tượng ArgumentParser: Tạo một đối tượng ArgumentParser để định nghĩa các tham số và tùy chọn cho chương trình.

  3. Thêm các tham số: Sử dụng phương thức .add_argument() của đối tượng ArgumentParser để thêm các tham số và tùy chọn cần thiết cho chương trình. Mỗi tham số có thể có tên, kiểu dữ liệu, mô tả và nhiều thuộc tính khác.

  4. Phân tích tham số: Sử dụng phương thức .parse_args() của đối tượng ArgumentParser để phân tích các tham số từ dòng lệnh và lưu chúng vào một đối tượng.

  5. Sử dụng các tham số: Sử dụng các giá trị được lưu trữ trong đối tượng đã phân tích từ bước trước để thực hiện các hành động tương ứng với các tùy chọn được cung cấp từ dòng lệnh.

Ví dụ: Dưới đây là một ví dụ đơn giản về cách sử dụng argparse để tính tổng của hai số từ dòng lệnh:

import argparse

# Định nghĩa đối tượng ArgumentParser
parser = argparse.ArgumentParser(description='Calculate the sum of two numbers.')

# Thêm các tham số vào ArgumentParser
parser.add_argument('num1', type=int, help='First number')
parser.add_argument('num2', type=int, help='Second number')

# Phân tích các tham số từ dòng lệnh
args = parser.parse_args()

# Sử dụng các tham số để tính tổng
sum_result = args.num1 + args.num2
print(f'The sum is: {sum_result}')

Khi chạy chương trình với các tham số, ví dụ: python my_program.py 10 20, kết quả sẽ là: The sum is: 30, và nó sẽ hiển thị tổng của hai số đã được cung cấp từ dòng lệnh.