add custom error handler for lnurl
This commit is contained in:
parent
527d1c9ae1
commit
1072db72c1
1 changed files with 27 additions and 1 deletions
28
__init__.py
28
__init__.py
|
|
@ -1,8 +1,12 @@
|
|||
from fastapi import APIRouter
|
||||
from fastapi import APIRouter, Request, Response
|
||||
from fastapi.routing import APIRoute
|
||||
|
||||
from fastapi.staticfiles import StaticFiles
|
||||
from fastapi.responses import JSONResponse
|
||||
|
||||
from lnbits.db import Database
|
||||
from lnbits.helpers import template_renderer
|
||||
from typing import Callable
|
||||
|
||||
db = Database("ext_withdraw")
|
||||
|
||||
|
|
@ -15,7 +19,29 @@ withdraw_static_files = [
|
|||
]
|
||||
|
||||
|
||||
class LNURLErrorResponseHandler(APIRoute):
|
||||
def get_route_handler(self) -> Callable:
|
||||
original_route_handler = super().get_route_handler()
|
||||
|
||||
async def custom_route_handler(request: Request) -> Response:
|
||||
try:
|
||||
response = await original_route_handler(request)
|
||||
except HTTPException as exc:
|
||||
logger.debug(f"HTTPException: {exc}")
|
||||
response = JSONResponse(
|
||||
status_code=exc.status_code,
|
||||
content={"status": "ERROR", "reason": f"{exc.detail}"},
|
||||
)
|
||||
except Exception as exc:
|
||||
raise exc
|
||||
|
||||
return response
|
||||
|
||||
return custom_route_handler
|
||||
|
||||
|
||||
withdraw_ext: APIRouter = APIRouter(prefix="/withdraw", tags=["withdraw"])
|
||||
withdraw_ext.route_class = LNURLErrorResponseHandler
|
||||
|
||||
|
||||
def withdraw_renderer():
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue