Puppeteer 自己封装的好用的函数

会不断更新!!!

依赖

  • puppeteer
  • expect-puppeteer
const expect = require('expect-puppeteer');
const puppeteer = require('puppeteer');

函数

1. 上传文件

filePath:从执行命令行的上下文开始算起

async function uploadfile(page, elementSelector, filePath) {
    const inputUploadHandle = await page.$(elementSelector);
    await inputUploadHandle.uploadFile(filePath);
}
2. 校验元素中包含指定字符
async function assertElementHaveText(page, elementSelector, assertText) {
    await page.waitFor(elementSelector, {visible: true});
    const element = await expect(page).toMatchElement(elementSelector, {timeout: 15000});
    await expect(element).toMatch(assertText);
}
3. 通过字符找到元素并点击

tag: 比如 'a', 'li'

async function clickElementByText(page, tag, text) {
    await page.waitFor(500);
    await expect(page).toMatch(text, {timeout: 5000});
    await expect(page).toClick(tag, {text: text});
    await page.waitFor(500);
}
4. 屏蔽指定的域名list

blockList: ['']

async function blockDomainsList(page, blockList){
    await page.setRequestInterception(true);

    page.on('request', interceptedRequest => {
        if (blockList.includes(new URL(interceptedRequest.url()).host)) {
            interceptedRequest.abort();
        } else {
            interceptedRequest.continue();
        }
    })
}
5. 清除输入框中的内容,并输入

puppeteer中好像没有自带清除的参数

async function clearAndType(page, elementSelector, text) {
    const ainput = await page.$(elementSelector);
    await ainput.click({ clickCount: 3 });
    await ainput.press('Backspace');
    await page.type(elementSelector, text);
}
6. 等待某个url请求返回
await page.waitForResponse(response => {
    // return里书写类似以下语句
    return response.request().url().includes('api/v1/my/url');
});
7. 校验当前页面url是否正确
async function assertCurrentUrl(page, url) {
    const currentUrl = await page.evaluate(() => location.href);
    if (currentUrl.indexOf(url) < 0) {
        throw new Error('url not in current url')
    }
}
8. 允许下载文件
await page._client.send('Page.setDownloadBehavior', {
        behavior: 'allow',
        // This path must match the WORKSPACE_DIR in Step 1
        downloadPath: path.join(process.cwd() , 'download')
});
9. 通过Xpath点击元素
async function clickByXpathText(page, tag, text){
    const xpath = `//${tag}[text()='${text}']`;
    await page.waitForXPath(xpath, {visible: true});
    await page.waitForTimeout(500);
    const elements = await page.$x(xpath);
    await elements[0].click();
}
10. 判断元素是否可见
async function elementIsHidden(page, elementSelector) {
    const isHidden = await page.$eval(elementSelector, (elem) => {
        return elem.style.display === 'none'
    });
    return isHidden;
}
11. 依次点击匹配的元素
async function clickEveryone(page, selector='', x_path=''){
    let elHandleArray = '';
    if (selector){
        await page.waitForSelector(selector);
        elHandleArray = await page.$$(selector);
    }
    if (x_path){
        await page.waitForXPath(x_path);
        elHandleArray = await page.$x(x_path);
    }
    if (!elHandleArray){
        throw new Error('没有找到要点击的元素');
    }
    elHandleArray.forEach(async el => {
        await el.click({delay:50}).catch(()=>{
            //console.log('try click every downside confirm')
        })
    });
}
12. 发送请求并获取返回结果
async function getToken(attr){
    return new Promise(resolve => {
        let obj='';
        const data = JSON.stringify({
            username: attr["username"],
            password: attr["password"]
        });
        const options = {
            hostname: attr["base_url"],
            port: 443,
            path: '/pa/api/auth/login',
            method: 'POST',
            headers: {
                'Content-Type': 'application/json',
                'Content-Length': data.length
            }
        };
        callback = function(response){
            var str='';
            response.on('data',function(chunk){
                str+=chunk;
            });
            response.on('end',function(){
                obj=JSON.parse(str);
                resolve(obj);
            });
        };

        let request = https.request(options,callback);
        request.write(data);
        request.end();
    });
}
13. 滚动屏幕
    await page.evaluate(_ => {
        window.scrollBy(0, 600);
    });
14. 滚动元素至可见
await page.evaluate((element_selector) => {
    document.querySelector(element_selector).scrollIntoView();
}, element_selector);
15. 控制新页面
    const newPagePromise = new Promise(res =>
        browser.once('targetcreated',
            target => res(target.page())
        )
    );
    // 执行打开新页面操作
    const page2 = await newPagePromise;
    await page2.bringToFront();
16. 获取元素属性值
async function getElementAttribute(page, selector, attribute) {
    return await page.$eval(selector, (el,attribute)=>
        el.getAttribute(attribute), attribute)
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,723评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,080评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,604评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,440评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,431评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,499评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,893评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,541评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,751评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,547评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,619评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,320评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,890评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,896评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,137评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,796评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,335评论 2 342

推荐阅读更多精彩内容