Refactor imports across multiple files to improve code organization and readability.

tests/conftest.py

@@ -13,10 +13,10 @@ if str(_project_root) not in sys.path:
 import pytest
 from aiogram.fsm.context import FSMContext
 from aiogram.types import Chat, Message, User
-from database.async_db import AsyncBotDB
 
 # Импортируем моки в самом начале
 import tests.mocks
+from database.async_db import AsyncBotDB
 
 # Настройка pytest-asyncio
 pytest_plugins = ("pytest_asyncio",)

tests/conftest_message_repository.py

@@ -3,6 +3,7 @@ import tempfile
 from datetime import datetime
 
 import pytest
+
 from database.models import UserMessage
 from database.repositories.message_repository import MessageRepository
 

tests/conftest_post_repository.py

@@ -5,6 +5,7 @@ from datetime import datetime
 from unittest.mock import AsyncMock, Mock
 
 import pytest
+
 from database.models import MessageContentLink, PostContent, TelegramPost
 from database.repositories.post_repository import PostRepository
 

tests/test_admin_dependencies.py

@@ -5,6 +5,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.handlers.admin.dependencies import (
     AdminAccessMiddleware,
     get_bot_db,

tests/test_admin_handlers.py

@@ -7,6 +7,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 import pytest
 from aiogram import types
 from aiogram.fsm.context import FSMContext
+
 from helper_bot.handlers.admin.admin_handlers import (
     admin_panel,
     cancel_ban_process,

tests/test_admin_repository.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
+
 from database.models import Admin
 from database.repositories.admin_repository import AdminRepository
 

tests/test_admin_utils.py

@@ -5,6 +5,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.handlers.admin.exceptions import AdminError
 from helper_bot.handlers.admin.utils import (
     escape_html,

tests/test_album_middleware.py

@@ -6,6 +6,7 @@ import asyncio
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.middlewares.album_middleware import AlbumGetter, AlbumMiddleware
 
 

tests/test_async_db.py

@@ -1,6 +1,7 @@
 from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
+
 from database.async_db import AsyncBotDB
 
 

tests/test_audio_file_service.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from unittest.mock import AsyncMock, MagicMock, Mock, mock_open, patch
 
 import pytest
+
 from helper_bot.handlers.voice.exceptions import DatabaseError, FileOperationError
 from helper_bot.handlers.voice.services import AudioFileService
 

tests/test_audio_repository.py

@@ -3,6 +3,7 @@ from datetime import datetime, timezone
 from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
+
 from database.models import AudioListenRecord, AudioMessage, AudioModerate
 from database.repositories.audio_repository import AudioRepository
 

tests/test_audio_repository_schema.py

@@ -3,6 +3,7 @@ from datetime import datetime, timezone
 from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
+
 from database.repositories.audio_repository import AudioRepository
 
 

tests/test_auto_moderation_service.py

@@ -3,6 +3,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.handlers.private.services import AutoModerationService, BotSettings
 
 
@@ -108,9 +109,7 @@ class TestAutoModerationService:
         assert result == "manual"
 
     @pytest.mark.asyncio
-    async def test_check_auto_action_publish_at_exact_threshold(
-        self, service, mock_db
-    ):
+    async def test_check_auto_action_publish_at_exact_threshold(self, service, mock_db):
         """Тест: возвращает publish когда score равен порогу."""
         mock_db.get_auto_moderation_settings.return_value = {
             "auto_publish_enabled": True,
@@ -124,9 +123,7 @@ class TestAutoModerationService:
         assert result == "publish"
 
     @pytest.mark.asyncio
-    async def test_check_auto_action_decline_at_exact_threshold(
-        self, service, mock_db
-    ):
+    async def test_check_auto_action_decline_at_exact_threshold(self, service, mock_db):
         """Тест: возвращает decline когда score равен порогу."""
         mock_db.get_auto_moderation_settings.return_value = {
             "auto_publish_enabled": False,

tests/test_auto_unban_integration.py

@@ -4,6 +4,7 @@ from datetime import datetime, timedelta, timezone
 from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
+
 from helper_bot.utils.auto_unban_scheduler import AutoUnbanScheduler
 
 

tests/test_auto_unban_scheduler.py

@@ -3,6 +3,7 @@ from datetime import datetime, timedelta, timezone
 from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
+
 from helper_bot.utils.auto_unban_scheduler import (
     AutoUnbanScheduler,
     get_auto_unban_scheduler,

tests/test_blacklist_history_repository.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
+
 from database.models import BlacklistHistoryRecord
 from database.repositories.blacklist_history_repository import (
     BlacklistHistoryRepository,

tests/test_blacklist_middleware.py

@@ -6,6 +6,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 from aiogram.types import CallbackQuery, Message
+
 from helper_bot.middlewares.blacklist_middleware import BlacklistMiddleware
 
 

tests/test_blacklist_repository.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
+
 from database.models import BlacklistUser
 from database.repositories.blacklist_repository import BlacklistRepository
 

tests/test_bot_settings_repository.py

@@ -3,6 +3,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from database.repositories.bot_settings_repository import BotSettingsRepository
 
 
@@ -115,11 +116,14 @@ class TestBotSettingsRepository:
     @pytest.mark.asyncio
     async def test_get_auto_moderation_settings(self, repository):
         """Тест получения всех настроек авто-модерации."""
-        with patch.object(
-            repository, "get_bool_setting", new_callable=AsyncMock
-        ) as mock_bool, patch.object(
-            repository, "get_float_setting", new_callable=AsyncMock
-        ) as mock_float:
+        with (
+            patch.object(
+                repository, "get_bool_setting", new_callable=AsyncMock
+            ) as mock_bool,
+            patch.object(
+                repository, "get_float_setting", new_callable=AsyncMock
+            ) as mock_float,
+        ):
             mock_bool.side_effect = [True, False]
             mock_float.side_effect = [0.8, 0.4]
 
@@ -133,11 +137,14 @@ class TestBotSettingsRepository:
     @pytest.mark.asyncio
     async def test_toggle_auto_publish(self, repository):
         """Тест переключения авто-публикации."""
-        with patch.object(
-            repository, "get_bool_setting", new_callable=AsyncMock
-        ) as mock_get, patch.object(
-            repository, "set_bool_setting", new_callable=AsyncMock
-        ) as mock_set:
+        with (
+            patch.object(
+                repository, "get_bool_setting", new_callable=AsyncMock
+            ) as mock_get,
+            patch.object(
+                repository, "set_bool_setting", new_callable=AsyncMock
+            ) as mock_set,
+        ):
             mock_get.return_value = False
 
             result = await repository.toggle_auto_publish()
@@ -148,11 +155,14 @@ class TestBotSettingsRepository:
     @pytest.mark.asyncio
     async def test_toggle_auto_decline(self, repository):
         """Тест переключения авто-отклонения."""
-        with patch.object(
-            repository, "get_bool_setting", new_callable=AsyncMock
-        ) as mock_get, patch.object(
-            repository, "set_bool_setting", new_callable=AsyncMock
-        ) as mock_set:
+        with (
+            patch.object(
+                repository, "get_bool_setting", new_callable=AsyncMock
+            ) as mock_get,
+            patch.object(
+                repository, "set_bool_setting", new_callable=AsyncMock
+            ) as mock_set,
+        ):
             mock_get.return_value = True
 
             result = await repository.toggle_auto_decline()

tests/test_callback_dependency_factory.py

@@ -5,6 +5,7 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
+
 from helper_bot.handlers.callback.dependency_factory import (
     get_ban_service,
     get_post_publish_service,

tests/test_callback_handlers.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
+
 from helper_bot.handlers.callback.callback_handlers import (
     change_page,
     delete_voice_message,

tests/test_callback_services.py

@@ -6,6 +6,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 from aiogram.types import CallbackQuery, Message
+
 from helper_bot.handlers.callback.constants import CONTENT_TYPE_MEDIA_GROUP
 from helper_bot.handlers.callback.exceptions import (
     PostNotFoundError,

tests/test_decorators.py

@@ -6,6 +6,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 from aiogram import types
+
 from helper_bot.handlers.group.decorators import error_handler as group_error_handler
 from helper_bot.handlers.private.decorators import (
     error_handler as private_error_handler,

tests/test_deepseek_service.py

@@ -5,6 +5,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.services.scoring.deepseek_service import DeepSeekService
 from helper_bot.services.scoring.exceptions import (
     DeepSeekAPIError,

tests/test_dependencies_middleware.py

@@ -5,6 +5,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.middlewares.dependencies_middleware import DependenciesMiddleware
 
 

tests/test_improved_media_processing.py

@@ -8,6 +8,7 @@ from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
 from aiogram import types
+
 from helper_bot.utils.helper_func import (
     add_in_db_media,
     add_in_db_media_mediagroup,

tests/test_keyboards_and_filters.py

@@ -7,6 +7,7 @@ from aiogram.types import (
     KeyboardButton,
     ReplyKeyboardMarkup,
 )
+
 from database.async_db import AsyncBotDB
 from helper_bot.filters.main import ChatTypeFilter
 from helper_bot.keyboards.keyboards import (

tests/test_main.py

@@ -6,6 +6,7 @@ import asyncio
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.main import start_bot, start_bot_with_retry
 
 

tests/test_message_repository.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from unittest.mock import AsyncMock, MagicMock
 
 import pytest
+
 from database.models import UserMessage
 from database.repositories.message_repository import MessageRepository
 

tests/test_message_repository_integration.py

@@ -4,6 +4,7 @@ import tempfile
 from datetime import datetime
 
 import pytest
+
 from database.models import UserMessage
 from database.repositories.message_repository import MessageRepository
 

tests/test_metrics_middleware.py

@@ -7,6 +7,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 from aiogram.types import Message
+
 from helper_bot.middlewares.metrics_middleware import (
     DatabaseMetricsMiddleware,
     ErrorMetricsMiddleware,

tests/test_post_repository.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from unittest.mock import AsyncMock, MagicMock
 
 import pytest
+
 from database.models import MessageContentLink, PostContent, TelegramPost
 from database.repositories.post_repository import PostRepository
 

tests/test_post_repository_integration.py

@@ -4,6 +4,7 @@ import tempfile
 from datetime import datetime
 
 import pytest
+
 from database.models import MessageContentLink, PostContent, TelegramPost
 from database.repositories.post_repository import PostRepository
 

tests/test_post_service.py

@@ -5,6 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
 from aiogram import types
+
 from database.models import TelegramPost, User
 from helper_bot.handlers.private.services import BotSettings, PostService
 

tests/test_rag_client.py

@@ -5,6 +5,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.services.scoring.exceptions import (
     InsufficientExamplesError,
     ScoringError,

tests/test_rate_limit_middleware.py

@@ -6,6 +6,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 from aiogram.types import CallbackQuery, Message, Update
+
 from helper_bot.middlewares.rate_limit_middleware import RateLimitMiddleware
 
 

tests/test_rate_limit_monitor.py

@@ -7,6 +7,7 @@ from collections import deque
 from unittest.mock import patch
 
 import pytest
+
 from helper_bot.utils.rate_limit_monitor import (
     RateLimitMonitor,
     RateLimitStats,

tests/test_rate_limiter.py

@@ -7,6 +7,7 @@ import time
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.config.rate_limit_config import RateLimitSettings, get_rate_limit_config
 from helper_bot.utils.rate_limit_monitor import (
     RateLimitMonitor,

tests/test_refactored_admin_handlers.py

@@ -3,6 +3,7 @@ from unittest.mock import AsyncMock, Mock, patch
 import pytest
 from aiogram import types
 from aiogram.fsm.context import FSMContext
+
 from helper_bot.handlers.admin.exceptions import (
     InvalidInputError,
     UserAlreadyBannedError,

tests/test_refactored_group_handlers.py

@@ -5,6 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, Mock
 import pytest
 from aiogram import types
 from aiogram.fsm.context import FSMContext
+
 from helper_bot.handlers.group.constants import ERROR_MESSAGES, FSM_STATES
 from helper_bot.handlers.group.exceptions import (
     NoReplyToMessageError,

tests/test_refactored_private_handlers.py

@@ -5,6 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, Mock, patch
 import pytest
 from aiogram import types
 from aiogram.fsm.context import FSMContext
+
 from helper_bot.handlers.private.constants import BUTTON_TEXTS, FSM_STATES
 from helper_bot.handlers.private.private_handlers import (
     PrivateHandlers,

tests/test_s3_storage.py

@@ -7,6 +7,7 @@ from pathlib import Path
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.utils.s3_storage import S3StorageService
 
 

tests/test_scoring_services.py

@@ -137,6 +137,7 @@ class TestVectorStore:
         """Создает VectorStore для тестов."""
         try:
             import numpy as np
+
             from helper_bot.services.scoring.vector_store import VectorStore
 
             return VectorStore(vector_dim=768, max_examples=100)

tests/test_server_prometheus.py

@@ -6,6 +6,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 from aiohttp import web
+
 from helper_bot.server_prometheus import (
     MetricsServer,
     start_metrics_server,

tests/test_text_middleware.py

@@ -5,6 +5,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+
 from helper_bot.middlewares.text_middleware import BulkTextMiddleware
 
 

tests/test_utils.py

@@ -2,8 +2,9 @@ import os
 from datetime import datetime
 from unittest.mock import AsyncMock, Mock, patch
 
-import helper_bot.utils.messages as messages  # Import for patching constants
 import pytest
+
+import helper_bot.utils.messages as messages  # Import for patching constants
 from database.async_db import AsyncBotDB
 from helper_bot.utils.base_dependency_factory import (
     BaseDependencyFactory,

tests/test_voice_bot_architecture.py

@@ -3,6 +3,7 @@ from pathlib import Path
 from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
+
 from helper_bot.handlers.voice.exceptions import AudioProcessingError, VoiceMessageError
 from helper_bot.handlers.voice.services import VoiceBotService
 from helper_bot.handlers.voice.utils import (

tests/test_voice_constants.py

@@ -1,4 +1,5 @@
 import pytest
+
 from helper_bot.handlers.voice.constants import (
     BTN_LISTEN,
     BTN_SPEAK,

tests/test_voice_exceptions.py

@@ -1,4 +1,5 @@
 import pytest
+
 from helper_bot.handlers.voice.exceptions import (
     AudioProcessingError,
     VoiceBotError,

tests/test_voice_handler.py

@@ -3,6 +3,7 @@ from unittest.mock import AsyncMock, MagicMock, Mock, patch
 import pytest
 from aiogram import types
 from aiogram.fsm.context import FSMContext
+
 from helper_bot.handlers.voice.constants import STATE_STANDUP_WRITE, STATE_START
 from helper_bot.handlers.voice.voice_handler import VoiceHandlers
 

tests/test_voice_services.py

@@ -3,6 +3,7 @@ from pathlib import Path
 from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
+
 from helper_bot.handlers.voice.exceptions import AudioProcessingError, VoiceMessageError
 from helper_bot.handlers.voice.services import VoiceBotService
 

tests/test_voice_utils.py

@@ -3,6 +3,7 @@ from unittest.mock import Mock, patch
 
 import pytest
 from aiogram import types
+
 from helper_bot.handlers.voice.utils import (
     format_time_ago,
     get_last_message_text,