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

fei

Python + pyspider某小说站的爬虫，入数据库，火车头发布，资源下载到本地，另可写爬虫！

1. #!/usr/bin/env python
   ; W9 Y/ {- V8 }% c! A
2. # -*- encoding: utf-8 -*-
   5 ]5 P  R8 j% R" e# A- R( q, |2 S
3. # Created on 2019-05-05 21:43:11
   
4. # Project: XiaoShuo
   
5. 
   
6. from pyspider.libs.base_handler import *
   ( C1 b' f* v( |1 Q9 |0 |- d
7. import pymysql
   : P& ?( j- X6 ~$ {7 U# f
8. import random
   
9. import datetime
   ; f3 C9 W2 A. z, J8 `
10. import urllib2,HTMLParser,re
    & @/ o4 S9 R( x  `4 u
11. import os
    ; w) {4 p, D6 \' i  r! E
12. import sys
    
13. import re
    * \  W- N! o' P# b9 f
14. import codecs
    - ?+ {' y7 A( D  \1 f
15. import requests
    ) c& C, H  u" p, Z" c+ Q' H2 D- f
16. import json
    
17. 
    
18. class Handler(BaseHandler):
    6 f  X. N/ e% s8 X! i: ?
19.     global Datos
    
20.     global P_dir   
    
21.     P_dir = '/Tools/Debug/'  #采集时候图片保持到本地的路径
    ; `  ~( R, J1 j7 B0 [; \3 I
22.     global Datos
    
23.     Datos = {}
    ' C* }. r# O1 m+ F' Z2 p
24.     headers= {
    
25.     'Accept':'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
    * Q* _) j" v" @$ @
26.     'Accept-Encoding':'gzip, deflate, sdch',
    : a# A" n4 h7 H% k9 F5 G
27.     'Accept-Language':'zh-CN,zh;q=0.8',
    / r1 @3 J% |$ ]* ?2 S+ l
28.     'Cache-Control':'max-age=0',
      w  _8 M( a+ B. p! \
29.     'Connection':'keep-alive',
    ! F& q# h2 u, R2 z6 M, @& f
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'
    9 C" ^/ K+ h1 q. G* a/ z
31.     }
    4 u9 r3 b9 M$ Z! M9 U$ k
32.     crawl_config = {
    * k+ ~' T. G( _$ ]) ^( R
33.         'headers' : headers,
    ( O- T' ?9 I: g9 v6 g
34.         'timeout' : 300
    % r  f/ L$ w7 {, |5 p/ f$ K
35.     }
    
36.     def add_BookFile(self,Bookname, BookIDs, img, Locaimg, Book_Dates):
    
37.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
    % o5 `; N% L7 J# u6 D) p
38.         try:
    3 O/ s5 z, W' E: m6 E+ `
39.             cursor = db.cursor()
    
40.             #注意此处字符串的占位符要加双引号"%s"
    ! A/ U4 X9 A" X, g) `
41.             sql = 'insert into BookFile(Bookname, BookID, img, Locaimg, Book_Date) values ("%s","%s","%s","%s","%s")' % (Bookname, BookIDs, img, Locaimg, Book_Dates);
    : i$ R6 O# Y4 |* q9 Z
42. #            print(sql)
    - ]0 U$ T5 ~$ K/ \! [
43.             cursor.execute(sql)
    % Z) o6 N4 x: R8 g+ N1 x
44.             
    
45.             #qid = cursor.lastrowid
    
46.             #print(qid)
    
47.             
    ! ]1 q* d- I* ]
48.             db.commit()
    
49.         except Exception as err:
      Z' y! H7 N2 {
50.             print("Error %s for execute sql: %s" % (err, sql))
    
51.             db.rollback()
    
52.     def add_comment(self,Bookname, Booktitle, BookID, Titleid, Book_Date):
    
53.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
    * |! U+ t; J6 m1 F. T. M
54.         try:
    
55.             cursor = db.cursor()
      l* K4 a' [: O7 x
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);
    9 Y, ^  o7 `' Q+ T2 @& {
58. #            print(sql)
    : R1 W9 K1 r4 o6 X
59.             cursor.execute(sql)
    
60.             
    8 s- q; |3 `; y5 b9 A
