|
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 |
|
) |
|
|
