pyspider脚本编写指南

注意,虽然在本文中会涉及调度策略等内容,但实际执行效果取决于具体策略实现。

project

脚本分为不同的 project,不同的 project 之间的任务互相独立,建议为不同的站点建立不同的 project

project 属性
  • group – 分组标识,暂时仅用于区分

  • name – 脚本的名字,只允许 a-zA-Z0-9_

  • status – 任务状态,只有处于 DEBUG 和 RUNNING 状态任务才会被执行

    • TODO – 新建立的脚本所处的状态

    • STOP – 停止

    • CHECKING – 当任务在 DEBUG 或 RUNNING 状态被修改时,会自动变为 CHECKING,暂停任务执行

    • DEBUG – 任务可以开始执行,handler的 __env__['debug'] == True

    • RUNNING – 任务正常执行状态

  • rate – 每秒抓取页面个数

  • burst – 并发数

task

任务是 spider 调度的最小单位。

  • taskid 是任务的唯一区分标识,默认使用 url 的 md5 作为 taskid(即根据 url 对抓取去重)

  • 不同 project 间任务相互独立(不同 project 间 taskid 可以相同)

  • 任务状态

    • active(1) – 活动状态,表示任务在队列中等待被抓取(包括在队列中、抓取中、执行时间未到、重试中)

    • success(2) – 抓取成功状态

    • failed(3) – 经过重试后抓取失败

    • bad(4) – 损坏,暂未使用

任务调度

只有处于 active 状态的任务才会得到调度,scheduler 对于每个 project 有两个队列:

  • 时间队列 – 任务尚未到达执行时间,时间拉长重试也在这个队列中

  • 优先队列 – 任务已经到达执行时间,任务会根据优先级 priority在有流量配额情况下,依次发起调度

scheduler 会等待每一个调度任务返回,当执行失败时,会根据 retries 参数,拉长一定时间发起重试。当超时结果未返回时,会无限次地重新发起调度

新/旧任务请求
  • 当产生一个库中没有的任务请求时,会根据参数 exetime 中设定的执行时间决定是放入 时间队列 还是 优先队列。

  • 当新请求已经在队列中(即任务状态是 active 时),会忽略新的请求,即使它们的参数不一致。

    可以通过 force_update参数强制更新处于 active 状态的任务

  • 当任务状态是 success 或 failed 时。参数 age 会检测 上次抓取时间 + age 是否大于当前时间,若大于,则会重启任务。或者,您可以通过设置 itag (例如前链中提取的更新时间)参数,当本次请求的 itag 与上次不同时,会重启任务。

script

pyspider的脚本都是python环境的,没有做任何隔离,可以调用 host 机器环境中的任何 lib 和资源。

如果您担心安全问题,请使用 linux 的用户隔离机制隔离 processor

示例脚本:http://demo.pyspider.org/

执行环境
环境变量
  • __project__ == self.project 当前任务信息

  • self.project_name 当前脚本名

  • self.response 当前请求的response

  • self.task 当前请求的原始task信息

  • loggingloggerprint的信息会被采集,请使用 logger 打印日志(默认 logging 已经 import 到环境中,请不要再次导入)

  • 可以通过 projects 伪模块将其他 project 导入成 module,例如 from projects import some_project

