使用Python打造原创工具的Logo＆Help Info

当我们使用Python完成自己的原创的工具时，比如：端口扫描、弱口令爆破等。你是否想过添加自己的Logo，以及简要的帮助信息？

如下：

sqlmap的Logo

Nmap的说明文档

本文接下来介绍Linux下的figlet工具，以及Python用于解析命令行参数和选项的标准模块argparse。

figlet是Linux下的一款工具，用于将输入的字符串输出为各种奇形怪状的Logo。可用于撰写自己的工具后，插入代表着自己原创工具的Logo标识。

1. 安装

sudo apt-get install figlet

2. 目录文件

flc为控制文件，flf为字体模板文件。

3. 常用语法

figlet AnonySec -f standard.flf

AnonySec 为我们要生成Logo的字符串

-f standard.flf 为选择不同风格的模板

4. 效果展示

【注：其他使用参数，请自行查阅】

argparse是python用于解析命令行参数和选项的标准模块，向原创工具中加入帮助信息，编写可读性非常好的程序。

大家对linux下的工具一定有印象吧，使用- -help查看帮助文档。此模块正是打造这样的帮助文档。

1. 导入模块

from argparse import ArgumentParser

可以直接import argparse，只不过写代码的时候比较麻烦。

2. 实例化对象

parser = ArgumentParser()

实例化一个ArgumentParser类的对象

3. 添加说明文档

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=“******” 此参数的说明信息

parser.add_argument("-t", dest="thread_count", type=int, default=100, help="Please enter the thread number")

type=“int” 此参数的默认类型为int型

default=100 当不输入此参数时,此参数默认为100

4. 赋值

args = parser.parse_args()

将上述添加的参数进行赋值。

5. 完整代码

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 ips is: %s.' %args.ip

print 'The thread number is: %s.' %args.thread_count

print是用来测试是否赋值成功。若赋值成功，就可以调用args.ip与args.thread_count来使用了。

【注：其他使用参数，请自行查阅】

附:Base64-Logo

logo_code = 'ICAgIF8gICAgICAgICAgICAgICAgICAgICAgICAgICAgX19fXyAgICAgICAgICAgIAogICAvIFwgICBfIF9fICAgX19fICBfIF9fICBfICAgXy8gX19ffCAgX19fICBfX18gCiAgLyBfIFwgfCAnXyBcIC8gXyBcfCAnXyBcfCB8IHwgXF9fXyBcIC8gXyBcLyBfX3wKIC8gX19fIFx8IHwgfCB8IChfKSB8IHwgfCB8IHxffCB8X19fKSB8ICBfXy8gKF9fIAovXy8gICBcX1xffCB8X3xcX19fL3xffCB8X3xcX18sIHxfX19fLyBcX19ffFxfX198CiAgICAgICAgICAgICAgICAgICAgICAgICAgIHxfX18vCgpQb3dlcmVkIGJ5IEFub255U2VjCkJsb2cnczogaHR0cDovL3d3dy5hbm9ueXNlYy5jbg=='

logo = base64.b64decode(logo_code)

print logo

logo_code是事先使用base64加密过的，目的是让Logo的嵌入不显得那么突兀。只需要base64解密一次，就可以将它输出至屏幕。

实例效果图