前言

當我們使用Python完成自己的原創(chuàng)的工具時，比如：端口掃描、弱口令爆破等。你是否想過添加自己的Logo，以及簡要的幫助信息？

如下：

Sqlmap的Logo

Nmap的說明文檔

本文接下來介紹Linux下的figlet工具，以及Python用于解析命令行參數(shù)和選項的標準模塊argparse。

Figlet使用

figlet是Linux下的一款工具，用于將輸入的字符串輸出為各種奇形怪狀的Logo。可用于撰寫自己的工具后，插入代表著自己原創(chuàng)工具的Logo標識。

安裝

sudo apt-get install figlet

目錄文件

flc為控制文件，flf為字體模板文件。

常用語法

figlet AnonySec -f standard.flf

① AnonySec 為我們要生成Logo的字符串

② -f standard.flf 為選擇不同風格的模板

效果展示

【注：其他使用參數(shù)，請自行查閱】

Argparse模塊使用

argparse是python用于解析命令行參數(shù)和選項的標準模塊，向原創(chuàng)工具中加入幫助信息，編寫可讀性非常好的程序。
大家對linux下的工具一定有印象吧，使用- -help查看幫助文檔。此模塊正是打造這樣的幫助文檔。

導入模塊

from argparse import ArgumentParser

可以直接import argparse，只不過寫代碼的時候比較麻煩。

實例化對象

parser = ArgumentParser()

實例化一個ArgumentParser類的對象

添加說明文檔

parser.add_argument("-i", dest="ip", help="Please enter the ip,for example 192.168.1.1")

上述這行代碼的add_argument()方法是向parser中添加。

① -i   輸入時緊跟此參數(shù)的一個字符串為此參數(shù)的值

② dest=“ip”   將 -i 后面的字符串賦值給 ip

③ help=“******”   此參數(shù)的說明信息

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

① type=“int”   此參數(shù)的默認類型為int型

② default=100   當不輸入此參數(shù)時,此參數(shù)默認為100

賦值

args = parser.parse_args()

將上述添加的參數(shù)進行賦值。