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

fei

Python + pyspider某小说站的爬虫，入数据库，火车头发布，资源下载到本地，另可写爬虫！
9 q( M: @- b% |3 I# |' ~, Y

1. #!/usr/bin/env python
   : _& P' X8 f1 `5 a
2. # -*- encoding: utf-8 -*-
   $ E5 \! `. f9 C& P: A
3. # Created on 2019-05-05 21:43:11
   
4. # Project: XiaoShuo
   
5. 
   # Z2 V/ T$ w) G- h: r- x
6. from pyspider.libs.base_handler import *
   
7. import pymysql
   1 f7 e: G- C5 W# ~
8. import random
   
9. import datetime
   
10. import urllib2,HTMLParser,re
    
11. import os
    
12. import sys
    
13. import re
    
14. import codecs
    
15. import requests
    
16. import json
    
17. 
    6 X; b* |% x/ l/ W) c, E' ~; S! G
18. class Handler(BaseHandler):
    
19.     global Datos
    % F8 a. w5 N5 d0 W6 T
20.     global P_dir   
    
21.     P_dir = '/Tools/Debug/'  #采集时候图片保持到本地的路径
    
22.     global Datos
    
23.     Datos = {}
    
24.     headers= {
    0 S- [# Q! d( J  N: K8 P
25.     'Accept':'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
    # o3 n5 ^9 P) x" |3 L0 O
26.     'Accept-Encoding':'gzip, deflate, sdch',
    
27.     'Accept-Language':'zh-CN,zh;q=0.8',
    - b4 Q. \2 ~+ U% G
28.     'Cache-Control':'max-age=0',
    - _9 c8 U. }4 j* ~/ i, W' @
29.     'Connection':'keep-alive',
    * r2 F) h2 }/ l9 e6 E4 w# j# B
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'
    
31.     }
    6 e" {; W1 a, I* X; \5 j( B# H
32.     crawl_config = {
    8 i! b0 {2 m5 ]. H
33.         'headers' : headers,
    3 T; G6 e3 p1 X' i) d9 D
34.         'timeout' : 300
    
35.     }
    
36.     def add_BookFile(self,Bookname, BookIDs, img, Locaimg, Book_Dates):
    
37.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
    
38.         try:
    + ?, u7 ~5 z3 G2 F
39.             cursor = db.cursor()
    % C5 q4 Q0 |* ]4 {: E
40.             #注意此处字符串的占位符要加双引号"%s"
    
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)
    $ \* Q7 ^' Q+ U* }9 R1 N4 X4 y5 N/ J
43.             cursor.execute(sql)
    5 a  E7 W6 ]/ e: z, Y/ R3 M4 i3 Z/ i. f
44.             
    0 V4 }6 k% }% S, p( u7 a
45.             #qid = cursor.lastrowid
    ' S# t+ X; y1 t/ h1 {% p
46.             #print(qid)
    * j% {. Z) M0 V$ `7 P
47.             
    ) W/ c) o- }' y; J* A! T9 I8 ]! s7 k7 s
48.             db.commit()
    
49.         except Exception as err:
    
50.             print("Error %s for execute sql: %s" % (err, sql))
    : l" j( X* S4 X: y+ ^2 u" w6 Q
51.             db.rollback()
    % U0 ^# i8 P, D! E9 U% x) Y8 b; _# k
