mirror of
https://github.com/grillazz/fastapi-sqlalchemy-asyncpg.git
synced 2025-08-26 16:40:40 +03:00
add auth and redis endpoints
This commit is contained in:
parent
0b2d2e7933
commit
32346cd7e7
16
app/api/health.py
Normal file
16
app/api/health.py
Normal file
@ -0,0 +1,16 @@
|
||||
import logging
|
||||
|
||||
from fastapi import APIRouter, status, Request
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.get("/redis", status_code=status.HTTP_200_OK)
|
||||
async def redis_check(request: Request):
|
||||
_redis = await request.app.state.redis
|
||||
_info = None
|
||||
try:
|
||||
_info = await _redis.info()
|
||||
except Exception as e:
|
||||
logging.error(f"Redis error: {e}")
|
||||
return _info
|
31
app/main.py
31
app/main.py
@ -1,24 +1,35 @@
|
||||
from fastapi import FastAPI
|
||||
from contextlib import asynccontextmanager
|
||||
|
||||
from fastapi import FastAPI, Depends
|
||||
|
||||
from app.api.nonsense import router as nonsense_router
|
||||
from app.api.shakespeare import router as shakespeare_router
|
||||
from app.api.stuff import router as stuff_router
|
||||
from app.utils.logging import AppLogger
|
||||
from app.api.user import router as user_router
|
||||
from app.api.health import router as health_router
|
||||
from app.redis import get_redis
|
||||
from app.services.auth import AuthBearer
|
||||
|
||||
logger = AppLogger.__call__().get_logger()
|
||||
|
||||
app = FastAPI(title="Stuff And Nonsense API", version="0.5")
|
||||
|
||||
@asynccontextmanager
|
||||
async def lifespan(app: FastAPI):
|
||||
# Load the redis connection
|
||||
app.state.redis = await get_redis()
|
||||
yield
|
||||
# close redis connection and release the resources
|
||||
app.state.redis.close()
|
||||
|
||||
|
||||
app = FastAPI(title="Stuff And Nonsense API", version="0.6", lifespan=lifespan)
|
||||
|
||||
app.include_router(stuff_router)
|
||||
app.include_router(nonsense_router)
|
||||
app.include_router(shakespeare_router)
|
||||
app.include_router(user_router)
|
||||
|
||||
|
||||
@app.on_event("startup")
|
||||
async def startup_event():
|
||||
logger.info("Starting up...")
|
||||
|
||||
|
||||
@app.on_event("shutdown")
|
||||
async def shutdown_event():
|
||||
logger.info("Shutting down...")
|
||||
app.include_router(health_router, prefix="/v1/public/health", tags=["Health, Public"])
|
||||
app.include_router(health_router, prefix="/v1/health", tags=["Health, Bearer"], dependencies=[Depends(AuthBearer())])
|
||||
|
14
app/redis.py
Normal file
14
app/redis.py
Normal file
@ -0,0 +1,14 @@
|
||||
import redis.asyncio as redis
|
||||
|
||||
from app import config
|
||||
|
||||
|
||||
global_settings = config.get_settings()
|
||||
|
||||
|
||||
async def get_redis():
|
||||
return await redis.from_url(
|
||||
global_settings.redis_url.unicode_string(),
|
||||
encoding="utf-8",
|
||||
decode_responses=True,
|
||||
)
|
@ -38,7 +38,7 @@ async def create_access_token(user: User, request: Request):
|
||||
_payload = {
|
||||
"email": user.email,
|
||||
"expiry": time.time() + global_settings.jwt_expire,
|
||||
"platform": request.headers.get("User-Agent")
|
||||
"platform": request.headers.get("User-Agent"),
|
||||
}
|
||||
_token = jwt.encode(_payload, str(user.password), algorithm=global_settings.jwt_algorithm)
|
||||
|
||||
|
@ -1,6 +1,6 @@
|
||||
[tool.poetry]
|
||||
name = "fastapi-sqlalchemy-asyncpg"
|
||||
version = "0.0.5"
|
||||
version = "0.0.6"
|
||||
description = ""
|
||||
authors = ["Jakub Miazek <the@grillazz.com>"]
|
||||
packages = []
|
||||
|
Loading…
x
Reference in New Issue
Block a user