Python入门：函数封装之python调用sqlmap

该案例是想通过python来简化sqlmap的使用，故会使用到查找sqlmap路径，以及在当前路径下找到sqlmap.py文件，这里肯定会有人要问了，为什么不使用list加下标的方式提取sqlmap.py文件呢？首先我们需要考虑的是，以下标的方式提取文件是可变的，如：在sqlmap文件夹中多了一个文件，这时候，你想取到目标文件的下标可能会变化，所以不能使用下标来提取目标文件，最简单的办法就是使用一个判断来找到目标文件，然后与目标path进行拼接，最终组合成想到的绝对文件路径

#test.py
import os

path = '/Users/frankslg/tools/sqlmap/'
os.chdir(path)
find_sqlmap = os.listdir(path)
num = len(find_sqlmap)


for sql in find_sqlmap:
    if sql == 'sqlmap.py':
        sqlmap_path = path + sql

comm = input('please enter the command parameters:')
os.system(sqlmap_path + ' ' + comm)

上下代码分别是无封装的结构和使用了封装的结构，当然使用了封装有相当多的好处，如果调用简单、模块化、代码结构清晰等等。

#test1.py
def call_sqlmap(comm):
    path = '/Users/frankslg/tools/sqlmap/'
    os.chdir(path)
    find_sqlmap = os.listdir(path)

    for sql in find_sqlmap:
        if sql == 'sqlmap.py':
            sqlmap_path = path + sql
    return os.system(sqlmap_path + ' ' + comm)

comm = input('please enter the command parameters:')

if __name__ == '__main__':
    call_sqlmap(comm)

演示结果

please enter the command parameters:--help

    sqlmap/0.9 - automatic SQL injection and database takeover tool
    http://sqlmap.sourceforge.net

Usage: python /Users/frankslg/tools/sqlmap/sqlmap.py [options]

Options:
  --version             show program's version number and exit
  -h, --help            show this help message and exit
  -v VERBOSE            Verbosity level: 0-6 (default 1)

补充案例

#burp.py

path1 = "/Users/frankslg/tools/burp/"
file1 = os.listdir(path1)
comm = ''
for burp in file1:
    if burp == "burpsuite1.6.jar":
        comm = "burpsuite1.6.jar"

os.chdir(path1)
os.system('open' + ' ' + comm)

#burp.py
def burp_run(burp_path):
    file1 = os.listdir(burp_path)
    comm = ''
    for burp in file1:
        if burp == "burpsuite1.6.jar":
            comm = "burpsuite1.6.jar"
    os.chdir(burp_path)
    return os.system('open' + ' ' + comm)

burp_path = input('Plase enter burp_path:')
if __name__ == '__main__':
    burp_run(burp_path)

运行结果

在Plase enter burp_path:后输入/Users/frankslg/tools/burp/

Plase enter burp_path:/Users/frankslg/tools/burp/

burp工具就被打开了