依赖lxml的索引

This commit is contained in:
grt1st
2016-11-09 21:51:27 +08:00
committed by GitHub
parent 4d0a61c565
commit 547faeadc7
2 changed files with 99 additions and 0 deletions

52
app2.py Normal file
View File

@@ -0,0 +1,52 @@
#coding=utf-8
import os
import sys
import MySQLdb
from lxml import etree
path='bugs'
#模式
pattern0=u'漏洞类型:'
pattern1=u'提交时间:'
pattern2=u'漏洞标题:'
for docs in os.listdir(path):
#打开文件,提取内容
if os.path.isdir('bugs/'+docs):
print "目录跳过"
continue
doc=open('bugs/'+docs,'r')
html=doc.read()
doc.close()
#提取信息
xml=etree.HTML(html)
corp=xml.xpath("//h3[@class='wybug_corp']//a")[0].text.replace(' ','').replace('\n','')
author=xml.xpath("//h3[@class='wybug_author']//a")[0].get('href').replace('http://www.wooyun.org/whitehats/','')
types=xml.xpath("//h3[@class='wybug_type']")[0].text.replace(' ','')
typr1=types.replace(pattern0,'')
date=xml.xpath("//h3[@class='wybug_date']")[0].text.replace(' ','')
date1=date.replace(pattern1,'')
title=xml.xpath("//h3[@class='wybug_title']")[0].text.replace(' ','')
title1=title.replace(pattern2,'')
print corp,author,type1,date1,title1
#连接数据库
try:
conn=MySQLdb.connect(host='localhost',port=3306,user='root',passwd='',db='wooyun',charset='utf8')
cur=conn.cursor()
reload(sys)
sys.setdefaultencoding('utf-8')
tmp=(title1,date1,author1,type1,corp1,docs)
cur.execute("INSERT INTO `bugs`(`title`,`dates`,`author`,`type`,`corp`,`doc`) VALUES(%s,%s,%s,%s,%s,%s)",tmp)
conn.commit()
cur.close()
conn.close()
except MySQLdb.Error,e:
print "Mysql Error %d: %s" % (e.args[0], e.args[1])

47
app3.py Normal file
View File

@@ -0,0 +1,47 @@
#coding:utf-8
import os
import re
import sys
import MySQLdb
from lxml import etree
path='drops'
pattern0=u' | WooYun知识库'
pattern1=re.compile(r'(.*)(?=-)')
for docs in os.listdir(path):
if os.path.isdir('drops/'+docs):
print "目录跳过"
continue
#打开文件,提取内容
doc=open('drops/'+docs,'r')
html=doc.read()
doc.close()
#提取信息
xml=etree.HTML(html)
title=xml.xpath("//title")[0].text.replace(pattern0,'')
author=xml.xpath("//a[@class='author name ng-binding']")[0].text.replace(' ','').replace(' ','').replace('\n','')
time=xml.xpath("//time[@class='published ng-binding ng-isolate-scope']")[0].text
doc=re.findall(pattern1,docs)
#doc[0]
print title,author,time,doc[0],docs
try:
conn=MySQLdb.connect(host='localhost',port=3306,user='root',passwd='',db='wooyun',charset='utf8')
cur=conn.cursor()
reload(sys)
sys.setdefaultencoding('utf-8')
tmp=(title,time,author,doc[0],docs)
cur.execute("INSERT INTO `drops`(`title`,`dates`,`author`,`type`,`doc`) VALUES(%s,%s,%s,%s,%s)",tmp)
conn.commit()
cur.close()
conn.close()
except MySQLdb.Error,e:
print "Mysql Error %d: %s" % (e.args[0], e.args[1])