forked from paperen/website-clone
-
Notifications
You must be signed in to change notification settings - Fork 0
/
clone.py
128 lines (109 loc) · 3.52 KB
/
clone.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#!/usr/bin/env
# -*- coding: UTF-8 -*-
#Filename: clone.py
import urllib,re,time,sys,core,webbrowser,getopt,os
def usage():
print u'''
-h or --help 说明
-u or --url 站点网址(site url)
-d or --dir [可选]保存本地目录，默认为站点域名(which directory to save files[option])
例子(example):
py clone.py -u paperen.com
clone -u paperen.com -d test (windows下)
'''
return 0
def show_progress():
sys.stdout.write("=")
time.sleep(0.1)
def main():
url = ""
root = None
try :
opts,args = getopt.getopt(sys.argv[1:], 'u:d:h:',
[
'url=',
'dir=',
'help=',
]
)
for opt, val in opts:
if opt in ( '-u', '--url' ):
url = val
elif opt in ( '-d', '--dir' ):
root = val
elif opt in ( '-h', '--help' ):
usage()
else:
raise Exception
except getopt.GetoptError, err:
usage()
try :
if root != None and os.path.exists(root) == False:
print u"保存路径不存在，请先创建该目录(directory not exists)";
raise Exception
if re.search(r"^((https|http)?:\/\/)[^\s]+",url) == None:
print u"输入的网址有误(url invalid)";
raise Exception
t = core.thief(url,root)
# 爬取网页
opener = urllib.FancyURLopener({})
f = opener.open(url)
html = f.read()
if len(html) == 0 :
print u"获取网页失败(url fetch error)"
raise Exception
# css
# <link href="http://paperen.com/theme/paperen/bootstrap.css" rel="stylesheet" />
# <link rel="stylesheet" type="text/css" href="http://events.hytera.com/theme/hytera/common.css" media="all" />
# <link href="http://fonts.googleapis.com/css?family=Montserrat" rel="stylesheet" />
sys.stdout.write( u"开始爬取css" )
css = re.findall(r"<\s*link\s+[^>]*?href\s*=\s*[\"|\'](.*?)[\"|\'][\s\S]*?>", html)
for v in css:
html = html.replace(v,t.get(v))
show_progress()
sys.stdout.flush()
print u"爬取css完成"
# 页面内css
# <style>body{background:url("images/bg.jpg")}</style>
# <body style="background:url("images/bg.jpg")">
sys.stdout.write( u"开始爬取页面内css" )
intercss = re.findall(r"url\((.*?)\)", html)
for v in intercss:
if v.find('data:') == -1 : html = html.replace(v,t.get(v))
show_progress()
sys.stdout.flush()
print u"爬取页面内css完成"
# js
# <script src="http://paperen.com/js/jquery-1.7.1.min.js"></script>
# <script src="/js/jquery-1.7.1.min.js"></script>
# <script src="http://fonts.googleapis.com/js/jquery-1.7.1.min.js"></script>
sys.stdout.write( u"开始爬取js" )
js = re.findall(r"<\s*script\s+[^>]*?src\s*=\s*[\"|\'](.*?)[\"|\'][\s\S]*?>", html)
for v in js:
html = html.replace(v,t.get(v))
show_progress()
sys.stdout.flush()
print u"爬取js完成"
# image
# <img src="http://paperen.com/file/182" alt="http://paperen.com/file/182" />
# <img src="http://events.hytera.com/upload/2016/03/22/62150172eebd2d423c42cdc511446bd7.jpg" class="dpb main-banner-img">
# <img src="/upload/2016/03/22/62150172eebd2d423c42cdc511446bd7.jpg" class="dpb main-banner-img">
sys.stdout.write( u"开始爬取图片" )
img = re.findall(r"<\s*img\s+[^>]*?src\s*=\s*[\"|\'](.*?)[\"|\'][\s\S]*?>", html)
for v in img:
html = html.replace(v,t.get(v))
show_progress()
sys.stdout.flush()
print u"爬取图片完成"
# 保存网页
index = t.getroot()+'index.html'
fp = open(index, 'wb')
fp.write(html)
fp.close()
print u"生成页面完成，请打开 " + index + u" 查看效果"
# 打开网页
webbrowser.open(index, 2, True)
except Exception:
print u"爬取失败"
if __name__ == "__main__":
main()