脚本约束
  • 接口请从 from libs.base_handler import * 中导入

  • 脚本必须有至少一个的继承自 BaseHandler 的类

    例外:当环境中有多个继承自 BaseHandler (例如:二次派生时),请指定 __class__ = Handler

  • 每个 task 由参数 callback 指定的成员函数解析(默认的 callback__call__

    完整的 callback 函数原型是:def foo(self, response, task) 您可以忽略您不需要的参数(但需要顺序一致)例如:foo(self, res)

    • response 是一个抓取结果的对象,完整 api 见下文

    • task 是原始的请求 json,和您在 调试器 左侧 绿色区域 看到的结果一致

  • 默认 status_code 不为 200 的请求不会发给回调函数解析,请用 @catch_status_code_error 修饰回调函数,以捕获抓取异常。

  • on_start 是脚本执行的入口,当您在主页/上点击 Run 时,会触发 on_start 执行

  • 函数的返回内容会被 on_result(self, result) 函数捕获,您可以在这里将其输出到数据库中

API接口

BaseHandler

source: libs/base_handler.py

特殊接口

def on_start(self):
当在首页上点击 project 的 run 按钮时被调用

def on_result(self, result):
所有 callback 函数 return 时被调用(包括返回值为 None 时)。默认的 on_result 回调会将结果发送给 result_work,默认的 result_work 会将结果储存到 resultdb 中。如果覆盖这个方法,会阻止默认行为。使用super(Handler, self).on_result(result) 调用父类方法以将结果发送给 result_worker。

def on_message(self, project, message):
当收到其他 project 发送的消息时被调用,project 为发送消息的 project 的名字,message 可能为任意类型。使用 send_message发送消息。

module方法

@config(**kwargs)
修饰回调函数,给 self.crawl 接口设定 function 级默认值

@every(minutes, seconds)
修饰 on_cronjob 回调函数,每 minutes 分钟执行

@catch_status_code_error
修饰回调函数,回调函数接受抓取异常数据。默认的 callback 会直接对非200返回直接抛出异常。

@not_send_status
修饰回调函数,回调函数不发送状态包给 scheduler,不建议使用.

类成员变量

crawl_config:dictself.crawl 接口设定 class 级默认值

方法

BaseHandler.crawl(url, **kwargs)
参数列表,以下参数,除了 url 都是可选的。您可以使用 crawl_config 和/或 @config(_config:dict) 给类或方法设定默认值。但 直接调用的优先级 高于 方法级默认值 高于 类默认值。修改默认值不会影响已经创建了的任务。

基本
  • url – 需要抓取的url,支持url数组,支持dataurl:data:,content > 当有params参数存在时,最终url为拼接后的
    > 当url中有中文时,会进行 % 编码

  • taskid – 默认取 url 的 md5

  • callback – 回调函数,可以是 self.foo 也可以是函数的名字 "foo",默认 __call__

  • save – 传递给下一个请求的数据,可以是任何可 json 的对象

  • force_update – 强制修改队列中的任务信息(见上文)

调度
  • priority – 调度优先级,越高越好

  • retries – 重试次数

  • exetime – 执行时间,为 time.time() 的 timestamp

  • age – 页面有效期,在有效期内抓取过的页面不会重新抓取

  • itag – 能反映页面最后更新时间的标记,一般取自最后更新时间

抓取
  • method – POST/GET/HEAD 等

  • params – url中 ?a=b 的部分,类型 dict

  • data – POST正文的内容,string 或 dict > GET有data可能会导致抓取失败

  • files – 上传文件,{'key': ('file.name': 'content')}

  • headers – 请球头,类型dict

  • cookies – 请求的 Cookies,类型 dict

  • timeout – 请求超时

  • allow_redirects – 是否跟随30x跳转,默认 True

  • proxy – 是否使用代理抓取,默认 True(需要 fetcher 支持)

  • etag – 是否启用etag(当页面无变化时不会抓取内容,也不会触发解析)

  • last_modifed – 是否启用last_modifed(当页面无变化时不会抓取内容,也不会触发解析)

使用JS渲染
  • fetch_type – 设置为 phantomjs或js时,调用 phantomjs 进行js渲染,加载 ajax 内容。需要安装 phantomjs,并为 fetcher 设置 phantomjs_proxy。

  • js_script – 设置一段 JS 脚本,在页面加载完成(前)时执行。进行模拟点击等动作。

  • js_run_at – document-start/document-end

  • load_images – 开启图片,默认关闭。

BaseHandler.send_message(project, msg, [url])
向其他 project 发送 msg,对方脚本可以通过 on_message 回调获得消息。默认的 url 是固定的,在 on_message 中发往on_result 的数据会相互覆盖,使用 url 参数为每个结果设置不同的 URL地址(例如:http://example.com/page-3.html#result-1)

Response

source: libs/response.py

属性

Response.status_code
请求返回的状态码

Response.url
请求的url(当发生跳转时,为跳转后的url)

Response.orig_url
原请求的url

Response.headers
返回头

Response.cookies
返回的cookies

Response.error
请求错误信息,string类型

Response.time
请求用时

Response.ok
请求是否成功

Response.encoding
页面的编码,会根据header,content自动检测;当检测错误时,可以手动设定编码覆盖

Response.text
页面经过转码成为 unicode 的数据

Response.content
页面未转码内容

Response.save
从上一个任务中传下来的数据

Response.json
尝试解析 json

Response.doc
将页面建树,返回一个 pyquery 对象(已将相对地址转为绝对地址 make_links_absolute

方法

Response.raise_for_status()
尝试抛出抓取异常

FAQ

如何删除一个任务

将任务的 group 设置为 delete ,并将状态设置为 STOP,没有更新 24小时 之后,scheduler 会自动删除任务。

来源: <http://www.jishubu.net/yunwei/python/411.html>

posted @ 2015-05-23 16:55  silianpan  阅读(1861)  评论(0编辑  收藏  举报