File size: 2,409 Bytes
0f8ad38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from typing import List, Tuple
import requests
from bs4 import BeautifulSoup
app = FastAPI()
class SearchQuery(BaseModel):
query: str
class LinkResults(BaseModel):
four_links_click: List[str]
direct4_links: List[str]
magnet_links: List[str]
def scrape_links(query: str) -> List[str]:
try:
url = f"https://www.full4movies.fyi/?s={query}"
response = requests.get(url)
if response.status_code == 200:
soup = BeautifulSoup(response.text, 'html.parser')
elements = soup.find_all('h2', class_='blog-entry-title entry-title')
links = [element.find('a')['href'] for element in elements]
return links
else:
raise HTTPException(status_code=response.status_code, detail="Failed to retrieve the webpage.")
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
def find_links(url: str) -> Tuple[List[str], List[str], List[str]]:
try:
response = requests.get(url)
if response.status_code == 200:
soup = BeautifulSoup(response.text, 'html.parser')
links = soup.find_all('a', href=True)
four_links_click = [link['href'] for link in links if '4links.click' in link['href']]
direct4_links = [link['href'] for link in links if 'direct4.link' in link['href']]
magnet_links = [link['href'] for link in links if link['href'].startswith('magnet:')]
return four_links_click, direct4_links, magnet_links
else:
raise HTTPException(status_code=response.status_code, detail="Failed to retrieve the selected webpage.")
except Exception as e:
raise HTTPException(status_code=500, detail=f"An error occurred while scraping the selected link: {str(e)}")
@app.post("/search", response_model=List[str])
def search_links(payload: SearchQuery):
links = scrape_links(payload.query)
if not links:
raise HTTPException(status_code=404, detail="No links found.")
return links
@app.get("/details", response_model=LinkResults)
def get_details(url: str):
four_links_click, direct4_links, magnet_links = find_links(url)
return LinkResults(
four_links_click=four_links_click,
direct4_links=direct4_links,
magnet_links=magnet_links
)
|