Spaces:
Sleeping
Sleeping
Update scraper.py
Browse files- scraper.py +33 -1
scraper.py
CHANGED
@@ -8,6 +8,38 @@ import requests
|
|
8 |
|
9 |
|
10 |
class Scraper:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
@staticmethod
|
12 |
async def power_scrapper(url):
|
13 |
async with async_playwright() as p:
|
@@ -76,6 +108,6 @@ class Scraper:
|
|
76 |
|
77 |
if not links:
|
78 |
print("Running alternative scrapper")
|
79 |
-
links, text_content = await Scraper.
|
80 |
|
81 |
return {"title": title, "URL": links, "Content": text_content}
|
|
|
8 |
|
9 |
|
10 |
class Scraper:
|
11 |
+
@staticmethod
|
12 |
+
async def power_scrapper_2(url):
|
13 |
+
async with async_playwright() as p:
|
14 |
+
browser = await p.chromium.launch(headless=True)
|
15 |
+
page = await browser.new_page()
|
16 |
+
await page.goto(url)
|
17 |
+
|
18 |
+
# Get the title
|
19 |
+
#title = await page.title()
|
20 |
+
|
21 |
+
# Get all links
|
22 |
+
page_url = await page.evaluate("""() => {
|
23 |
+
return Array.from(document.querySelectorAll('a')).map(a => a.href);
|
24 |
+
}""")
|
25 |
+
|
26 |
+
# Get page content (paragraphs, headers)
|
27 |
+
page_content = await page.evaluate("""() => {
|
28 |
+
let elements = Array.from(document.querySelectorAll('p, h1, h2, h3, h4, h5, h6'));
|
29 |
+
return elements.map(element => ({
|
30 |
+
tag: element.tagName,
|
31 |
+
text: element.innerText
|
32 |
+
}));
|
33 |
+
}""")
|
34 |
+
|
35 |
+
# Print the results
|
36 |
+
# print(f"Title: {title}")
|
37 |
+
# print(f"Links: {links}")
|
38 |
+
# print(f"Content: {content}")
|
39 |
+
|
40 |
+
await browser.close()
|
41 |
+
return page_url, page_content
|
42 |
+
|
43 |
@staticmethod
|
44 |
async def power_scrapper(url):
|
45 |
async with async_playwright() as p:
|
|
|
108 |
|
109 |
if not links:
|
110 |
print("Running alternative scrapper")
|
111 |
+
links, text_content = await Scraper.power_scrapper_2(url)
|
112 |
|
113 |
return {"title": title, "URL": links, "Content": text_content}
|