scrapy抓取拉勾网职位信息(五)——代码优化

时间:2023-11-11 21:32:56

上一篇我们已经让代码跑起来,各个字段也能在控制台输出,但是以item类字典的形式写的代码过于冗长,且有些字段出现的结果不统一,比如发布日期。

而且后续要把数据存到数据库,目前的字段基本都是string类型,会导致占用空间较多,查询时速度会较慢,所以本篇先对目前已写好的代码进行适当优化。


本篇目的:使用item loader以及processor对代码进行优化,对字段数据进行清洗

1、修改一下items.py文件的字段

我们对工资和工作经验字段进行分割让其更适合数据库存储:

import scrapy

class LagouItem(scrapy.Item):
url = scrapy.Field()
name = scrapy.Field()
ssalary = scrapy.Field() #最低工资
esalary = scrapy.Field() #最高工资
location = scrapy.Field()
syear = scrapy.Field() #最低工作经验
eyear = scrapy.Field() #最高工作经验
edu_background = scrapy.Field()
type = scrapy.Field()
tags = scrapy.Field()
release_time = scrapy.Field()
advantage = scrapy.Field()
job_desc = scrapy.Field()
work_addr = scrapy.Field()
company = scrapy.Field()

小贴士:我们要使用item loaders,需要对其有一个基本的了解。item loaders可以看成是对item对象的一个封装,也就是说它同样可以对字段进行操作,另外对于每一个字段,都有一个input processor和一个output processor,也就是输入处理器,和输出处理器,一般情况下,output processor即输出处理器使用的更多,它可以对我们使用css或者xpath选择出的结果进行进一步的处理。


2、编写lagou_c.py文件

导入LagouCLoader以及修改parse_item()函数

   from lagou.loaders import LagouCLoader  #这个loaders.py文件是新建立的文件,它包括一个LagouCLoader类使用处理器对item进行处理

   def parse_item(self,response):
loader = LagouCLoader(item=LagouItem(),response=response)
loader.add_value('url',response.url)
loader.add_css('name','.name::text')
loader.add_css('ssalary','.salary::text')
loader.add_css('esalary','.salary::text')
loader.add_xpath('location','//*[@class="job_request"]//span[2]/text()')
loader.add_xpath('syear','//*[@class="job_request"]//span[3]/text()')
loader.add_xpath('eyear','//*[@class="job_request"]//span[3]/text()')
loader.add_xpath('edu_background','//*[@class="job_request"]//span[4]/text()')
loader.add_xpath('type','//*[@class="job_request"]//span[5]/text()')
loader.add_css('tags','.labels::text')
loader.add_css('release_time','.publish_time::text')
loader.add_css('advantage','.job-advantage p::text')
loader.add_css('job_desc','.job_bt p::text')
loader.add_xpath('work_addr','//div[@class="work_addr"]/a[1]/text()') #这个work_addr分成了三次进行填充,最终需要把得到的数据进行合并
loader.add_xpath('work_addr','//div[@class="work_addr"]/a[2]/text()')
addr_value = response.css('.work_addr::text').extract()[-2].strip()
loader.add_value('work_addr',addr_value)
loader.add_css('company','.job_company img::attr(alt)')
yield loader.load_item()

解释:可以看到我们先生成了一个LagouCLoader实例对象,这个类使用了两个参数,一个是LagouItem对象,还有一个是response对象,而下方调用了一系列的方法对字段进行填充,add_xpath,add_css,add_value就是字面上的意思,对response使用xpath,css或者直接用一个value值进行填充。这和我们之前写的parse_item()比较类似,不同的是之前的每一个字段我们都直接进行了处理。最后使用yield loader.load_item()方法将字段分配出去。


强行解释一波:

上面的解释可能大家看的还是有点懵,我就用现实生活中的例子来说吧。现在我要做3个菜,青椒肉丝,西红柿炒鸡蛋,酸辣土豆丝。

items (空字段):可以看成这三个菜,但是最开始是没有的,需要加工出来

input processor:可以看成是前置处理,洗菜,切菜,准备配料等。

add_xpath,add_css,add_value:开始炒菜咯,放点油,放点盐,多加点辣椒。。。

output processor: 可以看成是后置处理,菜做好了, 撒点葱,香油

items(成品):现在可以上菜了,yield laoder.load_item()


3、编写loaders.py文件

细心的童鞋可能发现了,我们在lagou_c.py文件中只是把字段提取了出来,但是还没有做处理,所以我们在items.py文件的同级目录下建立一个loaders.py文件

  • 首先导入loaders以及几个自带的processor处理器,Join,Compose,TakeFirst,MapCompose
 from scrapy.loader.processors import Join, Compose, TakeFirst, MapCompose
