Python + pyspider某小说站的爬虫，入数据库，火车头发布，资...

fei

Python + pyspider某小说站的爬虫，入数据库，火车头发布，资源下载到本地，另可写爬虫！
, l5 c% r6 W$ _' Q4 T8 I

1. #!/usr/bin/env python
   
2. # -*- encoding: utf-8 -*-
   
3. # Created on 2019-05-05 21:43:11
   ' y8 o! V) k9 k! a9 n
4. # Project: XiaoShuo
   " l) ~9 l6 S. {0 ~) I: n$ {
5. 
   1 d$ K8 O1 I& c/ `$ z, X9 r
6. from pyspider.libs.base_handler import *
   6 W" s, Z/ e" ^6 S& `
7. import pymysql
   ( N# N  X! p9 B% m) {% n
8. import random
   ! e8 \' E2 r: m0 L
9. import datetime
   
10. import urllib2,HTMLParser,re
    
11. import os
    ) C6 R8 \* P8 \$ F' M9 }
12. import sys
    0 t- d2 x" g2 a5 b
13. import re
    % R( Q4 V( v1 i/ E  ^- Q4 Y6 ]
14. import codecs
    " s: y6 p/ d$ V/ n4 d8 w" N$ E
15. import requests
    3 k: O6 F8 y6 d' x' G
16. import json
    
17. 
    ) ]. z2 J; s3 W) t" w
18. class Handler(BaseHandler):
    
19.     global Datos
    
20.     global P_dir   
    
21.     P_dir = '/Tools/Debug/'  #采集时候图片保持到本地的路径
    
22.     global Datos
    
23.     Datos = {}
    ) X' V6 b$ A9 X% Z' @
24.     headers= {
    ' q; J6 W  ]6 _6 P% t6 n+ z# l( n
25.     'Accept':'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
    . Z, }* u6 d7 U6 ~. V
26.     'Accept-Encoding':'gzip, deflate, sdch',
    
27.     'Accept-Language':'zh-CN,zh;q=0.8',
    + P$ G( O; \, n1 a# N; _
28.     'Cache-Control':'max-age=0',
    2 H& p! W0 `+ Y" z# _! ^; o, z
29.     'Connection':'keep-alive',
    ' b& A$ q3 a/ Q6 C9 e& ~5 Z( K
30.     'User-Agent':'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/41.0.2272.101 Safari/537.36'
    ( N8 Z# `" [- {! }3 k
31.     }
    
