分享一款懒人必备的Python爬虫神器

前言

今天把最近的一个应用做好了,测试了一下运行没有问题,剩下的就是检验一下结果如何.从光谱到Lab值通常使用matlab中的roo2lab(),不过经过我最近的测试发现转换的结果并不理想,而且这个转化的代码也不是我写的所以另寻他法,找到了下面这个网页。

动手

有了这个网页,很简单就想到去解析.然后很快找到了这个api,可以看到用post提交表单请求就可以返回结果

于是一番操作,修改form表单,构造传入的spectrum,但是最终请求得到的却是一个页面,并不是想要的json.然后许久没写爬虫的我直接恼火,想到自动化工具模拟操作.可是selenium很难用而且还得去找浏览器新版本的驱动,随后直接搜索一番,发现了这个神器—playwright。首先老规矩去它的首页看看教程

接着安装一下playwright以及浏览器的驱动


pip install pytest-playwright
playwright install

第二步的时候很慢,所以我只等他下载好了chrome和ffmpeg直接就ctrl+c停止了,毕竟我也用不着其他的浏览器驱动。然后稍微浏览一下这个文档功能非常丰富,不过我用到的功能也不多,接下来的使用才是真正适合我们懒人的.

记住一行代码

python -m playwright codegen xxx.com

其中xxx.com就是我们的目标网址,运行后会创建一个熟悉的自动化页面,然后我们就进行一些我们想要的操作,比如设置开始的最小波长为400nm,然后观察以及光源改为D65/10.一系列操作后会看到对应的代码已经生成好了

至今为止我还没有写一行代码,不过构造输入的光谱还是得自己来的把这一段代码复制下来,然后自己写构造输入的函数(就几行),然后通过选择器(直接左键元素复制xpath)得到lab值,这样目标就搞定了

整体代码如下


import time
from playwright.sync_api import Playwright, sync_playwright, expect
import numpy as np

data_test=np.loadtxt('./dist/1_res.csv',delimiter=',')

def get_str(arr):
    arr_str=""
    for i in arr:
        arr_str+=str(format(i,".2f"))+"\r\n"
    return arr_str

labs=[]

def run(playwright: Playwright) -> None:
    browser = playwright.chromium.launch(headless=False)
    context = browser.new_context()

    # Open new page
    page = context.new_page()

    # Go to https://www.qtccolor.com/secaiku/tool/spectrum
    page.goto("https://www.qtccolor.com/secaiku/tool/spectrum")

    # Click div[role="tab"]:has-text("光谱数据")
    page.locator("div[role=\"tab\"]:has-text(\"光谱数据\")").click()

    # Click text=最小波长：nm
    page.locator("text=最小波长：nm").click()

    # Fill [placeholder="\33 80"]
    page.locator("[placeholder=\"\\33 80\"]").fill("400")

    # Select 1964
    page.locator("select[name=\"obs\"]").select_option("1964")

    # Select D65
    page.locator("select[name=\"ill\"]").select_option("D65")
    # Fill textarea[name="spectrum"]
    for i in range(len(data_test)):
        inputs=get_str(data_test[i,:])
        # Click textarea[name="spectrum"]
        page.locator("textarea[name=\"spectrum\"]").click()
        page.locator("textarea[name=\"spectrum\"]").press("Control+a")
        page.locator("textarea[name=\"spectrum\"]").fill(inputs)

        # Click button:has-text("转换颜色")
        page.locator("button:has-text(\"转换颜色\")").click()
        time.sleep(1)
        # Click text=Lab0.000.000.00 >> td >> nth=1
        L=float(page.locator('xpath=//*[@id="scroll_container"]/div[1]/div/div[2]/table/tbody/tr[2]/td[2]').inner_text())
        # Click text=Lab0.000.000.00 >> td >> nth=2
        a=float(page.locator('xpath=//*[@id="scroll_container"]/div[1]/div/div[2]/table/tbody/tr[2]/td[3]').inner_text())
        # Click text=Lab0.000.000.00 >> td >> nth=3
        b=float(page.locator('xpath=//*[@id="scroll_container"]/div[1]/div/div[2]/table/tbody/tr[2]/td[4]').inner_text())
        print(L,a,b)
        labs.append([L,a,b])
    # ---------------------
    context.close()
    browser.close()

with sync_playwright() as playwright:
    run(playwright)
    np.savetxt('./1_lab_res.csv',labs,delimiter=",")

可以说从安装到实现就几分钟,而且特别容易上手,我第一次用也一下就能实现效果

剩下的就是简单的写个函数计算色差啥的就没难度了.

最后

大厂出品果然不同,使用它在不考虑运行效率(有异步但是我懒得看了)的情况下可以轻松实现复杂操作,懒人最爱!

来源：https://juejin.cn/post/7140542063061237773（侵删）