This commit is contained in:
@@ -522,6 +522,398 @@ async def cache_stats(x_api_key: Optional[str] = Header(None)):
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@app.get("/pagination")
|
||||||
|
async def detect_pagination(url: str, x_api_key: Optional[str] = Header(None)):
|
||||||
|
"""Detect pagination on a website and determine the pagination pattern"""
|
||||||
|
# Validate API key
|
||||||
|
if not x_api_key or x_api_key != API_KEY:
|
||||||
|
raise HTTPException(status_code=401, detail="Invalid API key")
|
||||||
|
|
||||||
|
# Decode URL if it's encoded
|
||||||
|
decoded_url = unquote(url)
|
||||||
|
|
||||||
|
# Check cache first
|
||||||
|
cached_result = get_cached_data(decoded_url, "pagination")
|
||||||
|
if cached_result:
|
||||||
|
return cached_result
|
||||||
|
|
||||||
|
try:
|
||||||
|
print(f"Detecting pagination on: {decoded_url}")
|
||||||
|
|
||||||
|
# Define the operation to perform with the browser
|
||||||
|
async def pagination_operation(page):
|
||||||
|
try:
|
||||||
|
# Navigate to the URL
|
||||||
|
await page.goto(decoded_url, waitUntil='networkidle2', timeout=30000)
|
||||||
|
original_url = page.url
|
||||||
|
|
||||||
|
print(f"Successfully loaded page: {original_url}")
|
||||||
|
|
||||||
|
# Before clicking any pagination links, try to determine the last page
|
||||||
|
# by analyzing all pagination-related links on the current page
|
||||||
|
last_page = await page.evaluate('''() => {
|
||||||
|
console.log("Analyzing page for last page number before navigation");
|
||||||
|
|
||||||
|
// Get all links on the page
|
||||||
|
const links = Array.from(document.querySelectorAll('a'));
|
||||||
|
console.log(`Found ${links.length} links to analyze`);
|
||||||
|
|
||||||
|
let lastPage = null;
|
||||||
|
let lastPageSource = '';
|
||||||
|
|
||||||
|
// Strategy 1: Find numeric links (e.g., page numbers)
|
||||||
|
const numericLinks = links.filter(link => {
|
||||||
|
const text = link.innerText.trim();
|
||||||
|
return /^[0-9]+$/.test(text) && link.href && link.href !== '#';
|
||||||
|
});
|
||||||
|
|
||||||
|
if (numericLinks.length > 0) {
|
||||||
|
const numericValues = numericLinks.map(link => parseInt(link.innerText.trim()));
|
||||||
|
const maxNumeric = Math.max(...numericValues);
|
||||||
|
console.log(`Found highest numeric link: ${maxNumeric}`);
|
||||||
|
if (maxNumeric > 1) {
|
||||||
|
lastPage = maxNumeric;
|
||||||
|
lastPageSource = 'numeric-links';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Strategy 2: Look for "last page" link
|
||||||
|
const lastLinks = links.filter(link => {
|
||||||
|
const text = link.innerText.trim().toLowerCase();
|
||||||
|
const classes = (link.className || '').toLowerCase();
|
||||||
|
const ariaLabel = (link.getAttribute('aria-label') || '').toLowerCase();
|
||||||
|
|
||||||
|
return (text === 'last' ||
|
||||||
|
classes.includes('last') ||
|
||||||
|
ariaLabel.includes('last') ||
|
||||||
|
link.getAttribute('rel') === 'last');
|
||||||
|
});
|
||||||
|
|
||||||
|
if (lastLinks.length > 0) {
|
||||||
|
console.log(`Found ${lastLinks.length} "last" links`);
|
||||||
|
// Try to extract page number from the URL
|
||||||
|
const lastLink = lastLinks[0];
|
||||||
|
const href = lastLink.href;
|
||||||
|
console.log(`Last link href: ${href}`);
|
||||||
|
|
||||||
|
// Common patterns: page=X, /page/X, etc.
|
||||||
|
const pagePatterns = [
|
||||||
|
/[?&]page=(\d+)/,
|
||||||
|
/[?&]p=(\d+)/,
|
||||||
|
/[?&]pg=(\d+)/,
|
||||||
|
/\/page\/(\d+)/,
|
||||||
|
/\/p\/(\d+)/,
|
||||||
|
/\/paged\/(\d+)/
|
||||||
|
];
|
||||||
|
|
||||||
|
for (const pattern of pagePatterns) {
|
||||||
|
const match = href.match(pattern);
|
||||||
|
if (match && match[1]) {
|
||||||
|
const pageNum = parseInt(match[1]);
|
||||||
|
console.log(`Found page number ${pageNum} in last link URL`);
|
||||||
|
if (pageNum > 1 && (lastPage === null || pageNum > lastPage)) {
|
||||||
|
lastPage = pageNum;
|
||||||
|
lastPageSource = 'last-link-url';
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Strategy 3: Look for pagination text like "Page 1 of 42"
|
||||||
|
const paginationTexts = [];
|
||||||
|
document.querySelectorAll('.pagination, .pager, .paginator, .paging, .page-numbers, .pages, .page-navigation')
|
||||||
|
.forEach(el => paginationTexts.push(el.innerText));
|
||||||
|
|
||||||
|
// Also check for any element that might contain pagination info
|
||||||
|
document.querySelectorAll('[class*="pag"], [id*="pag"]')
|
||||||
|
.forEach(el => paginationTexts.push(el.innerText));
|
||||||
|
|
||||||
|
const pageOfPatterns = [
|
||||||
|
/page\s+\d+\s+of\s+(\d+)/i,
|
||||||
|
/page\s+\d+\s*\/\s*(\d+)/i,
|
||||||
|
/\d+\s*\/\s*(\d+)\s+pages/i,
|
||||||
|
/\d+\s*-\s*\d+\s+of\s+\d+\s+\(\s*(\d+)\s+pages\s*\)/i,
|
||||||
|
/showing\s+\d+\s*-\s*\d+\s+of\s+\d+\s+\(\s*(\d+)\s+pages\s*\)/i,
|
||||||
|
/\d+\s*-\s*\d+\s+of\s+\d+\s+items\s+\(\s*(\d+)\s+pages\s*\)/i
|
||||||
|
];
|
||||||
|
|
||||||
|
for (const text of paginationTexts) {
|
||||||
|
for (const pattern of pageOfPatterns) {
|
||||||
|
const match = text.match(pattern);
|
||||||
|
if (match && match[1]) {
|
||||||
|
const pageNum = parseInt(match[1]);
|
||||||
|
console.log(`Found page count ${pageNum} in text`);
|
||||||
|
if (pageNum > 1 && (lastPage === null || pageNum > lastPage)) {
|
||||||
|
lastPage = pageNum;
|
||||||
|
lastPageSource = 'pagination-text';
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// NEW STRATEGY: Analyze all link URLs for page numbers
|
||||||
|
console.log("Analyzing all link URLs for page numbers");
|
||||||
|
|
||||||
|
// Common URL patterns that indicate pagination
|
||||||
|
const urlPagePatterns = [
|
||||||
|
/[?&]page=(\d+)/,
|
||||||
|
/[?&]p=(\d+)/,
|
||||||
|
/[?&]pg=(\d+)/,
|
||||||
|
/\/page\/(\d+)/,
|
||||||
|
/\/p\/(\d+)/,
|
||||||
|
/\/paged\/(\d+)/,
|
||||||
|
/\/pages\/(\d+)/,
|
||||||
|
/[?&]offset=(\d+)/,
|
||||||
|
/[?&]start=(\d+)/,
|
||||||
|
/[?&]from=(\d+)/,
|
||||||
|
/[?&]paged=(\d+)/,
|
||||||
|
/[?&]pagenum=(\d+)/,
|
||||||
|
/[?&]pageNumber=(\d+)/,
|
||||||
|
/[?&]currentpage=(\d+)/
|
||||||
|
];
|
||||||
|
|
||||||
|
// Extract page numbers from all link URLs
|
||||||
|
const pageNumbersFromUrls = [];
|
||||||
|
links.forEach(link => {
|
||||||
|
if (!link.href || link.href === '#') return;
|
||||||
|
|
||||||
|
const href = link.href;
|
||||||
|
for (const pattern of urlPagePatterns) {
|
||||||
|
const match = href.match(pattern);
|
||||||
|
if (match && match[1]) {
|
||||||
|
const pageNum = parseInt(match[1]);
|
||||||
|
if (pageNum > 1) {
|
||||||
|
pageNumbersFromUrls.push(pageNum);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
if (pageNumbersFromUrls.length > 0) {
|
||||||
|
const maxPageFromUrls = Math.max(...pageNumbersFromUrls);
|
||||||
|
console.log(`Found highest page number in URLs: ${maxPageFromUrls} (from ${pageNumbersFromUrls.length} links)`);
|
||||||
|
if (maxPageFromUrls > 1 && (lastPage === null || maxPageFromUrls > lastPage)) {
|
||||||
|
lastPage = maxPageFromUrls;
|
||||||
|
lastPageSource = 'url-analysis';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`Pre-navigation last page detection: ${lastPage} (source: ${lastPageSource})`);
|
||||||
|
return { lastPage, lastPageSource };
|
||||||
|
}''');
|
||||||
|
|
||||||
|
print(f"Pre-navigation last page detection: {last_page.get('lastPage')} (source: {last_page.get('lastPageSource')})")
|
||||||
|
|
||||||
|
# Find and click on pagination elements
|
||||||
|
pagination_clicked = await page.evaluate('''async () => {
|
||||||
|
// Common pagination link selectors to try
|
||||||
|
const selectors = [
|
||||||
|
'a.next', 'a.page-next', 'a[rel="next"]',
|
||||||
|
'a[aria-label="Next page"]', 'a[aria-label="next"]',
|
||||||
|
'.pagination a:nth-child(2)', // Often the "2" link
|
||||||
|
'.pagination li:nth-child(2) a',
|
||||||
|
'a:contains("2")', 'a:contains("Next")', 'a:contains("»")'
|
||||||
|
];
|
||||||
|
|
||||||
|
// Try to find a page "2" link first
|
||||||
|
const links = Array.from(document.querySelectorAll('a'));
|
||||||
|
const page2Link = links.find(link => {
|
||||||
|
const text = link.innerText.trim();
|
||||||
|
return text === '2' && link.href && link.href !== '#';
|
||||||
|
});
|
||||||
|
|
||||||
|
if (page2Link) {
|
||||||
|
// Store the href before clicking
|
||||||
|
const href = page2Link.href;
|
||||||
|
// Click the link
|
||||||
|
page2Link.click();
|
||||||
|
return { clicked: true, href: href, type: 'numeric' };
|
||||||
|
}
|
||||||
|
|
||||||
|
// Try common selectors
|
||||||
|
for (const selector of selectors) {
|
||||||
|
try {
|
||||||
|
if (selector.includes(':contains')) {
|
||||||
|
// Handle jQuery-style :contains selector
|
||||||
|
const text = selector.match(/:contains\\("(.+)"\\)/)[1];
|
||||||
|
const link = links.find(l => l.innerText.includes(text) && l.href && l.href !== '#');
|
||||||
|
if (link) {
|
||||||
|
const href = link.href;
|
||||||
|
link.click();
|
||||||
|
return { clicked: true, href: href, type: 'selector' };
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
const element = document.querySelector(selector);
|
||||||
|
if (element && element.href && element.href !== '#') {
|
||||||
|
const href = element.href;
|
||||||
|
element.click();
|
||||||
|
return { clicked: true, href: href, type: 'selector' };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
// Ignore errors for invalid selectors
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Look for any link that might be pagination
|
||||||
|
const paginationLinks = links.filter(link => {
|
||||||
|
if (!link.href || link.href === '#') return false;
|
||||||
|
|
||||||
|
const text = link.innerText.trim();
|
||||||
|
const href = link.href;
|
||||||
|
|
||||||
|
// Check for numeric text or next/prev indicators
|
||||||
|
const isNumeric = /^[0-9]+$/.test(text) && text !== '1';
|
||||||
|
const isNextPrev = /next|prev|previous|older|newer/i.test(text) ||
|
||||||
|
/[»«‹›<>]/.test(text);
|
||||||
|
|
||||||
|
// Check for page parameter in URL
|
||||||
|
const hasPageParam = /[?&]page=|[?&]p=|[?&]pg=|\/page\/|\/p\//.test(href);
|
||||||
|
|
||||||
|
return (isNumeric || isNextPrev || hasPageParam);
|
||||||
|
});
|
||||||
|
|
||||||
|
if (paginationLinks.length > 0) {
|
||||||
|
const link = paginationLinks[0];
|
||||||
|
const href = link.href;
|
||||||
|
link.click();
|
||||||
|
return { clicked: true, href: href, type: 'other' };
|
||||||
|
}
|
||||||
|
|
||||||
|
return { clicked: false };
|
||||||
|
}''')
|
||||||
|
|
||||||
|
# If no pagination was found or clicked
|
||||||
|
if not pagination_clicked.get('clicked', False):
|
||||||
|
return {
|
||||||
|
"status": "success",
|
||||||
|
"url": decoded_url,
|
||||||
|
"hasPagination": False,
|
||||||
|
"urlTemplate": None,
|
||||||
|
"lastPage": last_page.get('lastPage')
|
||||||
|
}
|
||||||
|
|
||||||
|
# Wait for navigation to complete after the click
|
||||||
|
try:
|
||||||
|
await page.waitForNavigation({'timeout': 10000, 'waitUntil': 'networkidle2'})
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Navigation timeout: {e}")
|
||||||
|
|
||||||
|
# Get the new URL after clicking
|
||||||
|
next_page_url = page.url
|
||||||
|
|
||||||
|
# If URL didn't change, pagination might be handled by AJAX
|
||||||
|
if next_page_url == original_url:
|
||||||
|
return {
|
||||||
|
"status": "success",
|
||||||
|
"url": decoded_url,
|
||||||
|
"hasPagination": True,
|
||||||
|
"urlTemplate": "AJAX pagination (URL doesn't change)",
|
||||||
|
"lastPage": last_page.get('lastPage'),
|
||||||
|
"nextPageUrl": next_page_url,
|
||||||
|
"originalUrl": original_url,
|
||||||
|
"expectedHref": pagination_clicked.get('href')
|
||||||
|
}
|
||||||
|
|
||||||
|
print(f"Navigation successful: {original_url} -> {next_page_url}")
|
||||||
|
|
||||||
|
# Analyze the URL structure to determine pagination pattern
|
||||||
|
url_template = None
|
||||||
|
|
||||||
|
# Compare the original and new URLs to find the pagination pattern
|
||||||
|
if '?' in next_page_url:
|
||||||
|
# Extract query parameters from both URLs
|
||||||
|
original_params = {}
|
||||||
|
if '?' in original_url:
|
||||||
|
original_query = original_url.split('?')[1].split('#')[0]
|
||||||
|
for param in original_query.split('&'):
|
||||||
|
if '=' in param:
|
||||||
|
key, value = param.split('=', 1)
|
||||||
|
original_params[key] = value
|
||||||
|
|
||||||
|
next_query = next_page_url.split('?')[1].split('#')[0]
|
||||||
|
next_params = {}
|
||||||
|
for param in next_query.split('&'):
|
||||||
|
if '=' in param:
|
||||||
|
key, value = param.split('=', 1)
|
||||||
|
next_params[key] = value
|
||||||
|
|
||||||
|
# Find parameters that changed or were added
|
||||||
|
pagination_param = None
|
||||||
|
for key, value in next_params.items():
|
||||||
|
# Check if parameter is new or changed
|
||||||
|
if key not in original_params or original_params[key] != value:
|
||||||
|
# Check if the value is numeric and could be a page number
|
||||||
|
if value.isdigit() and int(value) > 1:
|
||||||
|
pagination_param = key
|
||||||
|
break
|
||||||
|
|
||||||
|
# If we found a pagination parameter
|
||||||
|
if pagination_param:
|
||||||
|
base_url = next_page_url.split('?')[0]
|
||||||
|
|
||||||
|
# Reconstruct the URL template with all parameters
|
||||||
|
query_parts = []
|
||||||
|
for key, value in next_params.items():
|
||||||
|
if key == pagination_param:
|
||||||
|
query_parts.append(f"{key}={{PAGE_NUMBER}}")
|
||||||
|
else:
|
||||||
|
query_parts.append(f"{key}={value}")
|
||||||
|
|
||||||
|
url_template = f"{base_url}?{'&'.join(query_parts)}"
|
||||||
|
|
||||||
|
# If we couldn't determine from query parameters, check for path-based pagination
|
||||||
|
if not url_template:
|
||||||
|
# Path-based pagination patterns
|
||||||
|
path_patterns = ['/page/', '/p/', '/paged/', '/pages/']
|
||||||
|
for pattern in path_patterns:
|
||||||
|
if pattern in next_page_url:
|
||||||
|
parts = next_page_url.split(pattern)
|
||||||
|
url_template = f"{parts[0]}{pattern}{{PAGE_NUMBER}}"
|
||||||
|
if len(parts) > 1 and '/' in parts[1]:
|
||||||
|
suffix = parts[1].split('/', 1)[1]
|
||||||
|
if suffix:
|
||||||
|
url_template += f"/{suffix}"
|
||||||
|
break
|
||||||
|
|
||||||
|
# If we still couldn't determine the pattern, use the original and next URLs as examples
|
||||||
|
if not url_template:
|
||||||
|
url_template = f"Pattern unclear. Example: {original_url} → {next_page_url}"
|
||||||
|
|
||||||
|
result = {
|
||||||
|
"status": "success",
|
||||||
|
"hasPagination": True,
|
||||||
|
"urlTemplate": url_template,
|
||||||
|
"lastPage": last_page.get('lastPage')
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Error during pagination detection: {e}")
|
||||||
|
return {
|
||||||
|
"status": "error",
|
||||||
|
"url": decoded_url,
|
||||||
|
"error": str(e),
|
||||||
|
"hasPagination": False,
|
||||||
|
"urlTemplate": None,
|
||||||
|
"lastPage": None
|
||||||
|
}
|
||||||
|
|
||||||
|
# Perform the operation
|
||||||
|
result = await safe_browser_operation(decoded_url, pagination_operation)
|
||||||
|
|
||||||
|
# Save to cache
|
||||||
|
save_to_cache(decoded_url, "pagination", result)
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(status_code=500, detail=str(e))
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
import uvicorn
|
import uvicorn
|
||||||
uvicorn.run(app, host="0.0.0.0", port=8000)
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|
||||||
|
|||||||
Reference in New Issue
Block a user