diff --git a/db/__init__.py b/db/__init__.py index 618ec9e..26d70fd 100644 --- a/db/__init__.py +++ b/db/__init__.py @@ -5,6 +5,24 @@ from db.schemas import UserCreate, UserInDb, DutyCreate, DutyInDb, DutyWithUser from db.session import get_engine, get_session_factory, get_session from db.repository import get_or_create_user, get_duties, insert_duty +__all__ = [ + "Base", + "User", + "Duty", + "UserCreate", + "UserInDb", + "DutyCreate", + "DutyInDb", + "DutyWithUser", + "get_engine", + "get_session_factory", + "get_session", + "get_or_create_user", + "get_duties", + "insert_duty", + "init_db", +] + def init_db(database_url: str) -> None: """Create tables from metadata (Alembic migrations handle schema in production).""" diff --git a/handlers/commands.py b/handlers/commands.py index be687dd..b7dbd86 100644 --- a/handlers/commands.py +++ b/handlers/commands.py @@ -48,9 +48,7 @@ async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: async def help_cmd(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: if update.message: await update.message.reply_text( - "Доступные команды:\n" - "/start — Начать\n" - "/help — Показать эту справку" + "Доступные команды:\n/start — Начать\n/help — Показать эту справку" ) diff --git a/handlers/errors.py b/handlers/errors.py index e70d98d..1307424 100644 --- a/handlers/errors.py +++ b/handlers/errors.py @@ -8,7 +8,9 @@ from telegram.ext import ContextTypes logger = logging.getLogger(__name__) -async def error_handler(update: Update | None, context: ContextTypes.DEFAULT_TYPE) -> None: +async def error_handler( + update: Update | None, context: ContextTypes.DEFAULT_TYPE +) -> None: logger.exception("Exception while handling an update") if isinstance(update, Update) and update.effective_message: await update.effective_message.reply_text("Произошла ошибка. Попробуйте позже.") diff --git a/api/test_app.py b/tests/test_app.py similarity index 98% rename from api/test_app.py rename to tests/test_app.py index 14d4caf..7bc2a22 100644 --- a/api/test_app.py +++ b/tests/test_app.py @@ -8,7 +8,7 @@ from fastapi.testclient import TestClient import config from api.app import app -from api.test_telegram_auth import _make_init_data +from tests.test_telegram_auth import _make_init_data @pytest.fixture diff --git a/api/test_telegram_auth.py b/tests/test_telegram_auth.py similarity index 100% rename from api/test_telegram_auth.py rename to tests/test_telegram_auth.py