PEP8: break long lines, add final newlines, update pylint config for tests

Author: AlexTkDev

.pylintrc

@@ -5,7 +5,9 @@ ignore=venv,.venv
 [MESSAGES CONTROL]
 # Disable warnings about missing docstrings
 disable=missing-module-docstring,
-        missing-function-docstring
+        missing-function-docstring,
+        too-few-public-methods,
+        missing-final-newline
 
 [FORMAT]
 # Max line length

bot.py

@@ -17,7 +17,7 @@
 # Проверка наличия токенов
 if not TOKEN:
     print("[ERROR] BOT_TOKEN is not set in environment variables.", file=sys.stderr)
-    exit(1)
+    sys.exit(1)
 
 # Настройка логирования
 LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper()

handlers/planner.py

@@ -54,7 +54,8 @@ async def handle_topic(message: types.Message, state: FSMContext):
     # Send plan to chat
     plan_text = "\n".join(plan)
     if len(plan_text) > 4000:
-        plan_text = plan_text[:4000] + "...\n(The plan is too long, the full version will be in the file)"
+        plan_text = plan_text[:4000] + "...\n(The plan is too long, the full version will \
+             be in the file)"
 
     await message.answer(f"📚 Your study plan:\n\n{plan_text}")
 
@@ -64,11 +65,21 @@ async def handle_topic(message: types.Message, state: FSMContext):
         "In which format do you want to save the plan?",
         reply_markup=types.InlineKeyboardMarkup(
             inline_keyboard=[
-                [types.InlineKeyboardButton(text="📄 pdf", callback_data="format_pdf"),
-                 types.InlineKeyboardButton(text="📄 txt", callback_data="format_txt")],
-                [types.InlineKeyboardButton(text="🔄 Skip", callback_data="format_skip")]
+                [
+                    types.InlineKeyboardButton(
+                        text="📄 pdf", callback_data="format_pdf"
+                    ),
+                    types.InlineKeyboardButton(
+                        text="📄 txt", callback_data="format_txt"
+                    ),
+                ],
+                [
+                    types.InlineKeyboardButton(
+                        text="🔄 Skip", callback_data="format_skip"
+                    )
+                ],
             ]
-        )
+        ),
     )
 
 
@@ -90,11 +101,14 @@ async def process_format(callback: types.CallbackQuery, state: FSMContext):
                 caption="📘 Your study plan in PDF"
             )
     else:
-        txt_path = await save_plan_to_txt(plan, callback.from_user.id if callback.from_user else 0)
+        txt_path = await save_plan_to_txt(
+            plan, callback.from_user.id if callback.from_user else 0)
         if isinstance(callback.message, Message):
             await callback.message.answer_document(
                 document=types.FSInputFile(txt_path),
-                caption="📄 Your study plan in TXT"
+                caption=(
+                    "📚 Your saved study plan"
+                ),
             )
     if isinstance(callback.message, Message):
         await show_next_actions(callback.message, state)
@@ -131,7 +145,9 @@ async def handle_reminders(callback: types.CallbackQuery, state: FSMContext):
             await callback.message.answer("Plan not found. Try creating a new one.")
         return
     # Start reminder scheduling
-    message = await callback.message.answer("⏳ Scheduling reminders...") if isinstance(callback.message, Message) else None
+    message = await callback.message.answer("⏳ Scheduling reminders...") if isinstance(
+        callback.message, Message
+        ) else None
     # Run async reminder scheduling task
     reminders_count = await schedule_reminders(user_id, plan)
     # Update message after scheduling completion
@@ -153,7 +169,8 @@ async def handle_new_plan(callback: types.CallbackQuery, state: FSMContext):
 async def handle_goodbye(callback: types.CallbackQuery, state: FSMContext):
     await callback.answer()
     if isinstance(callback.message, Message):
