乐趣区

关于python:Python-Cli-编写指南

jiezi

Python Cli 编写指南

python 实现 cli

环境：

  • python 3.8

库

  • python 自带 argparse

指南

简略示例 : cli.py

import argparse


def cli():
    parser = argparse.ArgumentParser(description='CLI 形容')
    subparsers = parser.add_subparsers(metavar='子命令')

    # 增加子命令，演示没有参数
    one_parser = subparsers.add_parser('one', help='第一个命令')
    one_parser.set_defaults(handle=handle_one)

    # 解析命令
    args = parser.parse_args()
    # 1. 第一个命令会解析成 handle，应用 args.handle() 就可能调用
    if hasattr(args, 'handle'):
        args.handle(args)
    # 2. 如果没有 handle 属性，则示意未输出子命令，则打印帮忙信息
    else:
        parser.print_help()


def handle_one(args):
    print('handle_one')


if __name__ == '__main__':
    cli()

调用：

$ python cli.py
usage: cli_1.py [-h] 子命令 ...

CLI 形容

positional arguments:
  子命令
    one       第一个命令

optional arguments:
  -h, --help  show this help message and exit

$ python cli.py one
handle_one

减少第二个命令 

import argparse


def cli():
    parser = argparse.ArgumentParser(description='CLI 形容')
    subparsers = parser.add_subparsers(metavar='子命令')

    # 增加子命令，演示没有参数
    one_parser = subparsers.add_parser('one', help='第一个命令')
    one_parser.set_defaults(handle=handle_one)

    # 增加子命令，演示有参数
    two_parser = subparsers.add_parser('two', help='第二个命令')
    # 参数 (简写，全称，类型，是否必填，帮忙阐明)
    two_parser.add_argument('-s', '--str', type=str, required=True,
                            help='一个字符串类型参数')
    # 参数 (简写，全称，类型，默认值，帮忙阐明)
    two_parser.add_argument('-d', '--default', type=str, default='默认值',
                            help='这个命令有默认值')
    # 参数 (简写，全称，类型，帮忙阐明)
    two_parser.add_argument('-ts', '--the-str', type=str,
                            help='当全称有横线时，属性名转换为下划线，即 the_str')
    two_parser.set_defaults(handle=handle_two)

    # 解析命令
    args = parser.parse_args()
    # 1. 第一个命令会解析成 handle，应用 args.handle() 就可能调用
    if hasattr(args, 'handle'):
        # 1.1. 其余参数会被解析成 args 的属性，以命令全称为属性名
        args.handle(args)
    # 2. 如果没有 handle 属性，则示意未输出子命令
    else:
        parser.print_help()


def handle_one(args):
    # 无参数
    print('handle_one')


def handle_two(args):
    print('handle_two')
    print(f'str:{args.str}')
    print(f'default:{args.default}')
    print(f'the-str:{args.the_str}')


if __name__ == '__main__':
    cli()

  • 调用：

    $ python cli.py
usage: cli.py [-h] 子命令 ...

CLI 形容

positional arguments:
子命令
  one       第一个命令
  two       第二个命令

optional arguments:
-h, --help  show this help message and exit

  • 调用 one 不受影响 

    $ python cli.py one
handle_one

  • 调用 two 会显示须要必填项 -s

    $ python cli.py two
usage: cli.py two [-h] -s STR [-d DEFAULT] [-ts THE_STR]
cli.py two: error: the following arguments are required: -s/--str

  • 输出 - h 查看帮忙 

    $ python cli.py two -h
usage: cli.py two [-h] -s STR [-d DEFAULT] [-ts THE_STR]

optional arguments:
-h, --help            show this help message and exit
-s STR, --str STR     一个字符串类型参数
-d DEFAULT, --default DEFAULT
                      这个命令有默认值
-ts THE_STR, --the-str THE_STR
                      当全称有横线时，属性名转换为下划线，即 the_str

  • 输出 - s 参数（或全称 –str）

    • 能够看到加了默认值的会显示默认值
    • 没默认值且非必填项值为 None
    $ python cli.py two -s 哈哈
handle_two
str: 哈哈
default: 默认值
the-str:None

  • 输出有默认值的参数 -d (或全称 –default)

    • 输出的值会笼罩默认值 
    $ python cli.py two -s 哈哈 --default 改了
handle_two
str: 哈哈
default: 改了
the-str:None

  • 输出所有参数 

    $ python cli.py two -s 哈哈 --default 改了 -ts 赋值了
handle_two
str: 哈哈
default: 改了
the-str: 赋值了

其余代码应用

在 ”setup.py” 文件中 

import setuptools

setuptools.setup(
  # 省略其余参数
  entry_points={
    'console_scripts': ['my-cli = my.cli:cli']
  }
)

  • ‘my-cli = my.cli:cli’ 就是替换内容

    • ‘my-cli’

      • 上传 pypi 后，拉取这个库，就可能应用 ’my-cli’ 来调用 cli
      • 如 ’my-cli one’ / ‘my-cli two -h’

    • ‘my.cli:cli’

      • ‘ 残缺类名: 办法名 ’
      • 目录构造如下方构造
      • cli 命令须要放在模块下
      • 不能放在根目录下，否则会报错找不到模块

文件目录构造：

root

  • my

    • \_\_init\_\_.py
    • cli.py

参考资料：

https://zhuanlan.zhihu.com/p/…

退出移动版