mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-01-29 00:00:26 +00:00
9a2aea038a
- Add pyproject.toml with uv and ruff configuration - Pin Python version to 3.13 via .python-version - Add Makefile commands: lint, format, fix - Apply ruff formatting to entire codebase - Remove unused imports (base64 in yookassa/simple_subscription) - Update .gitignore for new config files
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import AsyncGenerator
|
|
|
|
from fastapi import Depends, HTTPException, Request, Security, status
|
|
from fastapi.security import APIKeyHeader
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database.database import AsyncSessionLocal
|
|
from app.database.models import WebApiToken
|
|
from app.services.web_api_token_service import web_api_token_service
|
|
|
|
|
|
api_key_header_scheme = APIKeyHeader(name='X-API-Key', auto_error=False)
|
|
|
|
|
|
async def get_db_session() -> AsyncGenerator[AsyncSession]:
|
|
async with AsyncSessionLocal() as session:
|
|
try:
|
|
yield session
|
|
finally:
|
|
await session.close()
|
|
|
|
|
|
async def require_api_token(
|
|
request: Request,
|
|
api_key_header: str | None = Security(api_key_header_scheme),
|
|
db: AsyncSession = Depends(get_db_session),
|
|
) -> WebApiToken:
|
|
api_key = api_key_header
|
|
|
|
if not api_key:
|
|
authorization = request.headers.get('Authorization')
|
|
if authorization:
|
|
scheme, _, credentials = authorization.partition(' ')
|
|
if scheme.lower() == 'bearer' and credentials:
|
|
api_key = credentials
|
|
|
|
if not api_key:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail='Missing API key',
|
|
)
|
|
|
|
token = await web_api_token_service.authenticate(
|
|
db,
|
|
api_key,
|
|
remote_ip=request.client.host if request.client else None,
|
|
)
|
|
|
|
if not token:
|
|
await db.rollback()
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail='Invalid or expired API key',
|
|
)
|
|
|
|
await db.commit()
|
|
return token
|