39 lines
1.3 KiB
JavaScript
39 lines
1.3 KiB
JavaScript
import fs from 'fs';
|
|
|
|
/**
|
|
* 设置页面的LocalStorage
|
|
* @param {*} page
|
|
* @param {String} file 用户信息
|
|
*/
|
|
export async function setLocalStorage(page, file) {
|
|
const outputLocalStorge = fs.readFileSync(file, 'utf-8');
|
|
let localStorageTest = JSON.parse(outputLocalStorge);
|
|
|
|
// 设置页面的localStorage
|
|
await page.evaluateOnNewDocument(localStorageTest => {
|
|
localStorage.clear();
|
|
localStorage.setItem('merchant_access_token', localStorageTest.merchant_access_token);
|
|
localStorage.setItem('merchant_refresh_token', localStorageTest.merchant_refresh_token);
|
|
localStorage.setItem('merchant_login_data', localStorageTest.merchant_login_data);
|
|
localStorage.setItem('merchant_metadata_data', localStorageTest.merchant_metadata_data);
|
|
}, localStorageTest);
|
|
|
|
console.log('set success');
|
|
}
|
|
|
|
/**
|
|
* @param {Page} page
|
|
* @param {String} webUrl 网站
|
|
* @param {String} cookieFile cookie地址Url
|
|
*/
|
|
export async function injectionCookie(page, webUrl, cookieFile) {
|
|
try {
|
|
await page.goto(webUrl);
|
|
const cookies = JSON.parse(fs.readFileSync(cookieFile, 'utf-8'));
|
|
console.log(cookies);
|
|
await page.setCookie(...cookies);
|
|
await page.goto(webUrl);
|
|
} catch (error) {
|
|
console.error(error);
|
|
}
|
|
} |