python如何使用linux命令行参数

Python 使用 Linux 命令行参数的方法有：argparse 模块、sys.argv 模块、subprocess 模块。 其中，argparse 模块是推荐使用的方法，因为它功能强大且易于使用。下面，我将详细描述如何使用 argparse 模块来解析命令行参数。

一、ARGPARSE 模块

argparse 模块是 Python 内置的命令行解析模块，它可以帮助我们轻松解析命令行参数，并生成帮助信息。以下是一个使用 argparse 模块的示例：

import argparse
def main():
    parser = argparse.ArgumentParser(description='This is a sample Python script using argparse.')
    parser.add_argument('--name', type=str, required=True, help='Your name')
    parser.add_argument('--age', type=int, required=True, help='Your age')
    parser.add_argument('--city', type=str, help='Your city')
    args = parser.parse_args()
    print(f'Hello, {args.name}!')
    print(f'You are {args.age} years old.')
    if args.city:
        print(f'You live in {args.city}.')
if __name__ == '__main__':
    main()

在这个示例中，我们定义了三个命令行参数：--name、--age 和 --city。其中，--name 和 --age 是必需参数，而 --city 是可选参数。运行这个脚本时，可以使用以下命令：

python script.py --name Alice --age 30 --city London

这将输出：

Hello, Alice! You are 30 years old. You live in London.

二、SYS.ARGV 模块

sys.argv 是 Python 的标准库模块，用于获取命令行参数。它是一个列表，其中包含所有传递给脚本的命令行参数。第一个元素（索引为 0）是脚本名称，后面的元素是传递给脚本的参数。

以下是一个使用 sys.argv 的示例：

import sys
def main():
    if len(sys.argv) < 3:
        print('Usage: python script.py <name> <age> [city]')
        sys.exit(1)
    name = sys.argv[1]
    age = int(sys.argv[2])
    city = sys.argv[3] if len(sys.argv) > 3 else None
    print(f'Hello, {name}!')
    print(f'You are {age} years old.')
    if city:
        print(f'You live in {city}.')
if __name__ == '__main__':
    main()

在这个示例中，我们手动解析命令行参数。运行这个脚本时，可以使用以下命令：

python script.py Alice 30 London

这将输出：

Hello, Alice! You are 30 years old. You live in London.

三、SUBPROCESS 模块

subprocess 模块允许我们在 Python 中执行外部命令。我们可以使用 subprocess 模块来运行 Linux 命令，并将其输出作为参数传递给 Python 脚本。

以下是一个使用 subprocess 模块的示例：

import subprocess
def main():
    result = subprocess.run(['ls', '-l'], capture_output=True, text=True)
    print('Output of "ls -l":')
    print(result.stdout)
if __name__ == '__main__':
    main()

在这个示例中，我们使用 subprocess.run() 函数来执行 ls -l 命令，并将其输出捕获到 result.stdout 中。运行这个脚本时，它将输出 ls -l 命令的结果。

四、ARGPARSE、SYS.ARGV 和 SUBPROCESS 的对比

argparse 模块适用于需要解析复杂命令行参数的场景。它提供了丰富的功能，如参数类型转换、默认值、必需参数、可选参数和生成帮助信息等。
sys.argv 模块适用于简单的命令行参数解析。它非常灵活，但需要手动处理参数解析和错误处理。
subprocess 模块适用于需要在 Python 脚本中执行外部命令的场景。它可以捕获外部命令的输出，并将其作为参数传递给 Python 脚本。

五、ARGPARSE 模块的高级用法

除了基本用法，argparse 模块还有许多高级功能，如子命令、互斥选项和默认值等。

子命令

子命令是 argparse 模块的一个强大功能，允许我们定义多个子命令，每个子命令有自己的一组参数。以下是一个示例：

import argparse
def greet(args):
    print(f'Hello, {args.name}!')
def farewell(args):
    print(f'Goodbye, {args.name}!')
def main():
    parser = argparse.ArgumentParser(description='A script with subcommands.')
    subparsers = parser.add_subparsers()
    greet_parser = subparsers.add_parser('greet', help='Greet someone')
    greet_parser.add_argument('name', type=str, help='The name of the person to greet')
    greet_parser.set_defaults(func=greet)
    farewell_parser = subparsers.add_parser('farewell', help='Bid farewell to someone')
    farewell_parser.add_argument('name', type=str, help='The name of the person to bid farewell to')
    farewell_parser.set_defaults(func=farewell)
    args = parser.parse_args()
    args.func(args)
if __name__ == '__main__':
    main()

在这个示例中，我们定义了两个子命令：greet 和 farewell，每个子命令有自己的一组参数。运行这个脚本时，可以使用以下命令：

python script.py greet Alice python script.py farewell Alice

这将分别输出：

Hello, Alice!

Goodbye, Alice!

互斥选项

互斥选项允许我们定义一组参数，这些参数在命令行中是互斥的，即只能指定其中一个。以下是一个示例：

import argparse
def main():
    parser = argparse.ArgumentParser(description='A script with mutually exclusive options.')
    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument('--verbose', action='store_true', help='Enable verbose mode')
    group.add_argument('--quiet', action='store_true', help='Enable quiet mode')
    args = parser.parse_args()
    if args.verbose:
        print('Verbose mode enabled.')
    elif args.quiet:
        print('Quiet mode enabled.')
if __name__ == '__main__':
    main()

在这个示例中，我们定义了两个互斥选项：--verbose 和 --quiet。运行这个脚本时，可以使用以下命令：

python script.py --verbose python script.py --quiet

这将分别输出：

Verbose mode enabled.

Quiet mode enabled.

默认值

argparse 模块允许我们为参数指定默认值，如果在命令行中未提供该参数，则使用默认值。以下是一个示例：

import argparse
def main():
    parser = argparse.ArgumentParser(description='A script with default values.')
    parser.add_argument('--name', type=str, default='World', help='Your name')
    args = parser.parse_args()
    print(f'Hello, {args.name}!')
if __name__ == '__main__':
    main()

在这个示例中，我们为 --name 参数指定了默认值 World。运行这个脚本时，如果未提供 --name 参数，它将使用默认值：

python script.py

这将输出：

Hello, World!

当然，如果提供了 --name 参数，它将使用提供的值：

python script.py --name Alice

这将输出：

Hello, Alice!

六、总结

Python 提供了多种解析命令行参数的方法，其中 argparse 模块 是推荐的选择，因为它功能强大且易于使用。sys.argv 模块 适用于简单的命令行参数解析，而 subprocess 模块 则适用于在 Python 脚本中执行外部命令。通过学习和掌握这些方法，我们可以编写更灵活和功能丰富的 Python 脚本。

无论是简单的命令行工具，还是复杂的命令行应用程序，argparse 模块都能满足我们的需求。希望这篇文章能帮助你更好地理解和使用 Python 的命令行参数解析功能。