fuzzall_report.py 文件源码

def _to_db(html,db,cursor,domain_name):
    html = lxml.html.fromstring(html)

    rows = html.xpath('//table')
    '''
    list_dns_record = []
    list_mx_record = []
    elements = rows[0].xpath('.//tr')
    for k in elements:
        c =  k.xpath('.//td//text()')
        #print c
        list_dns_record.append(c[0])
    #print list_dns


    elements = rows[1].xpath('.//tr')
    if  len(elements) < 1:
        return 3
    for k in elements:
        c =  k.xpath('.//td//text()')
        list_mx_record.append(c[0])




    print list_dns_record
    print list_mx_record'''
    dic_domain = {}


    elements = rows[3].xpath('.//tr')
    if  len(elements)  <2:
        return False
    for k in elements:
        c =  k.xpath('.//td//text()')

        if len(c)>0:
            dic_domain['domain'] = domain
            dic_domain['fuzzall_id'] = int(c[0])
            dic_domain['ip_range'] = c[1]
            #dic_domain['data_tag'] = c[2]
            dic_domain['fuzz_time'] = c[3]
            sql = "INSERT INTO app_domainiprange(domain_name, \
                        ip_range, fuzz_time, fuzzall_id) \
                       VALUES ('%s',   '%s', '%s', '%d' )" % \
                       (domain_name,  dic_domain['ip_range'], dic_domain['fuzz_time'], dic_domain['fuzzall_id'])

            cursor.execute(sql)

    db.commit()
    return True