|
from fastapi import FastAPI |
|
from playwright.async_api import async_playwright, TimeoutError |
|
import re |
|
|
|
app = FastAPI() |
|
|
|
async def scrape_google(query: str): |
|
url = f"https://www.google.com/search?q={query}" |
|
async with async_playwright() as pw: |
|
browser = await pw.chromium.launch(headless=True) |
|
context = await browser.new_context() |
|
page = await context.new_page() |
|
|
|
await page.goto(url, wait_until="domcontentloaded", timeout=60000) |
|
try: |
|
await page.wait_for_selector("div#search", timeout=10000) |
|
except TimeoutError: |
|
pass |
|
|
|
links = [] |
|
for h in await page.query_selector_all("h3"): |
|
try: |
|
a = await h.evaluate_handle("e => e.closest('a')") |
|
href = await a.get_attribute("href") |
|
title = await h.inner_text() |
|
links.append({"title": title, "link": href}) |
|
except: |
|
continue |
|
|
|
results = [] |
|
for item in links[:5]: |
|
await page.goto(item["link"], wait_until="domcontentloaded", timeout=30000) |
|
html = await page.content() |
|
emails = re.findall(r"[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Za-z]{2,}", html) |
|
phones = re.findall(r"\+?\d[\d\s\-/]{7,}\d", html) |
|
results.append({ |
|
**item, |
|
"emails": list(set(emails))[:2], |
|
"phones": list(set(phones))[:2] |
|
}) |
|
|
|
await browser.close() |
|
return results |
|
|
|
@app.get("/search") |
|
async def search(query: str): |
|
data = await scrape_google(query.replace(" ", "+")) |
|
return {"query": query, "results": data} |
|
|