improved types

This commit is contained in:
Robert 2024-02-28 12:37:22 +07:00
parent 9d14dc43cd
commit 83acdfdc89
No known key found for this signature in database
GPG Key ID: F631C7FD957D5F22
3 changed files with 14 additions and 12 deletions

View File

@ -1,14 +1,10 @@
from loguru import logger
from fastapi import APIRouter, HTTPException
from chain_service.database.models.chain import Chain
from chain_service.repositories.chain import ChainRepository
from chain_service.dependencies.chain_repository import get_chain_repository
from typing import Annotated
from fastapi import APIRouter, Depends, HTTPException
from chain_service.dependencies.chain_repository import ChainRepositoryDependency
router = APIRouter(prefix="/chain")
ChainRepositoryDependency = Annotated[ChainRepository, Depends(get_chain_repository)]
@router.post("/")

View File

@ -1,12 +1,12 @@
from .database import get_database
from chain_service.database.database import Database
from .database import DatabaseDependency
from chain_service.repositories.chain import ChainRepository
from fastapi import Depends
from typing import Annotated
from chain_service.repositories.chain import ChainRepository
def get_chain_repository(
database: Annotated[Database, Depends(get_database)]
) -> ChainRepository:
def get_chain_repository(database: DatabaseDependency) -> ChainRepository:
return ChainRepository(database=database)
ChainRepositoryDependency = Annotated[ChainRepository, Depends(get_chain_repository)]

View File

@ -1,6 +1,9 @@
from chain_service.settings import Settings
from chain_service.database.database import Database
from fastapi import Depends
from typing import Annotated
def get_database() -> Database:
settings = Settings()
@ -8,3 +11,6 @@ def get_database() -> Database:
return Database(
database_url=settings.database_url, database_name=settings.database_name
)
DatabaseDependency = Annotated[Database, Depends(get_database)]