61.             #qid = cursor.lastrowid
    - ~* J0 o2 j4 y% I, J
62.             #print(qid)
    # ]. Q2 I! a* V- @/ a
63.             
    , |7 p* K1 F$ j3 h  ^- {
64.             db.commit()
    
65.         except Exception as err:
    / d. ^  Q3 h& X: M! x. a' N; {
66.             print("Error %s for execute sql: %s" % (err, sql))
    : c7 q# q" b, D
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):
    5 q% n5 h2 f, z- g# g" {& i" |
69.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
    
70.         try:
    
71.             cursor = db.cursor()
    3 g2 v1 {# g' D
72.             #注意此处字符串的占位符要加双引号"%s"
    
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);
    # f& {4 h& w. e
74.             print(sql)
    ( O9 d: s, ~  i
75.             cursor.execute(sql)
    
76.             print(cursor.lastrowid)
    
77.             db.commit()
    7 q% M& {+ j4 E! r) x- }
78.         except Exception as err:
    
79. #        except:
    
80. #            print('Failed')
    
81.             print("Error %s for execute sql: %s" % (err, sql))
      ?! p% q+ _- |: _$ Y
82.             db.rollback()
    
83.         
    
84.     def add_locoy(self,Bookname,Cater_Name,Book_author,Book_Introduction,Book_Palabras,Book_img,Booktitle,BookConte,abover):
    & j, t" y! v" _8 {
85.             reload(sys)
    % T) t3 j) E& b3 P" n9 o1 n
86.             sys.setdefaultencoding("gbk")
    . x  L) O5 \  b3 B. m% D; A
87.             locoy_url = 'http://www.******.net/locoy/?my=book'  #697火车头发接口地址
    
88.             locoy_data = {
    
89.             'my_u':'用户名',   #后台用户名
    
90.             'my_p':'密码',   #后台密码
    6 n7 x* N: J/ A& P$ m/ L' E' P
91.             'subject_669977_net':Bookname.encode('gbk', 'ignore'),
    
92.             'caid':Cater_Name.encode('gbk', 'ignore'),
    7 U8 a7 m2 }) [9 k) W8 Y
93.             'title_669977_net':Booktitle.encode('gbk', 'ignore'),
    
94.             'article':BookConte.encode('gbk', 'ignore'),
    ' s. D3 |8 v$ e, U
95.             'author':Book_author.encode('gbk', 'ignore'),
    & P# b* F& y' |) Z- F# `+ _
96.             'ready_1':Book_Palabras.encode('gbk', 'ignore'),
    : X8 d- ?* ~8 _, `& G% A# M
97.             'thumb':Book_img,
    
98.             'content':Book_Introduction.encode('gbk', 'ignore'),
    ) q! t7 @1 [$ W' W0 B
99.             'abover':abover.encode('gbk', 'ignore')           
    
100.                 }
     ( j" R& v5 u/ ^  X7 y7 c
101.             res = requests.post(locoy_url, data=locoy_data)
     2 S$ F$ Q5 H3 m5 }7 G7 J1 ]
102.             print res.text
     
103.             print res.content
     
104. #            print Dsd
     . M% r( M% w5 P- Y1 z1 Q2 G
105.             return res
     
106.    
     8 y  L6 S9 H: S
107.     def __init__(self):
     . C) c( u( ~3 p! R  P0 k( p
108.         self.base_url1 = 'https://www.****.cc/'
     : ?2 W0 y. T1 Q. S  S1 }) j0 y- N+ b
109.         self.base_url2 = '/'
     
110.         self.CaterId = []
     
111.         self.CaterIds = ['xuanhuan', 'wuxia', 'yanqing', 'lishi', 'kehuan', 'kongbu', 'nvsheng']
     $ B$ O' C0 n2 U7 Q7 w
112.         self.page_num = 1
     
113.         self.total_num = 200   
     # h0 G6 \0 N) w. E  k1 _
114. 
     
115.     @every(minutes=8 * 60)
     0 V8 M. C" ]0 q0 U2 Y$ z3 x/ s  ^
116.     def on_start(self):
     . o; g+ o) y; Y# C& }& G* D. b$ l
117.         global Cater_Name
     ! ]$ j4 U% v; _0 `; S! r6 I: }
