使用click进行python下命令行工具的编写以及编码问题解决

用原生的参数解析工具非常繁琐,而click工具入手就非常简单了,而且直观。建立以下这个程序a.py[1], cli会有子命令make,make有选项--work:

  1 import click
  2 
  3 @click.group()
  4 def cli():
  5     print("cli")
  6 
  7 @cli.command()
  8 @click.option('--work', default=None)
  9 def make(work):
 10     print(type(work))
 11     print(work)
 12 
 13 if __name__ == '__main__':
 14     cli()

a.py在bash下执行:

$ python a.py
Usage: a.py [OPTIONS] COMMAND [ARGS]...

Options:
  --help  Show this message and exit.

Commands:
  make

在bash下执行:

$ python a.py make
cli
<type 'NoneType'>
None

a.py在bash下执行:

$ python a.py make --work='我'
cli
<type 'unicode'>
我

a.py修改, 增加make的参数name:

  7 @cli.command()
  8 @click.option('--work', default=None)
  9 @click.argument('name')
 10 def make(work, name):
 11     print(u'work: {}, type: {}'.format(work, type(work)))
 12     print(u'name: {}'.format(name))

在bash下执行:

$ python a.py make --work='我'
cli
Usage: a.py make [OPTIONS] NAME
Try "a.py make --help" for help.

Error: Missing argument "NAME".

注意执行时提示:Error: Missing argument "NAME"。 表示需要参数NAME。我们重新执行:

$ python a.py make --work='我' '张三'
cli
work: 我, type: <type 'unicode'>
name: 张三

可以用click.Choice限定参数或者选项的取值范围,修改代码:

  7 @cli.command()
  8 @click.option('--work', default=None, type=click.Choice(['全职', '兼职']))
  9 @click.argument('name')
 10 def make(work, name):
 11     print(u'work: {}, type: {}'.format(work, type(work)))
 12     print(u'name: {}'.format(name))

我们重新执行:

$ python a.py make --work='全' '张三'
cli
/home/jzhou/env/wf/lib/python2.7/site-packages/click/types.py:156: UnicodeWarning: Unicode equal comparison failed to convert both arguments to Unicode - interpreting them as being unequal
 if value in self.choices:
/home/jzhou/env/wf/lib/python2.7/site-packages/click/types.py:176: UnicodeWarning: Unicode equal comparison failed to convert both arguments to Unicode - interpreting them as being unequal
 if normed_value in normed_choices:
Traceback (most recent call last):
 File "a.py", line 16, in <module>
   cli()
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 764, in __call__
   return self.main(*args, **kwargs)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 717, in main
   rv = self.invoke(ctx)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 1135, in invoke
   sub_ctx = cmd.make_context(cmd_name, args, parent=ctx)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 641, in make_context
   self.parse_args(ctx, args)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 940, in parse_args
   value, args = param.handle_parse_result(ctx, opts, args)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 1469, in handle_parse_result
   value = self.full_process_value(ctx, value)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 1790, in full_process_value
   return Parameter.full_process_value(self, ctx, value)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 1438, in full_process_value
   value = self.process_value(ctx, value)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 1428, in process_value
   return self.type_cast_value(ctx, value)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 1417, in type_cast_value
   return _convert(value, (self.nargs != 1) + bool(self.multiple))
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/core.py", line 1415, in _convert
   return self.type(value, self, ctx)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/types.py", line 39, in __call__
   return self.convert(value, param, ctx)
 File "/home/jzhou/env/wf/lib/python2.7/site-packages/click/types.py", line 180, in convert
   (value, ', '.join(self.choices)), param, ctx)
UnicodeDecodeError: 'ascii' codec can't decode byte 0xe5 in position 16: ordinal not in range(128)

这个是因为click在python 2环境下没有考虑把参数或者选项转化为unicode,下载click_patch[2], clone后在目录下安装:

$ git clone https://github.com/fininfotech/click_patch.git
Cloning into 'click_patch'...
remote: Enumerating objects: 11, done.
remote: Counting objects: 100% (11/11), done.
remote: Compressing objects: 100% (8/8), done.
remote: Total 11 (delta 1), reused 8 (delta 1), pack-reused 0
Unpacking objects: 100% (11/11), done.
$ cd click_patch
$ python setup.py install
running install
....

修改a.py:

  1 # -*- coding:utf-8 -*-
  2 import click
  3 import click_patch
  4 
  5 @click.group()
  6 def cli():
  7     print("cli")
  8 
  9 @cli.command()
 10 @click.option('--work', default=None, type=click.Choice([u'全职', u'兼职']))
 11 @click.argument('name')
 12 def make(work, name):
 13     print(u'work: {}, type: {}'.format(work, type(work)))
 14     print(u'name: {}'.format(name))
 15 
 16 if __name__ == '__main__':
 17     cli()

继续执行:

$ python a.py make --work='全' '张三'
cli
Usage: a.py make [OPTIONS] NAME
Try "a.py make --help" for help.

Error: Invalid value for "--work": invalid choice: 全. (choose from 全职, 兼职)
$ python a.py make --work='全职' '张三'
cli
work: 全职, type: <type 'unicode'>
name: 张三

总的来说,click用来做命令行工具非常方便,但是在python 2.7环境下还存在一些字符编码上的问题,可以使用click_patch来改正。


  1. 本文所有代码在fedora 27环境下测试,python 2.7.15。

  2. click_patch目前只是参考实现,不建议在生产上使用,如果要在生产上用,请务必自己确认和测试代码。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,922评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,591评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,546评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,467评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,553评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,580评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,588评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,334评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,780评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,092评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,270评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,925评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,573评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,194评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,437评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,154评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,127评论 2 352

推荐阅读更多精彩内容