feat: handle clickNext pagination by tracking already scraped items

This commit is contained in:
karishmas6
2024-08-19 01:38:41 +05:30
parent 0d1f83c201
commit b6bcc1d516

View File

@@ -371,7 +371,8 @@ export default class Interpreter extends EventEmitter {
private async handlePagination(page: Page, config: { listSelector: string, fields: any, limit?: number, pagination: any }) { private async handlePagination(page: Page, config: { listSelector: string, fields: any, limit?: number, pagination: any }) {
let allResults: Record<string, any>[] = []; let allResults: Record<string, any>[] = [];
let previousHeight = 0 let previousHeight = 0;
let scrapedItems: Set<string> = new Set(); // Track unique items to avoid re-scraping
while (true) { while (true) {
switch (config.pagination.type) { switch (config.pagination.type) {
@@ -391,16 +392,35 @@ export default class Interpreter extends EventEmitter {
case 'scrollUp': case 'scrollUp':
break; break;
case 'clickNext': case 'clickNext':
const nextButton = await page.$(config.pagination.selector); const pageResults = await page.evaluate((cfg) => window.scrapeList(cfg), config);
if (!nextButton) {
const finalResults = await page.evaluate((cfg) => window.scrapeList(cfg), config); // Filter out items that have already been scraped
allResults = allResults.concat(finalResults); const newResults = pageResults.filter(item => {
return allResults; const uniqueKey = JSON.stringify(item);
if (scrapedItems.has(uniqueKey)) return false;
scrapedItems.add(uniqueKey);
return true;
});
allResults = allResults.concat(newResults);
// If the limit is reached, return the required number of items
if (config.limit && allResults.length >= config.limit) {
return allResults.slice(0, config.limit);
} }
// Check if there's a next page button
const nextButton = await page.$(config.pagination.selector);
if (!nextButton) {
return allResults; // No more pages to navigate
}
// Click the next button and wait for the navigation to complete
await Promise.all([ await Promise.all([
nextButton.click(), nextButton.click(),
page.waitForNavigation({ waitUntil: 'networkidle' }) page.waitForNavigation({ waitUntil: 'networkidle' })
]); ]);
break; break;
case 'clickLoadMore': case 'clickLoadMore':
const loadMoreButton = await page.$(config.pagination.selector); const loadMoreButton = await page.$(config.pagination.selector);