# -*-coding=utf-8-*-
import os
import re
import time
from urllib.parse import urljoin
from selenium import webdriver
import requests
from bs4 import BeautifulSoup
from url_decode import urldecode
class GovSpider:
def __init__(self):
self.site_url = 'http://www.jiangxi.gov.cn/kzw/jxszfxxgkml/zfjg/zfld_12550/'
self.spath = re.sub('.*//', '', self.site_url)
sindex = self.spath.find('/')
if sindex != -1:
self.spath = self.spath[:sindex]
self.base_path = '~/Desktop/gov_spider/'+self.spath
print('base:',self.base_path)
def start_page(self):
req = webdriver.Chrome()
req.get(self.site_url)
time.sleep(3)
html1 = req.page_source
soup = BeautifulSoup(html1,'lxml')
iframes = soup.findAll('iframe')
self.gs_runner(urldecode(html1))
if iframes:
print('有隐藏页面:',iframes)
for ifr_url in iframes:
if 'https:' not in ifr_url or 'http' not in ifr_url:
ifr_url = urljoin(self.site_url,ifr_url.get('src'))
print('进入隐藏页面:',ifr_url,',并开启隐藏页面抓取img、css、js的程序........')
try:
html2 = requests.get(ifr_url).text
self.gs_runner(urldecode(html2))
except Exception as e:
print('隐藏页面的地址格式不正确,无法访问:',ifr_url)
#程序运行完毕,自动关闭浏览器
req.quit()
def download_imgs(self,html):
# 多筛选 防止无格式或不在src下的图片漏下
liResult = re.findall('(".*?")',html)
if liResult:
print('>>>>>>>>>>>>>>>>>>>>>>>>>正在下载img文件>>>>>>>>>>>>>>>>>>>>>>>>>')
for img_li in liResult:
if img_li and ('{' not in img_li and '</' not in img_li) and('.jpg' in img_li or '.ico' in img_li or '.png' in
img_li or '.svg' in img_li or '.gif' in img_li or '.webp' in img_li or 'data:image' in img_li):
if '\/' not in img_li:
img_urls = [re.sub('[";]|(.*=)|(.*\()|(\).*)|(")','',img_li)]
else:
img_urls = [re.sub('[";]','',img_li).replace('\/\/','//').replace('\/','//')]
if ',' in img_urls[0]:
print('检测到有js代码中有多路径img_url,正在深入抓取......')
img_urls = img_urls[0].split(',')
for img_url in img_urls:
try:
if 'http:' not in img_url or 'https:' not in img_url:
img_url = urljoin(self.site_url,img_url)
img_name = re.sub('(.*//)', '', img_url)
f_index = img_name.find('/')
l_index = img_name.rfind('/')
dir_path = self.spath+img_name[f_index:l_index]
if not os.path.exists(dir_path):
os.system('mkdir -p %s'%dir_path)
img_name = re.sub('(.*/)', '', img_url)
print('正在下载img文件:',img_url)
img = requests.get(img_url).content
with open('./%s' % (dir_path + '/' + img_name), 'wb') as f:
f.write(img)
except Exception as e:
print(img_url,e,'\r\nthis is uvaild url')
print('===========================img文件下载完毕=============================')
def download_js(self,html):
liResult = re.findall('(".*?")',html)
if liResult:
print('>>>>>>>>>>>>>>>>>>>>>>>>>正在下载js文件>>>>>>>>>>>>>>>>>>>>>>>>>')
for on_url in liResult:
if (on_url.endswith('.js"') or 'js?' in on_url) and 'src=' not in on_url:
js_links = [re.sub('(")|(.*:)', '', on_url)]
if ',' in js_links[0]:
print('检测到js代码中有多路径js_link,正在深入抓取.......')
js_links += js_links[0].split(',')
for js_link in js_links:
if 'http:' not in js_link or 'https:' not in js_link:
js_link = urljoin(self.site_url, js_link)
js_name = re.sub('(.*//)', '', js_link)
f_index = js_name.find('/')
l_index = js_name.rfind('/')
dir_path = self.spath+js_name[f_index:l_index]
if '/$' in dir_path:
dir_path = re.sub('(/\$)','/',dir_path)
if not os.path.exists(dir_path):
os.system('mkdir -p %s'%dir_path)
print('正在下载js文件:',js_link)
js_file = requests.get(js_link).content
js_name = re.sub('(.*/)', '', js_link)
if js_name[0] == '$':
js_name = js_name.replace('$','')
with open('./%s'%dir_path+'/'+js_name,'wb') as f:
f.write(js_file)
print('===========================js文件下载完毕===========================')
def download_css(self,html):
liResult = re.findall('(".*?")',html)
if liResult:
print('>>>>>>>>>>>>>>>>>>>>>>>>>正在下载css文件>>>>>>>>>>>>>>>>>>>>>>>>>')
for on_url in liResult:
if on_url.endswith('.css"') or 'css?' in on_url:
css_link = re.sub('["]','',on_url)
if 'http:' not in css_link or 'https:' not in css_link:
css_link = urljoin(self.site_url,css_link)
if css_link:
css_name = re.sub('(.*//)', '', css_link)
f_index = css_name.find('/')
l_index = css_name.rfind('/')
dir_path = self.spath+css_name[f_index:l_index]
if '/$' in dir_path:
dir_path = re.sub('(/\$)','/',dir_path)
if not os.path.exists(dir_path):
os.system('mkdir -p %s'%dir_path)
css_name = re.sub('(.*/)', '', css_link)
if css_name[0] == '$':
css_name = css_name.replace('$','')
print('正在下载css文件:',css_link)
css_file = requests.get(css_link).content
with open('./%s'%(dir_path+'/'+css_name) ,'wb') as f:
f.write(css_file)
print('===========================css文件下载完毕===========================')
def download_html(self,html):
print('>>>>>>>>>>>>>>>>>>>>>>>>>正在下载网页源码>>>>>>>>>>>>>>>>>>>>>>>>>')
if not os.path.exists(self.spath+'/htmls'):
print(self.base_path+'/htmls', '已经将html文件存在对应目录')
os.system('mkdir -p %s' % self.spath+'/htmls')
else:
print(self.base_path+'/htmls', '已经存在此htmls目录')
if html:
with open('./%s' % (self.spath+'/htmls/'+self.spath+'.html'), 'w', encoding='utf-8') as f:
f.write(html)
print('===========================下载网页源码完毕===========================')
def gs_runner(self,html):
self.download_html(html)
self.download_imgs(html)
self.download_css(html)
self.download_js(html)
if __name__ == '__main__':
gs = GovSpider()
gs.start_page()