selenium的截图功能在chrome下无法实现,但是可以操作滚动条来一屏一屏的截图,然后再合并成一张图,合并图片的代码在网上找的,十分感谢那位朋友,具体解决方案如下:直接上代码:
1 def capture(base_url, pix_w, pix_h, filename): 2 """chrome截屏 3 base_url- 要截屏的url 4 pix_w- 窗口宽 5 pix_h- 窗口高 6 filename-生成截图的文件名 7 """ 8 try: 9 driver = webdriver.Chrome() 10 driver.set_window_size(pix_w, pix_h+89) 11 driver.get(base_url) 12 time.sleep(2) 13 img_list = [] 14 i = 0 15 while i<20: 16 #滚屏 17 js="var q=document.body.scrollTop="+ str(i*pix_h)+";" 18 driver.execute_script(js) 19 js1 = "return document.body.scrollHeight.toString()+','+document.body.scrollTop.toString()" 20 js1_result = driver.execute_script(js1) 21 real_scroll_h, real_top = js1_result.split(',')[0], js1_result.split(',')[1] 22 #real_scroll_h, real_top 是当前滚动条长度和当前滚动条的top,作为是否继续执行的依据,由于存在滚动条向下拉动后会加载新内容的情况,所以需要以下的判断 23 #如果这次设置的top成功,则继续滚屏 24 if real_top == str(i*pix_h): 25 i += 1 26 driver.save_screenshot('static/capture/wap/wap-'+filename + str(i)+'.png') 27 img_list.append('static/capture/wap/wap-'+filename + str(i)+'.png') 28 last_t = real_top 29 else: 30 #如果本次设置失败,看这次的top和上一次记录的top值是否相等,相等则说明没有新加载内容,且已到页面底,跳出循环 31 if real_top != last_t: 32 last_t = real_top 33 else: 34 driver.save_screenshot(filename + str(i+1)+'.png') 35 img_list.append(filename + str(i+1)+'.png') 36 break 37 image_merge(img_list, "", filename+'.png') 38 except Exception,e: 39 print e 40 def image_merge(images, output_dir, output_name='merge.jpg', restriction_max_width=None, restriction_max_height=None): 41 """垂直合并多张图片 42 images - 要合并的图片路径列表 43 ouput_dir - 输出路径 44 output_name - 输出文件名 45 restriction_max_width - 限制合并后的图片最大宽度,如果超过将等比缩小 46 restriction_max_height - 限制合并后的图片最大高度,如果超过将等比缩小 47 """ 48 def image_resize(img, size=(1500, 1100)): 49 """调整图片大小 50 """ 51 try: 52 if img.mode not in ('L', 'RGB'): 53 img = img.convert('RGB') 54 img = img.resize(size) 55 except Exception, e: 56 pass 57 return img 58 max_width = 0 59 total_height = 0 60 # 计算合成后图片的宽度(以最宽的为准)和高度 61 for img_path in images: 62 if os.path.exists(img_path): 63 img = Image.open(img_path) 64 width, height = img.size 65 if width > max_ 66 max_width = width 67 total_height += height 68 69 # 产生一张空白图 70 new_img = Image.new('RGB', (max_width, total_height), 255) 71 # 合并 72 x = y = 0 73 for img_path in images: 74 if os.path.exists(img_path): 75 img = Image.open(img_path) 76 width, height = img.size 77 new_img.paste(img, (x, y)) 78 y += height 79 80 if restriction_max_width and max_width >= restriction_max_ 81 # 如果宽带超过限制 82 # 等比例缩小 83 ratio = restriction_max_height / float(max_width) 84 max_width = restriction_max_width 85 total_height = int(total_height * ratio) 86 new_img = image_resize(new_img, size=(max_width, total_height)) 87 88 if restriction_max_height and total_height >= restriction_max_height: 89 # 如果高度超过限制 90 # 等比例缩小 91 ratio = restriction_max_height / float(total_height) 92 max_width = int(max_width * ratio) 93 total_height = restriction_max_height 94 new_img = image_resize(new_img, size=(max_width, total_height)) 95 96 if not os.path.exists(output_dir): 97 os.makedirs(output_dir) 98 save_path = '%s/%s' % (output_dir, output_name) 99 new_img.save(save_path) 100 for img_path in images: 101 os.remove(img_path) 102 return save_path
大致就这么个思路