118.         Cater_Name = []
     
119.         while self.page_num <= self.total_num:
     1 {8 ?' `. _  B
120.             for self.CaterId in self.CaterIds:
     
121.                 if self.CaterId  == 'xuanhuan':
     6 W. R8 U$ \; B* K
122.                      Cater_Name = '玄幻'
     
123.                 if self.CaterId  == 'wuxia':
     ; j# y* t: N4 C
124.                     Cater_Name = '武侠'
     8 W$ d5 e. {' l7 W1 r- E, G
125.                 if self.CaterId  == 'lishi':
     $ F$ v/ S' l: ?: y; r
126.                     Cater_Name = '历史'            
     ! m6 m- }. r$ J+ B4 f& p
127.                 if self.CaterId  == 'yanqing':
     
128.                     Cater_Name = '都市'
     / \7 ~! t  f. [  w
129.                 if self.CaterId  == 'nvsheng':
     
130.                     Cater_Name = '都市'
     
131.                 if self.CaterId  == 'kehuan':
     
132.                     Cater_Name = '科幻'
     
133.                 if self.CaterId  == 'kongbu':
     
134.                     Cater_Name = '游戏'
     
135.                 print self.CaterId
     
136.                 url = self.base_url1 + str(self.CaterId) + self.base_url2 + str(self.page_num) + "/"         
     
137.                 self.crawl(url, callback=self.list_Caterg,save=Cater_Name)
     , K) y+ ]$ n8 ~) I3 ^
138.             self.page_num += 1
     
139.             
     $ C3 k; \+ N' J/ e4 Z0 i
140.     def list_Caterg(self, response):
     ' u7 C; l/ N8 p$ v) Y; R/ L, c: @
141.         Cater_Name = response.save
     
142.         for each in response.doc('.pic-list a[href^="http"]').items():
     2 Q5 o4 R+ z; D- m" W1 R
143.             self.crawl(each.attr.href, callback=self.list_Caterg_detail,save=Cater_Name)
     , v4 o1 ^; E1 w3 J3 Z" `
144.             
     
145.     def list_Caterg_detail(self, response):
     
146.         Cater_Name = response.save
     & _0 a& \  G  j- X4 A& Q: f
147. #        print Cater_Name
     9 R! Y" q5 A# P; D; s7 ]2 S. h# S
148.         Bookname = response.doc('h1').text()
     * _8 x! S! \' E! N3 G# s
149.         print Bookname
     
150.         Book_author = response.doc('.authorname > a').text()
     ! C- C5 |1 k- F3 R
151. #        print Book_author
     