32.     crawl_config = {
    0 Q( K7 D; V; a9 M) Q3 O
33.         'headers' : headers,
    
34.         'timeout' : 300
    
35.     }
    1 ~1 i9 Z/ [2 s. Z7 ?% a
36.     def add_BookFile(self,Bookname, BookIDs, img, Locaimg, Book_Dates):
    
37.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
      }+ t( U3 Z7 V6 g% i; F" i
38.         try:
    
39.             cursor = db.cursor()
    ; H! q+ x& K3 Z7 P
40.             #注意此处字符串的占位符要加双引号"%s"
    + C& P4 @& ?( l3 Z! Y: f# D0 y
41.             sql = 'insert into BookFile(Bookname, BookID, img, Locaimg, Book_Date) values ("%s","%s","%s","%s","%s")' % (Bookname, BookIDs, img, Locaimg, Book_Dates);
    
42. #            print(sql)
    
43.             cursor.execute(sql)
    
44.             
    3 p$ z: L1 a  [
45.             #qid = cursor.lastrowid
    
46.             #print(qid)
    ( p, i9 S" |2 j0 g' k- T% U; N8 `
47.             
    ( y, Z* D, S1 ^
48.             db.commit()
    * u, o. J0 J$ b1 n
49.         except Exception as err:
    
50.             print("Error %s for execute sql: %s" % (err, sql))
    
51.             db.rollback()
    / V0 {# }! r0 G" Z5 x4 j1 k
52.     def add_comment(self,Bookname, Booktitle, BookID, Titleid, Book_Date):
    
53.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
    6 w% e2 ^2 U, B  y" o
54.         try:
    & y# a; W6 Y. I; f" P6 J8 O  i
55.             cursor = db.cursor()
    ' @0 b- _6 q5 r% _
56.             #注意此处字符串的占位符要加双引号"%s"
    
57.             sql = 'insert into BookTitle(Bookname, Booktitle, BookID, Titleid, Book_Date) values ("%s","%s","%s","%s","%s")' % (Bookname, Booktitle, BookID, Titleid, Book_Date);
    
58. #            print(sql)
    0 l" a) w8 g4 a. c8 D" w. [
59.             cursor.execute(sql)
    / a) ]0 g* }# d8 f) m& t7 l$ B( I' c
60.             
    
61.             #qid = cursor.lastrowid
    
62.             #print(qid)
    : B6 O) V8 O4 Y8 z4 h9 {
63.             
    & u6 v: b* G4 w% Q
64.             db.commit()
    ( V3 Y' G$ g! c, [4 U/ e5 ^
65.         except Exception as err:
    
66.             print("Error %s for execute sql: %s" % (err, sql))
    
67.             db.rollback()
    
68.     def add_question(self,Bookname,Cater_Name,Book_author,Book_Introduction,Book_Synopsis,Book_Palabras,Bookurl,Booktitle,BookID,BookConte,Titleid,abover,Book_Date):
    9 L8 I2 Z- E- G; Q
69.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
    ( ], X2 ^/ Y  w- o* n
70.         try:
    ! @8 ?0 |; F: @% |( l
71.             cursor = db.cursor()
    ) Y/ e# G) N, R7 M& L; G1 y  s  e" \( B( r
72.             #注意此处字符串的占位符要加双引号"%s"
    1 p0 j- \3 r% u( s/ o' w
73.             sql = ' INSERT INTO BookConte (Bookname,Cater_Name,Book_author,Book_Introduction,Book_Synopsis,Book_Palabras,Bookurl,Booktitle,BookID,BookConte,Titleid,abover,Book_Date) VALUES("%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s")' % (Bookname,Cater_Name,Book_author,Book_Introduction,Book_Synopsis,Book_Palabras,Bookurl,Booktitle,BookID,BookConte,Titleid,abover,Book_Date);
    2 k. _7 f4 b) H3 A- o" F: o4 Y3 p! X
74.             print(sql)
    0 S& M+ h% b) V8 G( z8 W9 Y* t; ^* Y7 j
75.             cursor.execute(sql)
    
76.             print(cursor.lastrowid)
    
77.             db.commit()
    6 u) z# i* R/ C2 |" k1 w
78.         except Exception as err:
    
79. #        except:
    5 r8 ^0 g' F* R+ C5 Z
80. #            print('Failed')
    
81.             print("Error %s for execute sql: %s" % (err, sql))
    
82.             db.rollback()
    
83.         
    ; i3 U. Q  g: F: S( q' G
84.     def add_locoy(self,Bookname,Cater_Name,Book_author,Book_Introduction,Book_Palabras,Book_img,Booktitle,BookConte,abover):
    ) G! ?/ Q/ A: M7 N. Y% R: i
85.             reload(sys)
    
86.             sys.setdefaultencoding("gbk")
    
87.             locoy_url = 'http://www.******.net/locoy/?my=book'  #697火车头发接口地址
    + w6 L8 X4 l, U0 V5 P/ D
88.             locoy_data = {
    ! [3 C2 j% O& K; ^* o6 e
89.             'my_u':'用户名',   #后台用户名
    
90.             'my_p':'密码',   #后台密码
    3 O# L+ ~% @* [- y7 N. f- e
91.             'subject_669977_net':Bookname.encode('gbk', 'ignore'),
    # k* R9 T. S: T: U( {
92.             'caid':Cater_Name.encode('gbk', 'ignore'),
    
93.             'title_669977_net':Booktitle.encode('gbk', 'ignore'),
      M1 L# h$ n0 q! u( L6 `/ ]
94.             'article':BookConte.encode('gbk', 'ignore'),
    ' h0 J8 h& x! N+ `" w
95.             'author':Book_author.encode('gbk', 'ignore'),
    0 [5 d' y9 p* N  E" q1 \
96.             'ready_1':Book_Palabras.encode('gbk', 'ignore'),
    
97.             'thumb':Book_img,
    2 z, J, Q1 L7 w" J" U
98.             'content':Book_Introduction.encode('gbk', 'ignore'),
    
99.             'abover':abover.encode('gbk', 'ignore')           
    + n5 m3 R4 c+ S5 k8 N- \- C
100.                 }
     # q" U7 Y9 @3 E) o3 X8 F( E
101.             res = requests.post(locoy_url, data=locoy_data)
     
102.             print res.text
     
103.             print res.content
     0 q% P+ m- {: z, D( @1 E
104. #            print Dsd
     
105.             return res
       R& G2 M6 M1 U( I% T
106.    
     # ~+ k7 U" i2 r# u) I$ Z2 X
107.     def __init__(self):
     
108.         self.base_url1 = 'https://www.****.cc/'
     3 U" [+ d7 Z6 N) J6 _* ]
109.         self.base_url2 = '/'
     
110.         self.CaterId = []
     6 c% M* ?/ {, p# Q! Y: W) f$ U
111.         self.CaterIds = ['xuanhuan', 'wuxia', 'yanqing', 'lishi', 'kehuan', 'kongbu', 'nvsheng']
     + O3 }; F. U: u  Z2 W$ n) V+ c0 }
112.         self.page_num = 1
     9 o$ Y( C9 P3 ?' y, n
113.         self.total_num = 200   
     
114. 
     7 O6 x+ V4 k  ?6 W
115.     @every(minutes=8 * 60)
     
116.     def on_start(self):
     ; i0 B: X. f" Q* I& O9 ^
117.         global Cater_Name
     1 k6 Z4 _$ Y3 {4 g$ u
118.         Cater_Name = []
     3 Q# Q  _5 x) o+ c7 _% I( {8 G
119.         while self.page_num <= self.total_num:
     ' g/ J/ [* Y& |2 `- ?6 Z( f
120.             for self.CaterId in self.CaterIds:
     1 v, S' p7 x1 l" k, e1 `
121.                 if self.CaterId  == 'xuanhuan':
     ' ^/ p- }3 n8 U" E3 f. d
122.                      Cater_Name = '玄幻'
     # \% `. t0 I# a% C1 Q1 r6 H6 C. U
123.                 if self.CaterId  == 'wuxia':
     ( T% c' O" K/ l# B1 N! I! W
124.                     Cater_Name = '武侠'
     
125.                 if self.CaterId  == 'lishi':
     
126.                     Cater_Name = '历史'            
     
127.                 if self.CaterId  == 'yanqing':
     " \% X5 J7 }/ N, w1 s4 c
128.                     Cater_Name = '都市'
     
129.                 if self.CaterId  == 'nvsheng':
     
130.                     Cater_Name = '都市'
     
131.                 if self.CaterId  == 'kehuan':
     ' @- N% V' R6 B
132.                     Cater_Name = '科幻'
     
133.                 if self.CaterId  == 'kongbu':
     
134.                     Cater_Name = '游戏'
     ' d. t9 I4 p8 G/ ?+ G
135.                 print self.CaterId
     
136.                 url = self.base_url1 + str(self.CaterId) + self.base_url2 + str(self.page_num) + "/"         
     & L7 `0 M& l, r1 c  f- m
137.                 self.crawl(url, callback=self.list_Caterg,save=Cater_Name)
     
138.             self.page_num += 1
     
139.             
     
140.     def list_Caterg(self, response):
     # g# a( P- [5 ?  D! b; c, ~/ T4 P3 x
141.         Cater_Name = response.save
     . p, w: z- R8 X! ]  b2 s( W5 m+ a
142.         for each in response.doc('.pic-list a[href^="http"]').items():
     
143.             self.crawl(each.attr.href, callback=self.list_Caterg_detail,save=Cater_Name)
     
144.             
     
145.     def list_Caterg_detail(self, response):
     
146.         Cater_Name = response.save
     
147. #        print Cater_Name
     % N/ a4 X3 S* A" I
148.         Bookname = response.doc('h1').text()
     + @1 ?& n& }7 M
149.         print Bookname
     
150.         Book_author = response.doc('.authorname > a').text()
     . n, ~5 _" Z% P" t/ V
151. #        print Book_author
     ) p4 c8 }1 o( X6 x
152.         Book_Introduction = response.doc('.book-intro > div').text()
     " \4 p7 H- C8 ?) _# D; l4 `3 @
153. #        print Book_Introduction
     
154.         Book_Synopsis = response.doc('b').eq(1).text()
     ' Q0 o! o7 R/ I6 c2 A5 A& e
155. #        print Book_Synopsis
     1 ]. D8 I( O4 y6 e4 w( ^) O
156.         Book_Palabras = response.doc('.booktitle p').text().split(' ')[1].split('|')[0]
     
157. #        print Book_Palabras
     5 Y+ X1 G9 d( n& s3 V
158.         BookIDs = response.url.split("xiaoshuo/")[-1].split("/")[0]   #小说ID
     7 M3 ^: S" X  B; R9 Z4 }7 j) _$ s; x
159. #        print BookIDs
     $ v" @# m" ~& y) @5 Q
160.         Book_Dates = str(datetime.datetime.now())         
     
161.         for imgs in response.doc('.bigpic > img[src^="http"]').items():
     + \! @' b# r" o2 z. I
162.             img = imgs.attr.src
     8 ^/ E  K9 W$ Q
163.             print img
     & Y8 ^, {( y. ?  K- r9 A
164.                 #小说封面下载
     
165.             extension = self.getExtension(img)
     
166.             name = self.getname(img)
     
167.             file_name = name + "." + extension
     . P7 S! \! q# I- e0 N  s
168.             imgDir = P_dir + name
     
169.             Locaimg = imgDir + "/" + file_name
     
170.             print Locaimg
     ) W- z0 \, D- z9 v- R# @
171.             if(self.download(P_dir, imgDir, file_name, img)):   #这2行可注译，图片下载到本地
     % i. X' ^% d  [* K: o2 @0 U
172.                 print('attachment url is ' + img)               #
     - O0 r% O" w! H' V# F
173.             Datos = {
     
174.                     "Cater_Name":Cater_Name,
     % \/ P' y5 Z7 k
175.                     "Book_author":Book_author,
     : ?3 h  o5 L0 w9 ~
176.                     "Book_Introduction":Book_Introduction,
     , Y! u7 b& ?- c% C: C
177.                     "Book_Synopsis":Book_Synopsis,
     ; S% Q  ]* `. K1 j7 n* X* c, P) A* C
178.                     "Book_Palabras":Book_Palabras,
     ) L, M; S3 o- U( o% p) l; ?
179.                     "img":img,
     8 c; j3 Z, y7 j8 k  l  J& r4 m
180.                 }
     
181.             self.add_BookFile(Bookname, BookIDs, img, Locaimg, Book_Dates)  #这行可注译，数据库发布接口，方便其他系统的发布
     
182.         for each in response.doc('div[class="bookbtn-txt"]  a[class="catalogbtn"]').items():
     0 f! D' I* M6 D0 h
183.             self.crawl(each.attr.href, callback=self.index_page,save=Datos)
     : j3 i* N1 R6 L2 b/ g
184.             
     0 f- e9 b5 R& P. S
185.     @config(age=8 * 60 * 60)   
     0 ^5 F- D. c& B. n9 F) l) R5 d
186.     def index_page(self, response):
     
187.         Datos = {
     & Q3 V) O  i" O+ t; P- g' `
188.                   "Cater_Name":response.save['Cater_Name'],
     
189.                    "Book_author":response.save['Book_author'],
     ! d, j- g8 C) o5 @* I& Q
190.                    "Book_Introduction":response.save['Book_Introduction'],
     ; |% B6 v& K1 z0 v9 Q& u
191.                    "Book_Synopsis":response.save['Book_Synopsis'],
       ~% Y. t% e+ [5 E/ ~$ X
192.                    "Book_Palabras":response.save['Book_Palabras'],
     
193.                    "img":response.save['img'],
     
194.                      }
     1 j- V. K/ v. U3 A3 r
195.         for each in response.doc('.chapter-list li:first-child a[href^="http"]').items():
     6 a* J* m8 m5 I0 O
196. #        for each in response.doc('.chapter-list  a[href^="http"]').items():  
     
197.                     self.crawl(each.attr.href, callback=self.detail_page,save=Datos)
     2 h5 l8 l: r( s0 D/ G
198.     @config(priority=2)
     
199.     @catch_status_code_error
     3 I& |# E; d% z4 B5 D; K* H3 l
200.     def detail_page(self, response):        
     " D4 G& v) @( s
201.         NewRe1 = u'哈书'
     
202.         NewRe2 = u'huhjsd.CC'
     
203.         NewRe3 = r'^\\n\\n'
     
204.         NewRe5 = u'小说网'
     
205.         NewRe6 = u'fgdfgf'
     
206.         NewRe7 = u'fgfgf'
     : B5 C( a6 }  R8 ~# G
207.         NewRe8 = u'ffhgf'
     
208.         NewRe4 = r'[\f\t\v+\.\{\（\）\}\!\/_,$%^*(+"\')]+|[+——()?【】“”！，。？、~@#￥%……&*（）]+'
     
209.         ReC1 = u'静思'
     
210.         ReC2 = u'aghgf.com'
     
211.         ReC3 = u'aghgfh.com'
     
212.         ReC4 = u''
     # R, ]  w3 s4 P7 A, s% T: ]. d
213.         ReC5 = u'文学网'
     4 P' r' U4 }7 f
214.         ReC6 = r'<BR>'
     % S, a/ `/ e- }
215.         Bookname = response.doc('.readlocation a').eq(2).text()   #小说名称
     % S) n+ q# G) {
216.         print Bookname
     
217.         Cater_Name = response.save['Cater_Name']   # 小说分类
     
218.         Book_author = response.save['Book_author']   #小说作者
     
219.         Book_Introduction1 = response.save['Book_Introduction']   #小说简介
     
220.         Book_Synopsis = response.save['Book_Synopsis']   #最近更新
     
221.         Book_Palabras = response.save['Book_Palabras']   #小说字数
     " N+ r8 |$ D* m9 d2 y3 a( \( _( ^
222.         Bookurl = response.url   #小说网址
     
223.         Booktitle = response.doc('.article-title').text()   #章节名称
     ! K, q7 y; U( W. o* n* k: ]: f/ A
224.         BookID = response.doc('.readset-r span').text()   #小说ID
     ( P! U+ o* p: H) v2 Q0 d8 Z3 Z
225.         BookConte1 = response.doc('.article-con').text()   #小说章节内容
     # w  K( W4 c* \4 k+ Q) X
226.         abover = response.doc('.article-title').text() + response.save['Book_Synopsis'] + response.save['Book_Palabras'] + response.save['Book_Introduction']   #小说状态(连载还是完成)
     
227.         Book_Date = str(datetime.datetime.now())    # 采集时间
     
228.         BookConte2 = BookConte1.replace(NewRe1 , ReC1)
     9 E3 w4 J" I6 T, h  P
229.         BookConte3 = BookConte2.replace(NewRe2 , ReC2)
     ) W/ S' P5 w7 u2 s) A% ~
230.         BookConte5 = BookConte3.replace(NewRe5 , ReC5)
     $ @8 h) U" @8 g% Q4 i
231.         BookConte6 = BookConte5.replace(NewRe6 , ReC2)
     0 m  Q) v8 y3 R9 k
232.         BookConte7 = BookConte6.replace(NewRe7 , ReC2)
     % L  k& y" s3 i- f  Q
233.         BookConte8 = BookConte7.replace(NewRe3 , ReC6)
     8 ^. w7 ]+ ^) R
234.         BookConte4 = re.sub(NewRe4 , ReC4 , BookConte8)
     3 \+ H* j* [( ~4 k( }/ Z* r
235.         BookConte = BookConte4.replace("\n\n","<br>")
     
236.         print BookConte
     2 v0 w6 c; [. Q( j6 R* C0 l
237.         Book_Introduction2 = Book_Introduction1.replace(NewRe1 , ReC1)
     
238.         Book_Introduction3 = Book_Introduction2.replace(NewRe2 , ReC2)
     
239.         Book_Introduction4 = Book_Introduction3.replace(NewRe3 , ReC3)
     ' S% a  F$ @9 j9 _4 [4 g$ i$ N9 R
240.         Book_Introduction = re.sub(NewRe4 , ReC4 , Book_Introduction4)
     
241.         Titleid = response.url.split(BookID + "/")[-1].split("/")[0]     
     % `& C# h  o9 L4 K* }* Z
242.         Book_img = response.save['img'],  #小说图片
     
243.             
     ; I& P% O2 x( E; z: L" f5 o
244.         #insert into MySQL 小说入库
     2 e# `9 \( D( g' Q2 w& s
245.         self.add_question(Bookname,Cater_Name,Book_author,Book_Introduction,Book_Synopsis,Book_Palabras,Bookurl,Booktitle,BookID,BookConte,Titleid,abover,Book_Date)   #这行可注译，数据库发布接口，方便其他系统的发布
     * c, ]& e, b( R$ m9 c
246.         self.add_comment(Bookname, Booktitle, BookID, Titleid, Book_Date)   #这行可注译，数据库发布接口，方便其他系统的发布
     * ^( y2 g6 }+ `+ m# E& h. b5 S
247.         #post提交发布
     9 N* M) }' f3 y. z0 h
248.         self.add_locoy(Bookname,Cater_Name,Book_author,Book_Introduction,Book_Palabras,Book_img,Booktitle,BookConte,abover)  #这行可注译，火车头发布接口，不需要可取消
     " c8 e) t1 k, c! o1 F% S/ b/ E
249.         Datos = {
     
250.                   "Cater_Name":response.save['Cater_Name'],
     4 g0 u) r  {/ {# b6 f2 q; p$ V/ q, X
251.                    "Book_author":response.save['Book_author'],
     " a) F( {8 N  |6 p7 I7 y
252.                    "Book_Introduction":response.save['Book_Introduction'],
     : l( C; q4 W" |; L
253.                    "Book_Synopsis":response.save['Book_Synopsis'],
     ; i5 l7 [3 @, Z/ m6 i
254.                    "Book_Palabras":response.save['Book_Palabras'],
     
255.                    "img":response.save['img'],
     
256.                      }
     ' b% J3 t  N; s) [/ a
257.         for each in response.doc('.articlebtn > a:nth-child(4)[href*="/xiaoshuo"]').items():
     ( H. p; D2 \) q' t: R% G' |
258.             self.crawl(each.attr.href, callback=self.detail_page,save=Datos)
     " l# q/ o- s( ~0 k) r
259.         return {
     
260.             "Cater_Name":Cater_Name,
     
261.             "Bookname":Bookname,
     ) j2 h, @7 ?: }0 s! \7 B
262.             "Book_author":Book_author,
     
263.             "Book_Introduction":Book_Introduction,
     
264.             "Book_Synopsis":Book_Synopsis,
     # l$ A, `; N" R8 c7 O' N
265.             "Book_Palabras":Book_Palabras,
     $ D/ }% a2 K1 A) B
266.             "Book_img":Book_img,
     ; Q( c/ H) m. w3 b+ V3 D  z" ]. ^
267.             "Bookurl": response.url,
     ' s, \% c, k0 C( Y3 H/ b
268.             "Booktitle": Booktitle,
     
269.             "BookID": BookID,
     
270.             "BookConte": BookConte,
     
271.             "Titleid": Titleid,
     
272.             "abover":abover,
     7 u' Y* o6 ]* j/ f. F' J: m% q
273. #            "Book_Date" = str(datetime.datetime.now()),
     
274.         }
     1 _8 ^0 t9 D) Q; K, \1 e  I
275.     def download(self, P_dir, imgDir, file_name, Book_img):
     
276.         if not os.path.exists(imgDir):
     9 C& U; t& X8 V8 g
277.             os.makedirs(imgDir)
     
278.         file = imgDir + "/" + file_name
     
279. #        print file
     
280.         f = open(file, 'wb+')
     ( Q) n4 \2 \4 q6 C( V" m2 v
281.         imag = requests.get(Book_img)
     
282.         f.write(imag.content)
     
283.         f.close()
     
284.         #保存图片前
     
285.     def save_imgs(self,response):
     
286.         content = response.content
     ; |# n3 y( P% o: T* H
287.         file_name = response.save["file_name"]
     
288.         imgDir = response.save["imgDir"]
     
289.         file_path = imgDir + file_name
     
290.         self.save_img(content,imgDir,file_path)
     
291.     #保存图片
     : G+ p3 }; f4 p" u' M
292.     def save_img(self,content,imgDir,path):
     0 Q! g& p8 ^8 S( a& m) t( o% m* ?
293.         if not os.path.exists(imgDir):                        
     ; F5 N, I& [; b7 e4 ^
294.             os.makedirs(imgDir)
     - {# B0 r  B5 I# q: u
295.         f = open(path,"wb" )
     
296.         f.write(content)
     8 l/ L5 |- o9 ^/ O
297.         f.close()
     & ?3 g( C3 M$ r% v. b" P8 T
298.     #获取url后缀名
     - I. X9 Q  z3 T& e' C) q
299.     def getExtension(self,url):                           
     
300.         extension = url.split(".")[-1]
     
301.         return extension
     
302.    
     
303.     #获取图片名
     6 @. |+ e: d& O7 I
304.     def getname(self,url):
     - K3 n1 E+ P$ {) E- x& J
305.         name=url.split("/")[-1].split(".")[0]
     
306.         return name

复制代码

5 d4 f6 T# d2 r% w! P