improved types

chain_service/controllers/chain.py

@@ -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 ChainRepositoryDependency
-from chain_service.dependencies.chain_repository import get_chain_repository
-
-from typing import Annotated
-from fastapi import APIRouter, Depends, HTTPException
 
 router = APIRouter(prefix="/chain")
-ChainRepositoryDependency = Annotated[ChainRepository, Depends(get_chain_repository)]
 
 
 @router.post("/")

chain_service/dependencies/chain_repository.py

@@ -1,12 +1,12 @@
-from .database import get_database
+from .database import DatabaseDependency
-from chain_service.database.database import Database
+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(
+def get_chain_repository(database: DatabaseDependency) -> ChainRepository:
-    database: Annotated[Database, Depends(get_database)]
-) -> ChainRepository:
     return ChainRepository(database=database)
+
+
+ChainRepositoryDependency = Annotated[ChainRepository, Depends(get_chain_repository)]

chain_service/dependencies/database.py

@@ -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)]