27 lines
696 B
Python
27 lines
696 B
Python
|
from typing import Annotated, Union
|
||
|
|
|
||
|
|
from fastapi import BackgroundTasks, Depends, FastAPI
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
|
||
|
|
def write_log(message: str):
|
||
|
|
with open("log.txt", mode="a") as log:
|
||
|
|
log.write(message)
|
||
|
|
|
||
|
|
|
||
|
|
def get_query(background_tasks: BackgroundTasks, q: Union[str, None] = None):
|
||
|
|
if q:
|
||
|
|
message = f"found query: {q}\n"
|
||
|
|
background_tasks.add_task(write_log, message)
|
||
|
|
return q
|
||
|
|
|
||
|
|
|
||
|
|
@app.post("/send-notification/{email}")
|
||
|
|
async def send_notification(
|
||
|
|
email: str, background_tasks: BackgroundTasks, q: Annotated[str, Depends(get_query)]
|
||
|
|
):
|
||
|
|
message = f"message to {email}\n"
|
||
|
|
background_tasks.add_task(write_log, message)
|
||
|
|
return {"message": "Message sent"}
|