有个脚本一直调用一个xml文件，想让他更加灵活。就有了今天的这个需求。
程序使用配置文件生成任务，但是配置文件手动编写太过繁琐，配合调用脚本完成自动化
# -*-coding：utf-8 -*-
# BY WANGCC
import xml.dom.minidom as minidom
class Nsf_currency:
task_type='domain'
os_match = 'yes'
messtypes='5;6;7;8;9;10;11;12;13;14;15;16;17;18;20;21;22;23;24;25;26;27;28'
isguesspwd='no'
port_strategy='user'
port_speed='3'
port_tcp='T'
port_udp='yes'
live='yes'
live_icmp='yes'
live_udp='no'
live_tcp='yes'
live_tcp_ports='21,22,23,25,80,443,445,139,3389,6000'
scan_level='3'
timeout_plugins='40'
timeout_read='5'
enable_unsafe_plugins='no'
scan_alert='yes'
alert_msg=''
scan_huawei='no'
check_addtional='no'
scan_oracle='yes'
ifdebug='no'
encoding='UTF-8'
class Nsf_conf(Nsf_currency):
def __init__(self,targets, port_strategy_userports):
self.targets=targets
self.port_strategy_userports=port_strategy_userports
dom = minidom.getDOMImplementation().createDocument(None, 'config', None)
root = dom.documentElement
employee = dom.createElement('server')
root.appendChild(employee)
targets = dom.createElement('targets')
targets.appendChild(dom.createTextNode('\n\t'))
root.appendChild(targets)
report = dom.createElement('report')
report.appendChild(dom.createTextNode('\n\t'))
root.appendChild(report)
pwdguess = dom.createElement('pwdguess')
pwdguess.appendChild(dom.createTextNode('\n\t'))
root.appendChild(pwdguess)
#----------------
taskname = dom.createElement('taskname')
taskname.appendChild(dom.createCDATASection(''))
root.appendChild(taskname)
#--------------------plugin_template_id
plugin_template_id = dom.createElement('plugin_template_id')
plugin_template_id.appendChild(dom.createTextNode('0'))
root.appendChild(plugin_template_id)
#---------------------
scanpri = dom.createElement('scanpri')
scanpri.appendChild(dom.createTextNode('2'))
root.appendChild(scanpri)
def dom_createE(name,value):
element = dom.createElement('key')
element.setAttribute('name', name)
element.setAttribute('value', value)
employee.appendChild(element)
return
def writeInfoToXml(object):
#内容参数
dom_createE(name='task_type',value=Nsf_currency.task_type)
dom_createE(name='targets',value=object.targets)
dom_createE(name='os_match',value=object.os_match)
dom_createE(name='messtypes',value=object.messtypes)
dom_createE(name='isguesspwd', value=object.isguesspwd)
dom_createE(name='port_strategy', value=object.port_strategy)
dom_createE(name='port_strategy_userports', value=object.port_strategy_userports)
dom_createE(name='port_speed', value=object.port_speed)
dom_createE(name='port_tcp', value=object.port_tcp)
dom_createE(name='port_udp', value=object.port_udp)
dom_createE(name='live', value=object.live)
dom_createE(name='live_icmp', value=object.live_icmp)
dom_createE(name='live_udp', value=object.live_udp)
dom_createE(name='live_tcp', value=object.live_tcp)
dom_createE(name='live_tcp_ports', value=object.live_tcp_ports)
dom_createE(name='scan_level', value=object.scan_level)
dom_createE(name='timeout_plugins', value=object.timeout_plugins)
dom_createE(name='timeout_read', value=object.timeout_read)
dom_createE(name='enable_unsafe_plugins', value=object.enable_unsafe_plugins)
dom_createE(name='scan_alert', value=object.scan_alert)
dom_createE(name='scan_huawei', value=object.scan_huawei)
dom_createE(name='check_addtional', value=object.check_addtional)
dom_createE(name='scan_oracle', value=object.scan_oracle)
dom_createE(name='ifdebug', value=object.ifdebug)
dom_createE(name='encoding', value=object.encoding)
with open('test.xml', 'w', encoding='utf-8') as f:
dom.writexml(f, addindent='\t', newl='\n', encoding='utf-8')
if __name__ == '__main__':
object=Nsf_conf('10.10.10.10,20.20.20.20','1-100,443,445')
writeInfoToXml(object)
print(object.targets)
如上图所示，有点乱，还有很多可以优化的地方，先整出来，再优化吧
以下是今天借鉴的文章地址。
python 通过xml获取测试节点和属性
python中minidom对xml的常用处理
官方文档
