如何使用python自定义命令

dir、tree、cd等等，都是我们常见的命令。这些命令是开发者开发出来的，如果我们自己想按照自己的想法开发一个命令，应该怎么做呢？

以python语言来实现，问题就是：如何使用python自定义命令？

需求场景如下：

1、cmd007   作为自定义命令名称

2、cmd007  hello  --name=Lily    作为自定义功能1的名称hello打招呼，输出：Hi, Lily

3、cmd007  eat   --food=fish        作为自定义功能2的名称food吃东西，输出：I  like  to  eat  fish

 

（一）目录结构如下：

cmd007       #  文件夹必须和自定义命令同名

│ __init__.py
│ __main__.py

 

（二）实现自定义命令和功能1

在__main__.py中

from  __future__ import unicode_literals
import click

import os
import sys

__version__ = "1.1.dev2"


def verbose_option(f):
    def callback(ctx, param, value):
        pass

    return click.option("-v", "--version",
                        is_flag=True,
                        expose_value=False,
                        help="Enable verbose output",
                        callback=callback)(f)


def quiet_option(f):
    def callback(ctx, param, value):
        pass

    return click.option('-q', '--quiet',
                        is_flag=True,
                        expose_value=False,
                        help='Silence warnings',
                        callback=callback)(f)


def common_options(f):
    f = verbose_option(f)
    f = quiet_option(f)
    return f


pgk_dir = os.path.dirname(os.path.abspath(__file__))


@click.group(context_settings={'help_option_names': ['-h', '--help']})
@click.version_option(
    '{0} from {1} (Python {2})'.format(__version__, pgk_dir, sys.version[:3]),
    '-V', '--version')
@common_options
def cli():
    """
    cmd007 - my cmd named cmd007
    """


# 自定义功能：hello
@cli.command(name="hello")
@click.option('-n', '--name', help="please input name")
@click.option('-c', '--count', help="print times", default=1)
def hello_command(name, count):
    for x in range(count):
        print("Hi, {0}".format(name))


if __name__ == '__main__':
    cli()

来到cmd007的上级目录下，调出命令行客户端，执行命名：

python  cmd007

python   cmd007  -V

python   cmd007  hello  --name=Lily  --count=3

python   cmd007  hello  -n  Lilei  -c  2

 分别输出： 

提示：如果将这个python包cmd007打包发布，然后安装到python中去，那么每行命令就不用输入python关键字了。

 

 （三）实现自定义命令、功能1、功能2

 在__main__.py中增加代码

# 自定义功能2：eat
@cli.command(name="eat")
@click.option('-f', '--food', help="please input your like food")
@click.option('-a', '--and-you', help="ask echo other", default=False)
def hello_command(food, and_you):
    print("I like to eat {0}".format(food))
    if and_you in ['t', 'T', 'true', 'True', True, 1, "1"]:
        print("And what is your favorite food ?")

　

完整版：

from  __future__ import unicode_literals
import click

import os
import sys

__version__ = "1.1.dev2"


def verbose_option(f):
    def callback(ctx, param, value):
        pass

    return click.option("-v", "--version",
                        is_flag=True,
                        expose_value=False,
                        help="Enable verbose output",
                        callback=callback)(f)


def quiet_option(f):
    def callback(ctx, param, value):
        pass

    return click.option('-q', '--quiet',
                        is_flag=True,
                        expose_value=False,
                        help='Silence warnings',
                        callback=callback)(f)


def common_options(f):
    f = verbose_option(f)
    f = quiet_option(f)
    return f


pgk_dir = os.path.dirname(os.path.abspath(__file__))


@click.group(context_settings={'help_option_names': ['-h', '--help']})
@click.version_option(
    '{0} from {1} (Python {2})'.format(__version__, pgk_dir, sys.version[:3]),
    '-V', '--version')
@common_options
def cli():
    """
    cmd007 - my cmd named cmd007
    """


# 自定义功能1：hello
@cli.command(name="hello")
@click.option('-n', '--name', help="please input name")
@click.option('-c', '--count', help="print times", default=1)
def hello_command(name, count):
    for x in range(count):
        print("Hi, {0}".format(name))


# 自定义功能2：eat
@cli.command(name="eat")
@click.option('-f', '--food', help="please input your like food")
@click.option('-a', '--and-you', help="ask echo other", default=False)
def hello_command(food, and_you):
    print("I like to eat {0}".format(food))
    if and_you in ['t', 'T', 'true', 'True', True, 1, "1"]:
        print("And what is your favorite food ?")


if __name__ == '__main__':
    cli()

 

执行

python   cmd007  eat  -f  fish  --and-you=1

输出

 

延伸：

@click.option()中的参数详解：https://blog.csdn.net/weixin_31866177/article/details/82111586

每个参数解释如下:

• name or flags - 选项字符串的名字或者列表，例如 foo 或者 -f, --foo。
• action - 命令行遇到参数时的动作，默认值是 store。
  • store_const，表示赋值为const；
  • append，将遇到的值存储成列表，也就是如果参数重复则会保存多个值;
  • append_const，将参数规范中定义的一个值保存到一个列表；
  • count，存储遇到的次数；此外，也可以继承 argparse.Action 自定义参数解析；
• nargs - 应该读取的命令行参数个数，可以是具体的数字，或者是?号，当不指定值时对于 Positional argument 使用 default，对于 Optional argument 使用 const；或者是 * 号，表示 0 或多个参数；或者是 + 号表示 1 或多个参数。
• const - action 和 nargs 所需要的常量值。
• default - 不指定参数时的默认值。
• type - 命令行参数应该被转换成的类型。
• choices - 参数可允许的值的一个容器。
• required - 可选参数是否可以省略 (仅针对可选参数)。
• help - 参数的帮助信息，当指定为 argparse.SUPPRESS 时表示不显示该参数的帮助信息.
• metavar - 在 usage 说明中的参数名称，对于必选参数默认就是参数名称，对于可选参数默认是全大写的参数名称.
• dest - 解析后的参数名称，默认情况下，对于可选参数选取最长的名称，中划线转换为下划线.