book/v.py at main · kunki/book · GitHub

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
# -*- coding: utf-8 -*-
import http.cookiejar,urllib.request,re,os,sys,shutil

def getAtt(url):
	# 允許 cookie
	cookie = http.cookiejar.CookieJar()
	cookieProc = urllib.request.HTTPCookieProcessor(cookie)
	opener = urllib.request.build_opener(cookieProc)
	urllib.request.install_opener(opener)

	urllib.request.urlretrieve(url, 'temp.txt')

	# 取得各種變量
	data=open('temp.txt','rU',encoding='UTF-8').read()
	jpgPath=re.compile(r'(?<=jpgPath:")[0-9a-zA-Z\/]+(?=")')
	strf=jpgPath.search(str(data)).group()
	title=re.compile(u'(?<=<title>)[^<]+(?=</title>)')
	bookname=title.search(str(data)).group()
	ss=re.compile(r'(?<=ssid:.)\d{8}')
	ssid=ss.search(str(data)).group()

	# 構造鏈接和文件夾
	legurl="http://img.duxiu.com"+strf+"leg001?zoom=2"
	ssFolder="F:\\ss\\"+bookname+"_"+ssid+"\\"
	FolderExist=os.path.isdir(ssFolder)
	if FolderExist == False:
		os.mkdir(r'f:/ss/'+bookname+'_'+ssid+'/')
	legpath=ssFolder+"leg001.pdg"

	# 下載
	urllib.request.urlretrieve(legurl,legpath)
	os.remove("temp.txt")
	shutil.move(ssFolder,"F:\\ss\\leg\\"+bookname+"_"+ssid+"\\")

def main():
	getAtt(sys.argv[1])
if __name__ == "__main__":
	main()