152.         Book_Introduction = response.doc('.book-intro > div').text()
     : C$ b) A! ]  ^- [
153. #        print Book_Introduction
     
154.         Book_Synopsis = response.doc('b').eq(1).text()
     & Y  K. o9 ~& u
155. #        print Book_Synopsis
     7 ?; H! v* s# l3 ~: i
156.         Book_Palabras = response.doc('.booktitle p').text().split(' ')[1].split('|')[0]
     
157. #        print Book_Palabras
     & N" {% w7 m8 s% A8 P. i& U
158.         BookIDs = response.url.split("xiaoshuo/")[-1].split("/")[0]   #小说ID
     , G# p5 t# n3 w( y' H
159. #        print BookIDs
     . T- X& k4 Q' t1 Q
160.         Book_Dates = str(datetime.datetime.now())         
     ) U. m( V, f. ^7 W: |, s+ X1 I. O" o
161.         for imgs in response.doc('.bigpic > img[src^="http"]').items():
     $ G& F5 v6 x% T$ z8 x7 R0 ^+ M
162.             img = imgs.attr.src
     7 {. ]/ h- x2 k
163.             print img
     
164.                 #小说封面下载
     / c. C  b/ a- `. E
165.             extension = self.getExtension(img)
     ) t, T! R) G2 p% e& [0 o
166.             name = self.getname(img)
     
167.             file_name = name + "." + extension
     2 Q$ ?, |% G" _! J' A! L7 [
168.             imgDir = P_dir + name
     . D  t; i+ B' x, ]1 I5 `$ e3 P
169.             Locaimg = imgDir + "/" + file_name
     # N# h5 x% M% B7 k/ m( l8 p- L
170.             print Locaimg
     
171.             if(self.download(P_dir, imgDir, file_name, img)):   #这2行可注译，图片下载到本地
     
172.                 print('attachment url is ' + img)               #
     
173.             Datos = {
     , v* F( j- N% X8 j. k7 d, [4 n
174.                     "Cater_Name":Cater_Name,
     7 J' u+ P, B; Y5 r" w" L# d9 i
175.                     "Book_author":Book_author,
     " N2 L6 J* @/ [# E- G
176.                     "Book_Introduction":Book_Introduction,
     ) J; @, Q# M( I$ f
177.                     "Book_Synopsis":Book_Synopsis,
     4 }/ u/ f. E! |& H3 o
178.                     "Book_Palabras":Book_Palabras,
     
179.                     "img":img,
     
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():
     . H  ~( j3 d' c  x; b6 T. z
183.             self.crawl(each.attr.href, callback=self.index_page,save=Datos)
     
184.             
     ; I, V" D$ a1 z# T  T: v& W
185.     @config(age=8 * 60 * 60)   
     
186.     def index_page(self, response):
     
187.         Datos = {
     
188.                   "Cater_Name":response.save['Cater_Name'],
     
189.                    "Book_author":response.save['Book_author'],
     7 W6 r- G3 w4 h" u& u
190.                    "Book_Introduction":response.save['Book_Introduction'],
     ; ^) |9 C( S2 O7 O. \2 }* V" {
191.                    "Book_Synopsis":response.save['Book_Synopsis'],
     7 s* H' K3 J: Q6 k$ M! }
192.                    "Book_Palabras":response.save['Book_Palabras'],
     
193.                    "img":response.save['img'],
     3 k3 J. U% w+ P5 L+ b
194.                      }
     . c0 L; \5 z$ X5 u; u
195.         for each in response.doc('.chapter-list li:first-child a[href^="http"]').items():
     / ?* P! Y9 \1 t. g7 G
196. #        for each in response.doc('.chapter-list  a[href^="http"]').items():  
     
197.                     self.crawl(each.attr.href, callback=self.detail_page,save=Datos)
     
198.     @config(priority=2)
     $ h  O3 ?9 t* o
199.     @catch_status_code_error
     % ~2 ]! n; g+ A+ n- |) W# {' P1 V
200.     def detail_page(self, response):        
     ! s5 ?+ S( B: G
201.         NewRe1 = u'哈书'
     
202.         NewRe2 = u'huhjsd.CC'
     0 K+ c. o8 j& g& b- `4 c
203.         NewRe3 = r'^\\n\\n'
     2 L" r5 ]+ j  k
204.         NewRe5 = u'小说网'
     
205.         NewRe6 = u'fgdfgf'
     " f" J! @" R2 S2 y
206.         NewRe7 = u'fgfgf'
     
207.         NewRe8 = u'ffhgf'
     7 `9 }" D& r" |9 _
208.         NewRe4 = r'[\f\t\v+\.\{\（\）\}\!\/_,$%^*(+"\')]+|[+——()?【】“”！，。？、~@#￥%……&*（）]+'
     , Q: m; }4 }7 e
209.         ReC1 = u'静思'
     
210.         ReC2 = u'aghgf.com'
     2 [, c0 Z5 P( ]  N% S! Y2 O8 G$ k
211.         ReC3 = u'aghgfh.com'
     
212.         ReC4 = u''
     7 L* T+ x' D+ p
213.         ReC5 = u'文学网'
     ) B2 R6 n' u; v9 t* h
