mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 18:44:20 +01:00
95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
import logging
|
|
import sys
|
|
|
|
from fastapi import FastAPI
|
|
from loguru import logger
|
|
|
|
from cashu.core.settings import DEBUG, VERSION
|
|
|
|
from starlette_context.middleware import RawContextMiddleware
|
|
from starlette.middleware import Middleware
|
|
from starlette_context import context
|
|
|
|
|
|
from .router import router
|
|
from .startup import load_ledger
|
|
|
|
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
|
|
|
|
class CustomHeaderMiddleware(BaseHTTPMiddleware):
|
|
"""
|
|
Middleware for starlette that can set the context from request headers
|
|
"""
|
|
|
|
async def dispatch(self, request, call_next):
|
|
context["client-version"] = request.headers.get("Client-version")
|
|
response = await call_next(request)
|
|
return response
|
|
|
|
|
|
def create_app(config_object="core.settings") -> FastAPI:
|
|
def configure_logger() -> None:
|
|
class Formatter:
|
|
def __init__(self):
|
|
self.padding = 0
|
|
self.minimal_fmt: str = "<green>{time:YYYY-MM-DD HH:mm:ss.SS}</green> | <level>{level}</level> | <level>{message}</level>\n"
|
|
if DEBUG:
|
|
self.fmt: str = "<green>{time:YYYY-MM-DD HH:mm:ss.SS}</green> | <level>{level: <4}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> | <level>{message}</level>\n"
|
|
else:
|
|
self.fmt: str = self.minimal_fmt
|
|
|
|
def format(self, record):
|
|
function = "{function}".format(**record)
|
|
if function == "emit": # uvicorn logs
|
|
return self.minimal_fmt
|
|
return self.fmt
|
|
|
|
class InterceptHandler(logging.Handler):
|
|
def emit(self, record):
|
|
try:
|
|
level = logger.level(record.levelname).name
|
|
except ValueError:
|
|
level = record.levelno
|
|
logger.log(level, record.getMessage())
|
|
|
|
logger.remove()
|
|
log_level: str = "DEBUG" if DEBUG else "INFO"
|
|
formatter = Formatter()
|
|
logger.add(sys.stderr, level=log_level, format=formatter.format)
|
|
|
|
logging.getLogger("uvicorn").handlers = [InterceptHandler()]
|
|
logging.getLogger("uvicorn.access").handlers = [InterceptHandler()]
|
|
|
|
configure_logger()
|
|
|
|
middleware = [
|
|
Middleware(
|
|
RawContextMiddleware,
|
|
),
|
|
Middleware(CustomHeaderMiddleware),
|
|
]
|
|
|
|
app = FastAPI(
|
|
title="Cashu Mint",
|
|
description="Ecash wallet and mint with Bitcoin Lightning support.",
|
|
version=VERSION,
|
|
license_info={
|
|
"name": "MIT License",
|
|
"url": "https://raw.githubusercontent.com/callebtc/cashu/main/LICENSE",
|
|
},
|
|
middleware=middleware,
|
|
)
|
|
return app
|
|
|
|
|
|
app = create_app()
|
|
|
|
app.include_router(router=router)
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def startup_load_ledger():
|
|
await load_ledger()
|