scrapy操作MySQL数据库

scrapy爬虫框架简单Demo

转载自：http://blog.csdn.net/u013082989/article/details/52589791

github地址：https://github.com/lawlite19/PythonCrawler-Scrapy-Mysql-File-Template

使用scrapy爬虫框架将数据保存MySQL数据库和文件中

settings.py

• 修改Mysql的配置信息

#Mysql数据库的配置信息
MYSQL_HOST = '127.0.0.1'
MYSQL_DBNAME = 'testdb'         #数据库名字，请修改
MYSQL_USER = 'root'             #数据库账号，请修改 
MYSQL_PASSWD = '123456'         #数据库密码，请修改

MYSQL_PORT = 3306               #数据库端口，在dbhelper中使用
  
  

  
  

   
   
  
  
  
  

• 指定pipelines

ITEM_PIPELINES = {
    'webCrawler_scrapy.pipelines.WebcrawlerScrapyPipeline': 300,#保存到mysql数据库
    'webCrawler_scrapy.pipelines.JsonWithEncodingPipeline': 300,#保存到文件中
}
  
  

  
  

   
   
  
  
  
  

items.py

• 声明需要格式化处理的字段

class WebcrawlerScrapyItem(scrapy.Item):
    '''定义需要格式化的内容（或是需要保存到数据库的字段）'''
    # define the fields for your item here like:
    # name = scrapy.Field()
    name = scrapy.Field()   #修改你所需要的字段
    url = scrapy.Field()
  
  

  
  

   
   
  
  
  
  

pipelines.py

一、保存到数据库的类WebcrawlerScrapyPipeline（在settings中声明）

• 定义一个类方法from_settings，得到settings中的Mysql数据库配置信息，得到数据库连接池dbpool

    @classmethod
    def from_settings(cls,settings):
        '''1、@classmethod声明一个类方法，而对于平常我们见到的则叫做实例方法。 
           2、类方法的第一个参数cls（class的缩写，指这个类本身），而实例方法的第一个参数是self，表示该类的一个实例
           3、可以通过类来调用，就像C.f()，相当于java中的静态方法'''
        dbparams=dict(
            host=settings['MYSQL_HOST'],#读取settings中的配置
            db=settings['MYSQL_DBNAME'],
            user=settings['MYSQL_USER'],
            passwd=settings['MYSQL_PASSWD'],
            charset='utf8',#编码要加上，否则可能出现中文乱码问题
            cursorclass=MySQLdb.cursors.DictCursor,
            use_unicode=False,
        )
        dbpool=adbapi.ConnectionPool('MySQLdb',**dbparams)#**表示将字典扩展为关键字参数,相当于host=xxx,db=yyy....
        return cls(dbpool)#相当于dbpool付给了这个类，self中可以得到
  
  

  
  

   
   
  
  
  
  

• __init__中会得到连接池dbpool

    def __init__(self,dbpool):
        self.dbpool=dbpool
  
  
1
2
  
  

   
   
  
  
  
  
1
2

• process_item方法是pipeline默认调用的，进行数据库操作

    #pipeline默认调用
    def process_item(self, item, spider):
        query=self.dbpool.runInteraction(self._conditional_insert,item)#调用插入的方法
        query.addErrback(self._handle_error,item,spider)#调用异常处理方法
        return item
  
  

  
  

   
   
  
  
  
  

• 插入数据库方法_conditional_insert

    #写入数据库中
    def _conditional_insert(self,tx,item):
        #print item['name']
        sql="insert into testpictures(name,url) values(%s,%s)"
        params=(item["name"],item["url"])
        tx.execute(sql,params)
  
  

  
  

   
   
  
  
  
  

• 错误处理方法_handle_error

    #错误处理方法
    def _handle_error(self, failue, item, spider):
        print failue
  
  

  
  

   
   
  
  
  
  

二、保存到文件中的类JsonWithEncodingPipeline（在settings中声明）

