28 lines
777 B
Python
28 lines
777 B
Python
from fastapi import FastAPI
|
|
from pydantic import BaseModel
|
|
from missing_kids import MissingKidsFetcher
|
|
from logger import get_logger
|
|
logger = get_logger()
|
|
|
|
app = FastAPI()
|
|
|
|
@app.get("/get_missing_kids/")
|
|
def get_missing_kids(pages: int = -1):
|
|
try:
|
|
logger.info("Get missing kids, #pages={}".format(pages))
|
|
res = {"list_urls": MissingKidsFetcher().get_missing_kids_urls(first_n_pages=pages)}
|
|
except Exception as e:
|
|
res = {}
|
|
return res
|
|
|
|
class Body(BaseModel):
|
|
url: str
|
|
|
|
@app.post("/verify_missing_kid/")
|
|
def get_missing_kids(data: Body):
|
|
try:
|
|
logger.info("Verify missing kid, URL={}".format(data.url))
|
|
res = MissingKidsFetcher().verify_missing_kid_url(data.url)
|
|
except Exception as e:
|
|
res = {}
|
|
return res |