52.     def add_comment(self,Bookname, Booktitle, BookID, Titleid, Book_Date):
    / o5 {$ w  O( g1 H- \
53.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
    
54.         try:
    
55.             cursor = db.cursor()
    ) f3 V8 A4 a! S! t
56.             #注意此处字符串的占位符要加双引号"%s"
    ! r6 w8 J" F4 y6 `9 p& r
57.             sql = 'insert into BookTitle(Bookname, Booktitle, BookID, Titleid, Book_Date) values ("%s","%s","%s","%s","%s")' % (Bookname, Booktitle, BookID, Titleid, Book_Date);
    # \$ |4 Z& p1 n6 E8 S
58. #            print(sql)
    
59.             cursor.execute(sql)
    # h# T5 o  J  ?  I7 N8 v/ C
60.             
    & f1 s0 w, P3 U/ Q; v
61.             #qid = cursor.lastrowid
    % s/ U/ s4 f6 g# a0 o5 S
62.             #print(qid)
    # c5 j! R, h: u5 Q
63.             
    
64.             db.commit()
    - a. [. F4 Z) g: r! u7 c
65.         except Exception as err:
    
66.             print("Error %s for execute sql: %s" % (err, sql))
    
67.             db.rollback()
    " A* G" P, J/ H. n! J  a+ [
68.     def add_question(self,Bookname,Cater_Name,Book_author,Book_Introduction,Book_Synopsis,Book_Palabras,Bookurl,Booktitle,BookID,BookConte,Titleid,abover,Book_Date):
    
69.         db = pymysql.connect(host="localhost",user="数据库用户名",password="密码",db="数据库名",charset="utf8")
    
70.         try:
    9 d. P* D1 |9 C. h  L
71.             cursor = db.cursor()
    ( M/ m6 [* h- p8 {2 I- Q
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);
    : J: x, r# ~+ m. T" N& S) J
74.             print(sql)
    0 a( u7 f. F% T0 p
75.             cursor.execute(sql)
    6 g9 i; T9 u: o5 B7 q- E
76.             print(cursor.lastrowid)
    
77.             db.commit()
    * }3 S1 ^. L+ R( Y% h( f
78.         except Exception as err:
    2 W3 ^1 b) c4 J( r! s8 N
79. #        except:
    
80. #            print('Failed')
    
81.             print("Error %s for execute sql: %s" % (err, sql))
    
82.             db.rollback()
    
83.         
    ' I4 t6 n& A* |, ~" u4 {6 B- ]
84.     def add_locoy(self,Bookname,Cater_Name,Book_author,Book_Introduction,Book_Palabras,Book_img,Booktitle,BookConte,abover):
    / \: [( h, v% m1 C- R9 _/ v
85.             reload(sys)
    + m. U# S: A% C1 ^- f
86.             sys.setdefaultencoding("gbk")
    
87.             locoy_url = 'http://www.******.net/locoy/?my=book'  #697火车头发接口地址
    
88.             locoy_data = {
    ! I  a6 n, v/ s, W3 a& c- h( n
89.             'my_u':'用户名',   #后台用户名
    6 B3 J! f2 U( S4 n3 r; H
90.             'my_p':'密码',   #后台密码
    $ Q) s! ^. `$ |# h9 e8 ?% |
91.             'subject_669977_net':Bookname.encode('gbk', 'ignore'),
    
92.             'caid':Cater_Name.encode('gbk', 'ignore'),
    
93.             'title_669977_net':Booktitle.encode('gbk', 'ignore'),
    0 e  z9 l- |% B( H" X$ q" K
94.             'article':BookConte.encode('gbk', 'ignore'),
    3 U; y* ]6 @3 O: J
95.             'author':Book_author.encode('gbk', 'ignore'),
    ( a' l& r1 d* R7 B
96.             'ready_1':Book_Palabras.encode('gbk', 'ignore'),
    6 t* m3 S3 D# F* g- r* l
97.             'thumb':Book_img,
    
98.             'content':Book_Introduction.encode('gbk', 'ignore'),
    . e: P5 D) @- D. g( `" [1 s
99.             'abover':abover.encode('gbk', 'ignore')           
    
100.                 }
     * V! ?% V- }/ o7 S
101.             res = requests.post(locoy_url, data=locoy_data)
     # k8 A" L( F, \$ a) p2 @" I
102.             print res.text
     
103.             print res.content
     1 G. T7 Z) G4 e0 s* d& a
104. #            print Dsd
     
105.             return res
     + H% n; H$ S, e* H/ Y
106.    
     
107.     def __init__(self):
     ) a1 S  N, @( C  C# H) F6 [
108.         self.base_url1 = 'https://www.****.cc/'
     # C) U" K3 M2 [% @* x' t3 F
109.         self.base_url2 = '/'
     
110.         self.CaterId = []
     % V/ u" s& e$ b$ s6 g
111.         self.CaterIds = ['xuanhuan', 'wuxia', 'yanqing', 'lishi', 'kehuan', 'kongbu', 'nvsheng']
     8 H; c3 k1 t0 Z* T; G, G" a
112.         self.page_num = 1
     5 n, u+ _* e8 E0 u9 s4 C9 W% {! y
113.         self.total_num = 200   
     + e9 X4 Y5 C/ d$ j, m
114. 
     : D$ S* a% x; s% B  L" u. C
115.     @every(minutes=8 * 60)
     
116.     def on_start(self):
     
117.         global Cater_Name
     
118.         Cater_Name = []
     
119.         while self.page_num <= self.total_num:
     ; n+ G5 t, ^1 U3 j7 }( L
120.             for self.CaterId in self.CaterIds:
     8 W: p+ R( ~2 }, E
121.                 if self.CaterId  == 'xuanhuan':
     
122.                      Cater_Name = '玄幻'
     
123.                 if self.CaterId  == 'wuxia':
     
124.                     Cater_Name = '武侠'
     ' j: f, J; |! W: s# _4 i3 r
125.                 if self.CaterId  == 'lishi':
     & D* i# V  D6 v5 u9 z# M
126.                     Cater_Name = '历史'            
     % r& o% Q) Y9 ]6 w: f3 R6 A1 i4 t
127.                 if self.CaterId  == 'yanqing':
     
128.                     Cater_Name = '都市'
     " L8 x' {, N& ?
129.                 if self.CaterId  == 'nvsheng':
     " W' m; q7 E  N
130.                     Cater_Name = '都市'
     
131.                 if self.CaterId  == 'kehuan':
     # o- l; i" z- W9 e! B
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)
     ; s  K' z* u: B" p4 ^5 _0 ]
138.             self.page_num += 1
     7 l% b0 u; Q) H7 k2 X7 _, q2 K
139.             
     
140.     def list_Caterg(self, response):
     
141.         Cater_Name = response.save
     
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)
     / X9 D3 Q. A5 d, N1 p9 X  d& z
144.             
     
145.     def list_Caterg_detail(self, response):
     ( q6 _" G/ v. K
146.         Cater_Name = response.save
     
147. #        print Cater_Name
     , [* P; g9 s( u  s* I$ R
148.         Bookname = response.doc('h1').text()
     
149.         print Bookname
     1 R8 z: y: c8 L3 D8 q/ G4 ^! w2 j$ Y
150.         Book_author = response.doc('.authorname > a').text()
     0 J) j# _0 }. b$ `! E. v4 D+ f8 c8 ]2 R
151. #        print Book_author
     ( w+ B/ Y: }& g
152.         Book_Introduction = response.doc('.book-intro > div').text()
     ! q( O4 v. v4 W
153. #        print Book_Introduction
     ' V3 I, {- d' {3 ^) i3 h
154.         Book_Synopsis = response.doc('b').eq(1).text()
     
155. #        print Book_Synopsis
     $ Q/ b# R) V0 b) r
156.         Book_Palabras = response.doc('.booktitle p').text().split(' ')[1].split('|')[0]
     
157. #        print Book_Palabras
     
158.         BookIDs = response.url.split("xiaoshuo/")[-1].split("/")[0]   #小说ID
     / H# C% v/ Z* T" g" K  b
159. #        print BookIDs
     
160.         Book_Dates = str(datetime.datetime.now())         
     - _! U/ b" d! ]# u6 H, ^
161.         for imgs in response.doc('.bigpic > img[src^="http"]').items():
     ' `8 T4 [; j* B
162.             img = imgs.attr.src
     ( U; R% s2 S* `. H( _( w
163.             print img
     
164.                 #小说封面下载
     
165.             extension = self.getExtension(img)
     / t' }/ U2 i0 q# U/ n
166.             name = self.getname(img)
     
167.             file_name = name + "." + extension
     / S* E* U6 q: w& Z
168.             imgDir = P_dir + name
     2 S: J* q) f4 G3 U! C- {- X
169.             Locaimg = imgDir + "/" + file_name
     
170.             print Locaimg
     6 A0 I1 T2 T6 B3 R
171.             if(self.download(P_dir, imgDir, file_name, img)):   #这2行可注译，图片下载到本地
     ( n. q0 r+ e0 [, U3 t9 e2 c
172.                 print('attachment url is ' + img)               #
     
173.             Datos = {
     ! T0 W& I, U5 K! c7 G4 u. R. T
174.                     "Cater_Name":Cater_Name,
     
175.                     "Book_author":Book_author,
     . X' L* K+ x# D
176.                     "Book_Introduction":Book_Introduction,
     2 K9 z4 R! @: T8 e0 g: {; T9 ?8 [
177.                     "Book_Synopsis":Book_Synopsis,
       R6 c$ g! |0 q% x4 f
178.                     "Book_Palabras":Book_Palabras,
     
179.                     "img":img,
     
180.                 }
     
181.             self.add_BookFile(Bookname, BookIDs, img, Locaimg, Book_Dates)  #这行可注译，数据库发布接口，方便其他系统的发布
     ) {; L4 ]' T' b, A% y
182.         for each in response.doc('div[class="bookbtn-txt"]  a[class="catalogbtn"]').items():
     0 @1 e( l8 _% W
183.             self.crawl(each.attr.href, callback=self.index_page,save=Datos)
     ' \& l/ X! q" Q: H1 E! E
184.             
     
185.     @config(age=8 * 60 * 60)   
     
186.     def index_page(self, response):
     ) C1 ]) K3 j; y1 i) e7 S
187.         Datos = {
     
188.                   "Cater_Name":response.save['Cater_Name'],
     7 a8 K- A+ k4 [$ v
189.                    "Book_author":response.save['Book_author'],
     
190.                    "Book_Introduction":response.save['Book_Introduction'],
     2 |! m, _- l7 Z9 a
191.                    "Book_Synopsis":response.save['Book_Synopsis'],
     
192.                    "Book_Palabras":response.save['Book_Palabras'],
     # l6 B! j; R. `( b* S, `* j
193.                    "img":response.save['img'],
     ' D# n( _3 m+ D3 W' m0 w% t
194.                      }
     4 l* p- l/ o0 x, i5 v$ O% Y% u
195.         for each in response.doc('.chapter-list li:first-child a[href^="http"]').items():
     ; {4 y  S5 ^/ z  x8 j" D2 n
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)
     
199.     @catch_status_code_error
     
200.     def detail_page(self, response):        
     
201.         NewRe1 = u'哈书'
       o( _5 h0 A' P5 Q0 Q9 o
202.         NewRe2 = u'huhjsd.CC'
     
203.         NewRe3 = r'^\\n\\n'
     # P: F  U% O1 u5 C
204.         NewRe5 = u'小说网'
     - H( W/ c7 u3 {* @3 Y+ |: |- M4 ]
205.         NewRe6 = u'fgdfgf'
     
206.         NewRe7 = u'fgfgf'
     
207.         NewRe8 = u'ffhgf'
     ' J2 v5 K# h0 |9 ~) B7 F$ K3 q
208.         NewRe4 = r'[\f\t\v+\.\{\（\）\}\!\/_,$%^*(+"\')]+|[+——()?【】“”！，。？、~@#￥%……&*（）]+'
     ' n3 t3 w, ^# U, C' D7 ]
209.         ReC1 = u'静思'
     2 J2 G$ G2 L3 @7 R! F
210.         ReC2 = u'aghgf.com'
     / y% {1 |2 N- N1 M0 B5 k; M
211.         ReC3 = u'aghgfh.com'
     ) `# c# {- M7 U4 C, j  s2 W2 q4 k
212.         ReC4 = u''
     & Z3 ]6 e3 |0 j9 m
213.         ReC5 = u'文学网'
     
214.         ReC6 = r'<BR>'
     
215.         Bookname = response.doc('.readlocation a').eq(2).text()   #小说名称
     
216.         print Bookname
     ; K  W* ^5 A6 [* E* h
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']   #最近更新
     ; ]" A0 S1 ~3 Y* |4 T0 w
221.         Book_Palabras = response.save['Book_Palabras']   #小说字数
     7 t" c  l1 z( w* m6 |* j
222.         Bookurl = response.url   #小说网址
     
223.         Booktitle = response.doc('.article-title').text()   #章节名称
     : Y4 a) n6 g' A) Y& p) @
224.         BookID = response.doc('.readset-r span').text()   #小说ID
     3 b. u5 T' _" `7 d' {' o
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']   #小说状态(连载还是完成)
     5 p$ B8 L/ t0 B7 F
227.         Book_Date = str(datetime.datetime.now())    # 采集时间
     - k& I/ [+ `( n2 k# N
228.         BookConte2 = BookConte1.replace(NewRe1 , ReC1)
     
229.         BookConte3 = BookConte2.replace(NewRe2 , ReC2)
     # J) F# h3 a7 r7 F9 w4 I7 W" u! _
230.         BookConte5 = BookConte3.replace(NewRe5 , ReC5)
     
231.         BookConte6 = BookConte5.replace(NewRe6 , ReC2)
     ) t; J6 D) b! l6 u% t, G5 A, h
232.         BookConte7 = BookConte6.replace(NewRe7 , ReC2)
     
233.         BookConte8 = BookConte7.replace(NewRe3 , ReC6)
     
234.         BookConte4 = re.sub(NewRe4 , ReC4 , BookConte8)
     
235.         BookConte = BookConte4.replace("\n\n","<br>")
     2 m+ C; _- M0 ^4 j& {; l, B
236.         print BookConte
     & ]2 f( L2 r' o, O5 y0 ]9 J2 d
237.         Book_Introduction2 = Book_Introduction1.replace(NewRe1 , ReC1)
     7 W8 ]8 a$ N' {& F& y% O
238.         Book_Introduction3 = Book_Introduction2.replace(NewRe2 , ReC2)
     ' o0 m2 w/ \8 q) f/ x- R' O
239.         Book_Introduction4 = Book_Introduction3.replace(NewRe3 , ReC3)
     
240.         Book_Introduction = re.sub(NewRe4 , ReC4 , Book_Introduction4)
     
241.         Titleid = response.url.split(BookID + "/")[-1].split("/")[0]     
     * b& l+ }8 k3 P& |% X/ {: Q2 t
242.         Book_img = response.save['img'],  #小说图片
     + ]: H9 j0 x( ~3 N8 A4 D/ H
243.             
     ' e5 m/ L  t. A1 J2 ~
244.         #insert into MySQL 小说入库
     ( L+ h. u: A8 P& x) S( l! M; F8 @
245.         self.add_question(Bookname,Cater_Name,Book_author,Book_Introduction,Book_Synopsis,Book_Palabras,Bookurl,Booktitle,BookID,BookConte,Titleid,abover,Book_Date)   #这行可注译，数据库发布接口，方便其他系统的发布
     
246.         self.add_comment(Bookname, Booktitle, BookID, Titleid, Book_Date)   #这行可注译，数据库发布接口，方便其他系统的发布
     
247.         #post提交发布
     , i8 V- V$ I4 G) A0 d
248.         self.add_locoy(Bookname,Cater_Name,Book_author,Book_Introduction,Book_Palabras,Book_img,Booktitle,BookConte,abover)  #这行可注译，火车头发布接口，不需要可取消
     ! E1 ?/ D- s3 P% ~4 C9 n6 B# L
249.         Datos = {
     
250.                   "Cater_Name":response.save['Cater_Name'],
     - G6 S1 q1 [3 w: g
251.                    "Book_author":response.save['Book_author'],
     & _1 T5 E. W6 i# R) E
252.                    "Book_Introduction":response.save['Book_Introduction'],
     
253.                    "Book_Synopsis":response.save['Book_Synopsis'],
     ; \, V; e) d8 f! _1 y
254.                    "Book_Palabras":response.save['Book_Palabras'],
     0 \( A6 G9 h4 W+ j7 R; Q
255.                    "img":response.save['img'],
     
256.                      }
     & ~# g/ S# N. e1 v9 n3 n, V) u
257.         for each in response.doc('.articlebtn > a:nth-child(4)[href*="/xiaoshuo"]').items():
     
258.             self.crawl(each.attr.href, callback=self.detail_page,save=Datos)
     
259.         return {
     
260.             "Cater_Name":Cater_Name,
     6 b! V% n. u; E$ @/ b: B7 s
261.             "Bookname":Bookname,
     9 o3 x: n) I4 ?* k  S9 W( j  ?4 N" d& h
262.             "Book_author":Book_author,
     1 e3 \  H- G3 [$ a/ [. ~
263.             "Book_Introduction":Book_Introduction,
     . \8 l) P5 r! s+ d9 T' E
264.             "Book_Synopsis":Book_Synopsis,
       X7 S0 L# Y8 y# _7 I
265.             "Book_Palabras":Book_Palabras,
     
266.             "Book_img":Book_img,
     
267.             "Bookurl": response.url,
     " ^5 K3 H/ E8 d4 i$ M2 t/ s
268.             "Booktitle": Booktitle,
     9 u  B  m1 f5 e4 n& h+ R% z
269.             "BookID": BookID,
     . [& d! Y0 c% ^+ J' i$ H
270.             "BookConte": BookConte,
     7 u1 G6 G+ ?/ k7 G8 Q: M+ ~0 ?
271.             "Titleid": Titleid,
     0 s& _4 V- t& k* C5 Y
272.             "abover":abover,
     
273. #            "Book_Date" = str(datetime.datetime.now()),
     ; c5 \- m( ]2 l" u8 d% a2 t! o" U3 o
274.         }
     
275.     def download(self, P_dir, imgDir, file_name, Book_img):
     
276.         if not os.path.exists(imgDir):
     . ?$ u' e9 p5 k0 O
277.             os.makedirs(imgDir)
     
278.         file = imgDir + "/" + file_name
     0 T  ?; n, O; _. W0 h8 B) ~
279. #        print file
     
280.         f = open(file, 'wb+')
     5 k; u, }* J2 n7 G
281.         imag = requests.get(Book_img)
     
282.         f.write(imag.content)
     * x3 @2 e0 G$ C  q# |
283.         f.close()
     + U0 U* ~6 A1 V( ~" D
284.         #保存图片前
     7 \& e' ^( [; Q2 @2 @4 @% p
285.     def save_imgs(self,response):
     ! \7 p$ \+ p+ v- d) M
286.         content = response.content
     
287.         file_name = response.save["file_name"]
     : g0 e9 t- @2 ]6 d  k6 I% e* F
288.         imgDir = response.save["imgDir"]
     
289.         file_path = imgDir + file_name
     
290.         self.save_img(content,imgDir,file_path)
     4 X6 p  {5 E1 K! ]+ E+ Y% r/ _8 O
291.     #保存图片
     
292.     def save_img(self,content,imgDir,path):
     
293.         if not os.path.exists(imgDir):                        
     
294.             os.makedirs(imgDir)
     . W9 n! r: ]' M7 o
295.         f = open(path,"wb" )
     
296.         f.write(content)
     
297.         f.close()
     * [- d; u$ z" p, ^: V8 p# h! z
298.     #获取url后缀名
     
299.     def getExtension(self,url):                           
     : r- E" @1 k& x8 F/ t7 O
300.         extension = url.split(".")[-1]
     
301.         return extension
     
302.    
     8 u8 P2 O0 ]' a: P- q$ @
303.     #获取图片名
     * f( [" g5 i" R3 y
304.     def getname(self,url):
     ! N7 h! N* {. ]* f6 a: i- C5 Y
305.         name=url.split("/")[-1].split(".")[0]
     
306.         return name

复制代码

6 Q5 Q- [( {7 `7 S$ }9 Z