214.         ReC6 = r'<BR>'
     6 n# e0 Q$ t4 l$ S
215.         Bookname = response.doc('.readlocation a').eq(2).text()   #小说名称
     / ~. z3 E, s" X! A) d& X7 x) W9 Y6 c/ {
216.         print Bookname
     
217.         Cater_Name = response.save['Cater_Name']   # 小说分类
     : K% t2 n8 G" g2 s" u( [
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']   #小说字数
     ) |. w  a6 P! `: e2 q# z/ `, P: X
222.         Bookurl = response.url   #小说网址
     
223.         Booktitle = response.doc('.article-title').text()   #章节名称
     
224.         BookID = response.doc('.readset-r span').text()   #小说ID
     
225.         BookConte1 = response.doc('.article-con').text()   #小说章节内容
     
226.         abover = response.doc('.article-title').text() + response.save['Book_Synopsis'] + response.save['Book_Palabras'] + response.save['Book_Introduction']   #小说状态(连载还是完成)
     / q  v0 U  z: j
227.         Book_Date = str(datetime.datetime.now())    # 采集时间
     & ^# H/ d1 _1 R  m# ~
228.         BookConte2 = BookConte1.replace(NewRe1 , ReC1)
       o- L# H3 k* b3 C7 o( L8 g
229.         BookConte3 = BookConte2.replace(NewRe2 , ReC2)
     * ]* s+ A5 z1 {/ Z
230.         BookConte5 = BookConte3.replace(NewRe5 , ReC5)
     * }4 Z& R) O" ~8 M+ q4 l  E9 n& v/ @
231.         BookConte6 = BookConte5.replace(NewRe6 , ReC2)
     
232.         BookConte7 = BookConte6.replace(NewRe7 , ReC2)
     
233.         BookConte8 = BookConte7.replace(NewRe3 , ReC6)
     9 R% _9 d+ B& Q: a
234.         BookConte4 = re.sub(NewRe4 , ReC4 , BookConte8)
     
235.         BookConte = BookConte4.replace("\n\n","<br>")
     0 c! n, q% W6 z- G' T1 N
236.         print BookConte
     
237.         Book_Introduction2 = Book_Introduction1.replace(NewRe1 , ReC1)
     
238.         Book_Introduction3 = Book_Introduction2.replace(NewRe2 , ReC2)
     , z8 l1 p% l. [* Y- }; U" \
239.         Book_Introduction4 = Book_Introduction3.replace(NewRe3 , ReC3)
     
240.         Book_Introduction = re.sub(NewRe4 , ReC4 , Book_Introduction4)
     / I: a3 v; {6 w9 O- l; B" A" b$ W; F
241.         Titleid = response.url.split(BookID + "/")[-1].split("/")[0]     
     
242.         Book_img = response.save['img'],  #小说图片
     
243.             
     : E5 j9 y+ g4 q2 m2 ?5 S0 h2 _
244.         #insert into MySQL 小说入库
     7 T% {" I; y! f9 L" i2 j8 w& F
245.         self.add_question(Bookname,Cater_Name,Book_author,Book_Introduction,Book_Synopsis,Book_Palabras,Bookurl,Booktitle,BookID,BookConte,Titleid,abover,Book_Date)   #这行可注译，数据库发布接口，方便其他系统的发布
     1 i+ \7 q2 e) ]- t8 Q
246.         self.add_comment(Bookname, Booktitle, BookID, Titleid, Book_Date)   #这行可注译，数据库发布接口，方便其他系统的发布
     
247.         #post提交发布
     
248.         self.add_locoy(Bookname,Cater_Name,Book_author,Book_Introduction,Book_Palabras,Book_img,Booktitle,BookConte,abover)  #这行可注译，火车头发布接口，不需要可取消
     ; C; o; Y8 A: G* P& a6 v5 C
