|
from apscheduler.executors.asyncio import AsyncIOExecutor |
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler |
|
from contextlib import asynccontextmanager |
|
from datetime import datetime |
|
from fastapi import FastAPI, Request |
|
from fastapi.responses import Response, FileResponse |
|
from typing import Any |
|
|
|
import httpx |
|
import json |
|
import subprocess |
|
|
|
|
|
class PrettyJSONResponse(Response): |
|
media_type = "application/json" |
|
|
|
def render(self, content: Any) -> bytes: |
|
return json.dumps(content, indent=2).encode("utf-8") |
|
|
|
|
|
@asynccontextmanager |
|
async def lifespan(app: FastAPI): |
|
scheduler.start() |
|
yield |
|
scheduler.shutdown() |
|
|
|
|
|
app = FastAPI(lifespan=lifespan) |
|
|
|
|
|
@app.middleware("http") |
|
async def log_request(request: Request, call_next: Any): |
|
ts = datetime.now().strftime("%y%m%d%H%M%S%f") |
|
data = { |
|
"day": int(ts[:6]), |
|
"dt": int(ts[:-3]), |
|
"url": request.url, |
|
"query_params": request.query_params, |
|
"client": request.client.host, |
|
"method": request.method, |
|
"headers": dict(request.headers), |
|
} |
|
output = json.dumps( |
|
obj=data, |
|
default=str, |
|
indent=None, |
|
separators=(", ", ":"), |
|
) |
|
with open("a.json", "a") as f: |
|
separator = "\n" if f.tell() else "" |
|
f.write(separator + output) |
|
|
|
response = await call_next(request) |
|
return response |
|
|
|
|
|
@app.get("/") |
|
def read_root(request: Request): |
|
return {"Hello": request.client.host} |
|
|
|
|
|
@app.get("/a", response_class=PrettyJSONResponse) |
|
def get_analytics(n: int = 5): |
|
if n == 0: |
|
cmd = "cat a.json" |
|
else: |
|
cmd = f"tail -{n} a.json" |
|
json_lines = subprocess.run(cmd.split(), capture_output=True).stdout |
|
content = json.loads(f"[{json_lines.replace(b"\n", b",").decode()}]") |
|
return content |
|
|
|
|
|
@app.api_route("/qa", response_class=FileResponse, methods=["GET", "HEAD"]) |
|
def query_analytics(): |
|
return "a.json" |
|
|
|
|
|
@app.get("/favicon.ico") |
|
async def favicon(): |
|
return {"message": "woof!"} |
|
|
|
|
|
@app.get("/ping") |
|
async def ping(): |
|
return {"message": "woof!"} |
|
|
|
|
|
async def self_ping(): |
|
async with httpx.AsyncClient() as client: |
|
_ = await client.get("http://0.0.0.0:7860/ping") |
|
|
|
|
|
scheduler = AsyncIOScheduler(executors={"default": AsyncIOExecutor()}) |
|
scheduler.add_job(self_ping, "interval", minutes=60) |
|
|
