mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2025-02-27 17:25:03 +03:00
Add application
This commit is contained in:
parent
a1f779a9f3
commit
e000f821b3
2
examples/miniapps/fastapi-sqlalchemy/config.yml
Normal file
2
examples/miniapps/fastapi-sqlalchemy/config.yml
Normal file
|
@ -0,0 +1,2 @@
|
||||||
|
db:
|
||||||
|
url: "sqlite:///./sql_app.db"
|
5
examples/miniapps/fastapi-sqlalchemy/requirements.txt
Normal file
5
examples/miniapps/fastapi-sqlalchemy/requirements.txt
Normal file
|
@ -0,0 +1,5 @@
|
||||||
|
dependency-injector
|
||||||
|
fastapi
|
||||||
|
uvicorn
|
||||||
|
pyyaml
|
||||||
|
sqlalchemy
|
1
examples/miniapps/fastapi-sqlalchemy/webapp/__init__.py
Normal file
1
examples/miniapps/fastapi-sqlalchemy/webapp/__init__.py
Normal file
|
@ -0,0 +1 @@
|
||||||
|
"""Top-level package."""
|
23
examples/miniapps/fastapi-sqlalchemy/webapp/application.py
Normal file
23
examples/miniapps/fastapi-sqlalchemy/webapp/application.py
Normal file
|
@ -0,0 +1,23 @@
|
||||||
|
"""Application module."""
|
||||||
|
|
||||||
|
from fastapi import FastAPI
|
||||||
|
|
||||||
|
from .containers import Container
|
||||||
|
from . import endpoints
|
||||||
|
|
||||||
|
|
||||||
|
def create_app() -> FastAPI:
|
||||||
|
container = Container()
|
||||||
|
container.config.from_yaml('config.yml')
|
||||||
|
container.wire(modules=[endpoints])
|
||||||
|
|
||||||
|
db = container.db()
|
||||||
|
db.create_database()
|
||||||
|
|
||||||
|
app = FastAPI()
|
||||||
|
app.container = container
|
||||||
|
app.include_router(endpoints.router)
|
||||||
|
return app
|
||||||
|
|
||||||
|
|
||||||
|
app = create_app()
|
24
examples/miniapps/fastapi-sqlalchemy/webapp/containers.py
Normal file
24
examples/miniapps/fastapi-sqlalchemy/webapp/containers.py
Normal file
|
@ -0,0 +1,24 @@
|
||||||
|
"""Containers module."""
|
||||||
|
|
||||||
|
from dependency_injector import containers, providers
|
||||||
|
|
||||||
|
from .database import Database
|
||||||
|
from .repositories import UserRepository
|
||||||
|
from .services import UserService
|
||||||
|
|
||||||
|
|
||||||
|
class Container(containers.DeclarativeContainer):
|
||||||
|
|
||||||
|
config = providers.Configuration()
|
||||||
|
|
||||||
|
db = providers.Singleton(Database, db_url=config.db.url)
|
||||||
|
|
||||||
|
users_repository = providers.Factory(
|
||||||
|
UserRepository,
|
||||||
|
session_factory=db.provided.session,
|
||||||
|
)
|
||||||
|
|
||||||
|
user_service = providers.Factory(
|
||||||
|
UserService,
|
||||||
|
users_repository=users_repository,
|
||||||
|
)
|
41
examples/miniapps/fastapi-sqlalchemy/webapp/database.py
Normal file
41
examples/miniapps/fastapi-sqlalchemy/webapp/database.py
Normal file
|
@ -0,0 +1,41 @@
|
||||||
|
"""Database module."""
|
||||||
|
|
||||||
|
from contextlib import contextmanager, AbstractContextManager
|
||||||
|
from typing import Callable
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from sqlalchemy import create_engine, orm
|
||||||
|
from sqlalchemy.ext.declarative import declarative_base
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
Base = declarative_base()
|
||||||
|
|
||||||
|
|
||||||
|
class Database:
|
||||||
|
|
||||||
|
def __init__(self, db_url: str) -> None:
|
||||||
|
self._engine = create_engine(db_url, echo=True)
|
||||||
|
self._session_factory = orm.scoped_session(
|
||||||
|
orm.sessionmaker(
|
||||||
|
autocommit=False,
|
||||||
|
autoflush=False,
|
||||||
|
bind=self._engine,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
def create_database(self) -> None:
|
||||||
|
Base.metadata.create_all(self._engine)
|
||||||
|
|
||||||
|
@contextmanager
|
||||||
|
def session(self) -> Callable[..., AbstractContextManager[Session]]:
|
||||||
|
session: Session = self._session_factory()
|
||||||
|
try:
|
||||||
|
yield session
|
||||||
|
except Exception:
|
||||||
|
logger.exception('Session rollback because of exception')
|
||||||
|
session.rollback()
|
||||||
|
raise
|
||||||
|
finally:
|
||||||
|
session.close()
|
57
examples/miniapps/fastapi-sqlalchemy/webapp/endpoints.py
Normal file
57
examples/miniapps/fastapi-sqlalchemy/webapp/endpoints.py
Normal file
|
@ -0,0 +1,57 @@
|
||||||
|
"""Endpoints module."""
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Depends, Response, status
|
||||||
|
from dependency_injector.wiring import inject, Provide
|
||||||
|
|
||||||
|
from .containers import Container
|
||||||
|
from .services import UserService
|
||||||
|
from .repositories import NotFoundError
|
||||||
|
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/users')
|
||||||
|
@inject
|
||||||
|
def get_list(
|
||||||
|
user_service: UserService = Depends(Provide[Container.user_service]),
|
||||||
|
):
|
||||||
|
return user_service.get_users()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/users/{user_id}')
|
||||||
|
@inject
|
||||||
|
def get_by_id(
|
||||||
|
user_id: int,
|
||||||
|
response: Response,
|
||||||
|
user_service: UserService = Depends(Provide[Container.user_service]),
|
||||||
|
):
|
||||||
|
try:
|
||||||
|
return user_service.get_user_by_id(user_id)
|
||||||
|
except NotFoundError:
|
||||||
|
response.status_code = status.HTTP_404_NOT_FOUND
|
||||||
|
|
||||||
|
|
||||||
|
@router.post('/users', status_code=status.HTTP_201_CREATED)
|
||||||
|
@inject
|
||||||
|
def add(
|
||||||
|
user_service: UserService = Depends(Provide[Container.user_service]),
|
||||||
|
):
|
||||||
|
return user_service.create_user()
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete('/users/{user_id}', status_code=status.HTTP_204_NO_CONTENT)
|
||||||
|
@inject
|
||||||
|
def remove(
|
||||||
|
user_id: int,
|
||||||
|
response: Response,
|
||||||
|
user_service: UserService = Depends(Provide[Container.user_service]),
|
||||||
|
):
|
||||||
|
try:
|
||||||
|
user_service.delete_user_by_id(user_id)
|
||||||
|
except NotFoundError:
|
||||||
|
response.status_code = status.HTTP_404_NOT_FOUND
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/status')
|
||||||
|
def get_status():
|
||||||
|
return {'status': 'OK'}
|
21
examples/miniapps/fastapi-sqlalchemy/webapp/models.py
Normal file
21
examples/miniapps/fastapi-sqlalchemy/webapp/models.py
Normal file
|
@ -0,0 +1,21 @@
|
||||||
|
"""Models module."""
|
||||||
|
|
||||||
|
from sqlalchemy import Column, String, Boolean, Integer
|
||||||
|
|
||||||
|
from .database import Base
|
||||||
|
|
||||||
|
|
||||||
|
class User(Base):
|
||||||
|
|
||||||
|
__tablename__ = 'users'
|
||||||
|
|
||||||
|
id = Column(Integer, primary_key=True)
|
||||||
|
email = Column(String, unique=True)
|
||||||
|
hashed_password = Column(String)
|
||||||
|
is_active = Column(Boolean, default=True)
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f'<User(id="{self.id}", ' \
|
||||||
|
f'email="{self.email}", ' \
|
||||||
|
f'hashed_password="{self.hashed_password}", ' \
|
||||||
|
f'is_active="{self.is_active}")>'
|
54
examples/miniapps/fastapi-sqlalchemy/webapp/repositories.py
Normal file
54
examples/miniapps/fastapi-sqlalchemy/webapp/repositories.py
Normal file
|
@ -0,0 +1,54 @@
|
||||||
|
"""Repositories module."""
|
||||||
|
|
||||||
|
from contextlib import AbstractContextManager
|
||||||
|
from typing import Callable, Iterator
|
||||||
|
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
|
from .models import User
|
||||||
|
|
||||||
|
|
||||||
|
class UserRepository:
|
||||||
|
|
||||||
|
def __init__(self, session_factory: Callable[..., AbstractContextManager[Session]]) -> None:
|
||||||
|
self.session_factory = session_factory
|
||||||
|
|
||||||
|
def get_all(self) -> Iterator[User]:
|
||||||
|
with self.session_factory() as session:
|
||||||
|
return session.query(User).all()
|
||||||
|
|
||||||
|
def get_by_id(self, user_id: int) -> User:
|
||||||
|
with self.session_factory() as session:
|
||||||
|
user = session.query(User).filter(User.id == user_id).first()
|
||||||
|
if not user:
|
||||||
|
raise UserNotFoundError(user_id)
|
||||||
|
return user
|
||||||
|
|
||||||
|
def add(self, email: str, password: str, is_active: bool = True) -> User:
|
||||||
|
with self.session_factory() as session:
|
||||||
|
user = User(email=email, hashed_password=password, is_active=is_active)
|
||||||
|
session.add(user)
|
||||||
|
session.commit()
|
||||||
|
session.refresh(user)
|
||||||
|
return user
|
||||||
|
|
||||||
|
def delete_by_id(self, user_id: int) -> None:
|
||||||
|
with self.session_factory() as session:
|
||||||
|
entity: User = session.query(User).filter(User.id == user_id).first()
|
||||||
|
if not entity:
|
||||||
|
raise UserNotFoundError(user_id)
|
||||||
|
session.delete(entity)
|
||||||
|
session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
class NotFoundError(Exception):
|
||||||
|
|
||||||
|
entity_name: str
|
||||||
|
|
||||||
|
def __init__(self, entity_id):
|
||||||
|
super().__init__(f'{self.entity_name} not found, id: {entity_id}')
|
||||||
|
|
||||||
|
|
||||||
|
class UserNotFoundError(NotFoundError):
|
||||||
|
|
||||||
|
entity_name: str = 'User'
|
26
examples/miniapps/fastapi-sqlalchemy/webapp/services.py
Normal file
26
examples/miniapps/fastapi-sqlalchemy/webapp/services.py
Normal file
|
@ -0,0 +1,26 @@
|
||||||
|
"""Services module."""
|
||||||
|
|
||||||
|
from uuid import uuid4
|
||||||
|
from typing import Iterator
|
||||||
|
|
||||||
|
from .repositories import UserRepository
|
||||||
|
from .models import User
|
||||||
|
|
||||||
|
|
||||||
|
class UserService:
|
||||||
|
|
||||||
|
def __init__(self, users_repository: UserRepository) -> None:
|
||||||
|
self._repository: UserRepository = users_repository
|
||||||
|
|
||||||
|
def get_users(self) -> Iterator[User]:
|
||||||
|
return self._repository.get_all()
|
||||||
|
|
||||||
|
def get_user_by_id(self, user_id: int) -> User:
|
||||||
|
return self._repository.get_by_id(user_id)
|
||||||
|
|
||||||
|
def create_user(self) -> User:
|
||||||
|
uid = uuid4()
|
||||||
|
return self._repository.add(email=f'{uid}@email.com', password='pwd')
|
||||||
|
|
||||||
|
def delete_user_by_id(self, user_id: int) -> None:
|
||||||
|
return self._repository.delete_by_id(user_id)
|
Loading…
Reference in New Issue
Block a user