Enhance private handlers structure and add database support

- Introduced a new `PrivateHandlers` class to encapsulate private message handling logic, improving organization and maintainability.
- Added new dependencies in `requirements.txt` for database support with `aiosqlite`.
- Updated the private handlers to utilize modular components for better separation of concerns and easier testing.
- Implemented error handling and logging for improved robustness in message processing.

tests/test_async_db.py

@@ -0,0 +1,174 @@
+import pytest
+import asyncio
+import os
+import tempfile
+from database.async_db import AsyncBotDB
+
+
+@pytest.fixture
+async def temp_db():
+    """Создает временную базу данных для тестирования."""
+    with tempfile.NamedTemporaryFile(suffix='.db', delete=False) as tmp:
+        db_path = tmp.name
+    
+    db = AsyncBotDB(db_path)
+    yield db
+    
+    # Очистка
+    try:
+        os.unlink(db_path)
+    except:
+        pass
+
+
+@pytest.fixture(scope="function")
+def event_loop():
+    """Создает новый event loop для каждого теста."""
+    loop = asyncio.new_event_loop()
+    yield loop
+    loop.close()
+
+
+@pytest.mark.asyncio
+async def test_create_tables(temp_db):
+    """Тест создания таблиц."""
+    await temp_db.create_tables()
+    # Если не возникло исключение, значит таблицы созданы успешно
+    assert True
+
+
+@pytest.mark.asyncio
+async def test_add_and_get_user(temp_db):
+    """Тест добавления и получения пользователя."""
+    await temp_db.create_tables()
+    
+    # Добавляем пользователя
+    user_id = 12345
+    first_name = "Test"
+    full_name = "Test User"
+    username = "testuser"
+    
+    await temp_db.add_new_user(user_id, first_name, full_name, username)
+    
+    # Проверяем существование
+    exists = await temp_db.user_exists(user_id)
+    assert exists is True
+    
+    # Получаем информацию
+    user_info = await temp_db.get_user_info(user_id)
+    assert user_info is not None
+    assert user_info['username'] == username
+    assert user_info['full_name'] == full_name
+
+
+@pytest.mark.asyncio
+async def test_blacklist_operations(temp_db):
+    """Тест операций с черным списком."""
+    await temp_db.create_tables()
+    
+    user_id = 12345
+    user_name = "Test User"
+    message = "Test ban"
+    date_to_unban = "01-01-2025"
+    
+    # Добавляем в черный список
+    await temp_db.add_to_blacklist(user_id, user_name, message, date_to_unban)
+    
+    # Проверяем наличие
+    is_banned = await temp_db.check_blacklist(user_id)
+    assert is_banned is True
+    
+    # Получаем список
+    banned_users = await temp_db.get_blacklist_users()
+    assert len(banned_users) == 1
+    assert banned_users[0][1] == user_id  # user_id
+    
+    # Удаляем из черного списка
+    removed = await temp_db.remove_from_blacklist(user_id)
+    assert removed is True
+    
+    # Проверяем удаление
+    is_banned = await temp_db.check_blacklist(user_id)
+    assert is_banned is False
+
+
+@pytest.mark.asyncio
+async def test_admin_operations(temp_db):
+    """Тест операций с администраторами."""
+    await temp_db.create_tables()
+    
+    user_id = 12345
+    role = "admin"
+    
+    # Добавляем администратора
+    await temp_db.add_admin(user_id, role)
+    
+    # Проверяем права
+    is_admin = await temp_db.is_admin(user_id)
+    assert is_admin is True
+    
+    # Удаляем администратора
+    await temp_db.remove_admin(user_id)
+    
+    # Проверяем удаление
+    is_admin = await temp_db.is_admin(user_id)
+    assert is_admin is False
+
+
+@pytest.mark.asyncio
+async def test_audio_operations(temp_db):
+    """Тест операций с аудио."""
+    await temp_db.create_tables()
+    
+    user_id = 12345
+    file_name = "test_audio.mp3"
+    file_id = "test_file_id"
+    
+    # Добавляем аудио запись
+    await temp_db.add_audio_record(file_name, user_id, file_id)
+    
+    # Получаем file_id
+    retrieved_file_id = await temp_db.get_audio_file_id(user_id)
+    assert retrieved_file_id == file_id
+    
+    # Получаем имя файла
+    retrieved_file_name = await temp_db.get_audio_file_name(user_id)
+    assert retrieved_file_name == file_name
+
+
+@pytest.mark.asyncio
+async def test_post_operations(temp_db):
+    """Тест операций с постами."""
+    await temp_db.create_tables()
+    
+    message_id = 12345
+    text = "Test post text"
+    author_id = 67890
+    
+    # Добавляем пост
+    await temp_db.add_post(message_id, text, author_id)
+    
+    # Обновляем helper сообщение
+    helper_message_id = 54321
+    await temp_db.update_helper_message(message_id, helper_message_id)
+    
+    # Получаем текст поста
+    retrieved_text = await temp_db.get_post_text(helper_message_id)
+    assert retrieved_text == text
+    
+    # Получаем ID автора
+    retrieved_author_id = await temp_db.get_author_id_by_helper_message(helper_message_id)
+    assert retrieved_author_id == author_id
+
+
+@pytest.mark.asyncio
+async def test_error_handling(temp_db):
+    """Тест обработки ошибок."""
+    # Пытаемся получить пользователя без создания таблиц
+    with pytest.raises(Exception):
+        await temp_db.user_exists(12345)
+
+
+if __name__ == "__main__":
+    # Запуск тестов
+    pytest.main([__file__, "-v"])