249.         Datos = {
     4 O0 e! z. Z3 k" w4 c# A/ _
250.                   "Cater_Name":response.save['Cater_Name'],
     
251.                    "Book_author":response.save['Book_author'],
     7 x# _/ n- O  R/ u2 h  e+ p
252.                    "Book_Introduction":response.save['Book_Introduction'],
     * N' _. T* h: {- s. r  U" N" H
253.                    "Book_Synopsis":response.save['Book_Synopsis'],
     7 Y$ t) K6 Y; O8 y6 E1 {
254.                    "Book_Palabras":response.save['Book_Palabras'],
     
255.                    "img":response.save['img'],
     
256.                      }
     $ ^+ w3 s9 A: J; d( z0 |- R4 `+ z
257.         for each in response.doc('.articlebtn > a:nth-child(4)[href*="/xiaoshuo"]').items():
     5 w% l/ f& O7 c9 E
258.             self.crawl(each.attr.href, callback=self.detail_page,save=Datos)
     6 m% Z3 ^* I+ |' K5 _* R
259.         return {
     
260.             "Cater_Name":Cater_Name,
     
261.             "Bookname":Bookname,
     7 z" C  @9 U9 b6 B
262.             "Book_author":Book_author,
     
263.             "Book_Introduction":Book_Introduction,
     + V; B- L% I6 I1 y4 e
264.             "Book_Synopsis":Book_Synopsis,
     0 }/ q8 [3 L" k+ o
265.             "Book_Palabras":Book_Palabras,
     2 u7 ^* R6 x  ]( x, v. j/ \! C( e- b
266.             "Book_img":Book_img,
     
267.             "Bookurl": response.url,
     
268.             "Booktitle": Booktitle,
     
269.             "BookID": BookID,
     
270.             "BookConte": BookConte,
     
271.             "Titleid": Titleid,
     9 q$ i! R; r% s! m( k
272.             "abover":abover,
     
273. #            "Book_Date" = str(datetime.datetime.now()),
     
274.         }
     - h6 Z/ A2 ^, S- T; X. y+ w
275.     def download(self, P_dir, imgDir, file_name, Book_img):
     # Q  ^* m- `& A3 O8 `6 o% F+ G
276.         if not os.path.exists(imgDir):
     1 [$ \% B1 g* i4 i9 f/ O6 H
277.             os.makedirs(imgDir)
     5 D# D) R4 m* Q0 X2 p2 m* i
278.         file = imgDir + "/" + file_name
     8 L- |  o7 o5 Y: S
279. #        print file
     
280.         f = open(file, 'wb+')
     . G# D9 X( @, G& P
281.         imag = requests.get(Book_img)
     
282.         f.write(imag.content)
     
283.         f.close()
     6 W; u2 V  C7 N
284.         #保存图片前
     
285.     def save_imgs(self,response):
     
286.         content = response.content
     
287.         file_name = response.save["file_name"]
     * E8 c8 |5 c  _* s2 ?# d
288.         imgDir = response.save["imgDir"]
     3 ~. k# w2 G2 O( z
289.         file_path = imgDir + file_name
     
290.         self.save_img(content,imgDir,file_path)
     7 n8 Q9 Q: }5 P& Y
291.     #保存图片
     7 Q3 k5 }( X$ b  p$ w3 q" S, b
292.     def save_img(self,content,imgDir,path):
     
293.         if not os.path.exists(imgDir):                        
     9 v* D/ P% R0 _- T& b+ A
294.             os.makedirs(imgDir)
     9 p  }. [5 I1 r1 v1 p8 K& |  \0 l3 C# `
295.         f = open(path,"wb" )
     ( e7 y8 @, o0 j" c9 m1 Y' x) a
296.         f.write(content)
     
297.         f.close()
     
298.     #获取url后缀名
     6 o, p+ ?( q, U0 {% w
299.     def getExtension(self,url):                           
     6 d# I: k- H8 |9 M; F$ v
300.         extension = url.split(".")[-1]
     * B8 _7 ]7 f% I$ B+ H
301.         return extension
     
302.    
     
303.     #获取图片名
     
304.     def getname(self,url):
     ; S! q1 v# |' X$ Q+ u
305.         name=url.split("/")[-1].split(".")[0]
     
306.         return name

复制代码

9 s% G6 F4 M3 h) v/ E