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