mirror of
https://github.com/ThePhaseless/Byparr.git
synced 2025-03-15 01:40:21 +08:00
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
import time
|
|
|
|
import uvicorn.config
|
|
from fastapi import FastAPI
|
|
from fastapi.responses import RedirectResponse
|
|
from sbase import SB, BaseCase
|
|
|
|
from src.models.requests import LinkRequest, LinkResponse, Solution
|
|
from src.utils import logger
|
|
from src.utils.consts import LOG_LEVEL
|
|
|
|
app = FastAPI(debug=LOG_LEVEL == logging.DEBUG, log_level=LOG_LEVEL)
|
|
|
|
|
|
@app.get("/")
|
|
def read_root():
|
|
"""Redirect to /docs."""
|
|
logger.info("Redirecting to /docs")
|
|
return RedirectResponse(url="/docs", status_code=301)
|
|
|
|
|
|
@app.get("/health")
|
|
async def health_check():
|
|
"""Health check endpoint."""
|
|
logger.info("Health check")
|
|
# browser: Chrome = await new_browser()
|
|
# browser.get("https://google.com")
|
|
# browser.stop()
|
|
return {"status": "ok"}
|
|
|
|
|
|
@app.post("/v1")
|
|
def read_item(request: LinkRequest):
|
|
"""Handle POST requests."""
|
|
start_time = int(time.time() * 1000)
|
|
# request.url = "https://nowsecure.nl"
|
|
logger.info(f"Request: {request}")
|
|
response: LinkResponse
|
|
|
|
# start_time = int(time.time() * 1000)
|
|
with SB(uc=True, locale_code="en", test=False, xvfb=True, ad_block=True) as sb:
|
|
sb: BaseCase
|
|
sb.uc_open_with_reconnect(request.url)
|
|
sb.uc_gui_click_captcha()
|
|
logger.info(f"Got webpage: {request.url}")
|
|
sb.save_screenshot("screenshot.png")
|
|
logger.info(f"Got webpage: {request.url}")
|
|
|
|
response = LinkResponse(
|
|
message="Success",
|
|
solution=Solution(
|
|
userAgent=sb.get_user_agent(),
|
|
url=sb.get_current_url(),
|
|
status=200,
|
|
cookies=sb.get_cookies(),
|
|
headers={},
|
|
response=sb.get_page_source(),
|
|
),
|
|
startTimestamp=start_time,
|
|
)
|
|
|
|
return response
|
|
|
|
|
|
if __name__ == "__main__":
|
|
uvicorn.run(app, host="0.0.0.0", port=8191, log_level=LOG_LEVEL) # noqa: S104
|