from scrapy.loader import ItemLoader

Join:和python原生库中的Join用法类似,用来做拼接

Compose:参数传入多个函数对象,这几个函数会依次调用

MapCompose:和Compose类似,不同的是它可以对一个列表中的每个值分别调用函数

TakeFirst:和extract_first()类似,不过,TakeFirst是产生第一个非空的值

  • 建立一个LagouCLoader类,它继承自ItemLoader,字段的处理在这个类下面进行
class LagouCLoader(ItemLoader):
default_output_processor = TakeFirst() #设置默认的输出处理器为TakeFirst,也就是说,所有的字段信息提取出来后,会自动取第一个值,类似extract_first()

但是我们目前还有几个字段不能这样处理

location、edu_backgroud:提取出的字段需要去掉/和多余的空格

ssalary、esalary:最低工资和最高工资在同一个字段里,需要分别拿出来

syear、eyear:最低工作年限和最高工作年限需要进一步处理,提取出的字段有类似几种情况,a:如3-5年,b.如三年以上 c.如经验不限

release_time:发布时间同样分类似几种情况,a.如 2天前 b.如13:20c.如2018-11-11

tags、advantage、job_desc:需要使用Join方法进行拼接


  • 我这里直接给出代码
from scrapy.loader.processors import Join, Compose, TakeFirst, MapCompose
from scrapy.loader import ItemLoader
import re
import datetime def change_value(value):
return value.replace('/', '').strip() #把字段的’/‘更换为空,然后把首位多余的空格去掉 def time_trans(value):
if '天前' in value: #针对字段中有几天前的情况
num = re.search('(\d+).*', value).group(1)
time_pub = datetime.datetime.now() - datetime.timedelta(days=int(num)) #获取num天前的日期
return time_pub.strftime('%Y-%m-%d')
if '-' in value: #针对字段中有-的情况,也就是有具体发布日期的情况
time_pub = re.search('(\d+-\d+-\d+).*', value).group(1)
return time_pub
else:
return datetime.datetime.now().strftime('%Y-%m-%d') #把当前日期转为string格式 def exp_trans(value):
if '-' in value: #针对有具体工作经验年限要求的
result = re.findall('(\d+).*-(\d+).*', value)
elif '以上' in value: #针对含多少年以上工作经验的情况
result = re.search('(\d+).*', value)
else:
result = [(0, 0)] #和另外两种情况保持一致格式
return result #返回的结果是一个列表(内含元组),元组内第一个是最低工作年限,第二个值是最高工作年限 class LagouCLoader(ItemLoader):
default_output_processor = TakeFirst() #将所有的字段都进行后处理,只取第一个非空值,但是要注意这个优先级是低于_in,_out的
location_out = Compose(TakeFirst(), change_value) #字段名+_out代表的就是对输出后的处理
edu_background_out = Compose(TakeFirst(), change_value)
ssalary_out = Compose(TakeFirst(), lambda x: re.search('(\d+).*-(\d+).*', x).group(1))
esalary_out = Compose(TakeFirst(), lambda x: re.search('(\d+).*-(\d+).*', x).group(2))
syear_out = Compose(TakeFirst(), exp_trans, lambda x: x[0][0])
eyear_out = Compose(TakeFirst(), exp_trans, lambda x: x[0][1])
release_time_out = Compose(TakeFirst(), time_trans)
tags_out = Compose(Join(','))
advantage_out = Compose(Join(','))
job_desc_out = Compose(Join('\n'))
work_addr_out = Compose(Join(''))

小贴士:

  • default_output_processor = TakeFirst(),设置所有的默认输出处理器都使用TakeFirst(),这样对于所有的字段全都只会取第一个非空值,针对数据本身就很整齐,不需要额外进行清洗的,这种操作收益最大。
  • 如果需要对某个字段使用输入处理器,可以使用字段名+_in = processor()来对该字段进行输入的修饰处理,注意这里的processor是指Compose、TakeFirst等。比如字段名是name,name_in = Compose(lambda x:x.strip())就可以对输入进行处理了,处理的结果就是输入的值前后去除了多余的空格。同理,如果要对某个字段使用输出处理器,那就使用字段名+_out来进行操作。
  • 优先级说明:类似_in,_out这种操作,优先级是高于default_output_processor(或者default_input_processor)优先级的,所以如果对某个字段采取了_in,_out操作,这个操作就会把default_output_processor的操作给覆盖掉。

这样我们就完成了前期的代码优化,运行下程序看下结果,可以看到符合我们的预期要求。

scrapy抓取拉勾网职位信息(五)——代码优化