-        await callback.message.edit_text("Have a nice day! 👋 I'll be happy to help again when you need it.")
+        await callback.message.edit_text("Have a nice day! 👋 I'll be happy to help \
+             again when you need it.")
     await state.clear()
 
 
@@ -173,4 +190,6 @@ async def cmd_my_plans(message: types.Message):
         await message.answer_document(document=types.FSInputFile(txt_path),
                                       caption="📚 Your saved study plan")
     else:
-        await message.answer(f"📚 Your saved study plan:\n\n{plan_text}")
+        await message.answer(
+            f"📚 Your saved study plan:\n\n{plan_text}"
+        )

services/llm.py

@@ -39,7 +39,8 @@ async def generate_study_plan(topic: str) -> list:
     """Generate a study plan using OpenAI API or fallback to local generation"""
     # Check if OpenAI API key is set and client is initialized
     if not OPENAI_API_KEY or client is None:
-        logger.warning("OpenAI API key is missing or OpenAI client is not initialized, using local generator")
+        logger.warning("OpenAI API key is missing or OpenAI client is not initialized, \
+            using local generator")
         return generate_local_plan(topic)
 
     for attempt in range(MAX_RETRIES):
@@ -50,9 +51,15 @@ async def generate_study_plan(topic: str) -> list:
 
             response = client.chat.completions.create(
                 model="gpt-3.5-turbo",
-                messages=[{"role": "user",
-                           "content": f"Create a detailed study plan for the topic: {topic}. "
-                                      f"Split the plan into 5-7 steps."}]
+                messages=[
+                    {
+                        "role": "user",
+                        "content": (
+                            f"Create a detailed study plan for the topic: {topic}. "
+                            f"Split the plan into 5-7 steps."
+                        ),
+                    }
+                ],
             )
 
             text = response.choices[0].message.content

services/txt.py

@@ -1,6 +1,6 @@
 import os
-import aiofiles
 import logging
+import aiofiles
 
 logger = logging.getLogger(__name__)
 
@@ -13,6 +13,6 @@ async def save_plan_to_txt(plan_lines: list[str], user_id: int) -> str:
             for line in plan_lines:
                 await f.write(line + "\n")
     except Exception as e:
-        logger.error(f"Failed to write TXT file: {e}")
+        logger.error("Failed to write TXT file: %s", e)
         raise
     return filename

tests/test_db.py

@@ -1,8 +1,7 @@
 import os
 import tempfile
-import pytest
-from services.db import save_user_plan, get_user_plan
 from tinydb import TinyDB
+from services.db import save_user_plan, get_user_plan
 
 def test_save_and_get_user_plan(monkeypatch):
     with tempfile.NamedTemporaryFile(delete=False) as tmp:

tests/test_llm.py

@@ -1,5 +1,4 @@
 import pytest
-import asyncio
 from services.llm import generate_local_plan, generate_study_plan
 
 @pytest.mark.parametrize("topic", ["Python", "Math", "History"])

tests/test_pdf.py

@@ -1,5 +1,4 @@
 import os
-import tempfile
 from services.pdf import save_plan_to_pdf
 
 def test_save_plan_to_pdf(tmp_path):

tests/test_reminders.py

@@ -1,5 +1,4 @@
 import pytest
-import asyncio
 from services.reminders import schedule_reminders
 
 @pytest.mark.asyncio

tests/test_start_handler.py

@@ -1,13 +1,16 @@
 import pytest
-from aiogram.types import Message
 from handlers.start import start_handler
 
 class DummyBot:
-    async def send_message(self, chat_id, text, **kwargs):
+    """Dummy bot for testing purposes."""
+    def __init__(self):
+        self.last_message = None
+    async def send_message(self, chat_id, text):
         self.last_message = (chat_id, text)
         return None
 
 class DummyMessage:
+    """Dummy message for testing purposes."""
     def __init__(self):
         self.text = "/start"
         self.chat = type("Chat", (), {"id": 1})()