• 保存为json格式的文件，比较简单，代码如下

class JsonWithEncodingPipeline(object):
    '''保存到文件中对应的class
       1、在settings.py文件中配置
       2、在自己实现的爬虫类中yield item,会自动执行'''    
    def __init__(self):
        self.file = codecs.open('info.json', 'w', encoding='utf-8')#保存为json文件
    def process_item(self, item, spider):
        line = json.dumps(dict(item)) + "\n"#转为json的
        self.file.write(line)#写入文件中
        return item
    def spider_closed(self, spider):#爬虫结束时关闭文件
        self.file.close()
  
  

  
  

   
   
  
  
  
  

dbhelper.py

• 自己实现的操作Mysql数据库的类
• init方法，获取settings配置文件中的信息

    def __init__(self):
        self.settings=get_project_settings() #获取settings配置，设置需要的信息

        self.host=self.settings['MYSQL_HOST']
        self.port=self.settings['MYSQL_PORT']
        self.user=self.settings['MYSQL_USER']
        self.passwd=self.settings['MYSQL_PASSWD']
        self.db=self.settings['MYSQL_DBNAME']
  
  

  
  

   
   
  
  
  
  

• 连接到Mysql

    #连接到mysql，不是连接到具体的数据库
    def connectMysql(self):
        conn=MySQLdb.connect(host=self.host,
                             port=self.port,
                             user=self.user,
                             passwd=self.passwd,
                             #db=self.db,不指定数据库名
                             charset='utf8') #要指定编码，否则中文可能乱码
        return conn
  
  

  
  

   
   
  
  
  
  

• 连接到settings配置文件中的数据库名（MYSQL_DBNAME）

    #连接到具体的数据库（settings中设置的MYSQL_DBNAME）
    def connectDatabase(self):
        conn=MySQLdb.connect(host=self.host,
                             port=self.port,
                             user=self.user,
                             passwd=self.passwd,
                             db=self.db,
                             charset='utf8') #要指定编码，否则中文可能乱码
        return conn 
  
  

  
  

   
   
  
  
  
  

• 创建数据库（settings文件中配置的数据库名）

    #创建数据库
    def createDatabase(self):
        '''因为创建数据库直接修改settings中的配置MYSQL_DBNAME即可，所以就不要传sql语句了'''
        conn=self.connectMysql()#连接数据库

        sql="create database if not exists "+self.db
        cur=conn.cursor()
        cur.execute(sql)#执行sql语句
        cur.close()
        conn.close()
  
  

  
  

   
   
  
  
  
  

• 还有一些数据库操作方法传入sql语句和参数即可（具体看代码）

实现具体的爬虫.py（即模板中的pictureSpider_demo.py文件）

• 继承scrapy.spiders.Spider 类
• 声明三个属性

    name="webCrawler_scrapy"    #定义爬虫名，要和settings中的BOT_NAME属性对应的值一致

    allowed_domains=["desk.zol.com.cn"] #搜索的域名范围，也就是爬虫的约束区域，规定爬虫只爬取这个域名下的网页

    start_urls=["http://desk.zol.com.cn/fengjing/1920x1080/1.html"]   #开始爬取的地址
  
  

  
  

   
   
  
  
  
  

• 实现parse方法，该函数名不能改变，因为Scrapy源码中默认callback函数的函数名就是parse

    def parse(self, response):
  
  

  
  

   
   
  
  
  
  

• 返回item

    item=WebcrawlerScrapyItem()  #实例item（具体定义的item类）,将要保存的值放到事先声明的item属性中
    item['name']=file_name 
    item['url']=realUrl
    print item["name"],item["url"]    

    yield item  #返回item,这时会自定解析item
  
  

  
  

   
   
  
  
  
  

测试

• 测试DBHelper
  创建testdb数据库和testtable表

  

• 测试爬虫
  scrapy crawl webCrawler_scrapy运行爬虫后会将爬取得图片保存到本地，并且将name和url保存到数据库中