前言
当我们使用python完成自己的原创的工具时，比如：端口扫描、弱口令爆破等。你是否想过添加自己的logo，以及简要的帮助信息？
如下：
sqlmap的logo
nmap的说明文档
本文接下来介绍linux下的figlet工具，以及python用于解析命令行参数和选项的标准模块argparse。
figlet使用
figlet是linux下的一款工具，用于将输入的字符串输出为各种奇形怪状的logo。可用于撰写自己的工具后，插入代表着自己原创工具的logo标识。
安装
|
1
|
sudo apt-get install figlet |
目录文件
flc为控制文件，flf为字体模板文件。
常用语法
figlet anonysec -f standard.flf
① anonysec 为我们要生成logo的字符串
② -f standard.flf 为选择不同风格的模板
效果展示
【注：其他使用参数，请自行查阅】
argparse模块使用
argparse是python用于解析命令行参数和选项的标准模块，向原创工具中加入帮助信息，编写可读性非常好的程序。
大家对linux下的工具一定有印象吧，使用- -help查看帮助文档。此模块正是打造这样的帮助文档。
导入模块
|
1
|
from argparse import argumentparser |
可以直接import argparse，只不过写代码的时候比较麻烦。
实例化对象
|
1
|
parser = argumentparser() |
实例化一个argumentparser类的对象
添加说明文档
|
1
|
parser.add_argument("-i", dest="ip", help="please enter the ip,for example 192.168.1.1") |
上述这行代码的add_argument()方法是向parser中添加。
① -i 输入时紧跟此参数的一个字符串为此参数的值
② dest=“ip” 将 -i 后面的字符串赋值给 ip
③ help=“******” 此参数的说明信息
|
1
|
parser.add_argument("-t", dest="thread_count", type=int, default=100, help="please enter the thread number") |
① type=“int” 此参数的默认类型为int型
② default=100 当不输入此参数时,此参数默认为100
赋值
|
1
|
args = parser.parse_args() |
将上述添加的参数进行赋值。
完整代码
|
1
2
3
4
5
6
7
|
parser = argumentparser()parser.add_argument("-i", dest="ip", help="please enter the ip,for example 192.168.1.1")parser.add_argument("-t", dest="thread_count", type=int, default=100, help="please enter the thread number")args = parser.parse_args()print 'the ip is: %s .' %args.ipprint 'the thread number is: %s .' %args.thread_count |
print是用来测试是否赋值成功。若赋值成功，就可以调用args.ip与args.thread_count来使用了。
【注：其他使用参数，请自行查阅】
附:base64-logo
|
1
2
3
|
logo_code = 'icagif8gicagicagicagicagicagicagicagicagicagx19fxyagicagicagicagiaogicavifwgicbfif9ficagx19ficbfif9ficbficagxy8gx19ffcagx19ficbfx18gciaglybfifwgfcanxybcic8gxybcfcanxybcfcb8ihwgxf9fxybcic8gxybclybfx3wkic8gx19fifx8ihwgfcb8ichfksb8ihwgfcb8ihxffcb8x19fksb8icbfxy8gkf9fiaovxy8gicbcx1xffcb8x3xcx19fl3xffcb8x3xcx18sihxfx19flybcx19fffxfx198ciagicagicagicagicagicagicagicagicagihxfx18vcgpqb3dlcmvkigj5iefub255u2vjckjsb2cnczogahr0cdovl3d3dy5hbm9uexnlyy5jbg=='logo = base64.b64decode(logo_code)print logo |
logo_code是事先使用base64加密过的，目的是让logo的嵌入不显得那么突兀。只需要base64解密一次，就可以将它输出至屏幕。
实例效果图
附源码
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
# coding:utf-8# powered by anonysecimport base64from argparse import argumentparserlogo_code = 'icagif8gicagicagicagicagicagicagicagicagicagx19fxyagicagicagicagiaogicavifwgicbfif9ficagx19ficbfif9ficbficagxy8gx19ffcagx19ficbfx18gciaglybfifwgfcanxybcic8gxybcfcanxybcfcb8ihwgxf9fxybcic8gxybclybfx3wkic8gx19fifx8ihwgfcb8ichfksb8ihwgfcb8ihxffcb8x19fksb8icbfxy8gkf9fiaovxy8gicbcx1xffcb8x3xcx19fl3xffcb8x3xcx18sihxfx19flybcx19fffxfx198ciagicagicagicagicagicagicagicagicagihxfx18vcgpqb3dlcmvkigj5iefub255u2vjckjsb2cnczogahr0cdovl3d3dy5hbm9uexnlyy5jbg=='logo = base64.b64decode(logo_code)print logoprint ''parser = argumentparser()parser.add_argument("-i", dest="ip", help="please enter the ip,for example 192.168.1.1")parser.add_argument("-t", dest="thread_count", type=int, default=100, help="please enter the thread number")args = parser.parse_args()#parser.print_help()print 'the ip is: %s .' %args.ipprint 'the thread number is: %s .' %args.thread_countprint '' |
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对服务器之家的支持。
原文链接：https://www.jianshu.com/p/a129001557ed
