Merge pull request #8 from dmsnback/refactoring

refactoring

Author: dmsnback

Dockerfile

@@ -3,6 +3,6 @@ LABEL maintainer="Dmitry Titenkov <lt200711@yandex.ru>"
 LABEL version="1.0"
 LABEL description="Satire Pulp parser"
 WORKDIR /app
-COPY requirements.txt /app
+COPY requirements.txt .
 RUN pip3 install -r /app/requirements.txt --no-cache-dir
 COPY . .

bot/__init__.py

@@ -1,6 +1,6 @@
 import logging
 
-from models import LastSentNews, News
+from db.models import LastSentNews, News
 from sqlalchemy import select
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.ext.asyncio import AsyncSession

bot_storage.py bot/bot_storage.pybot_storage.py renamed to bot/bot_storage.py

@@ -1,13 +1,14 @@
 import logging
 
-from bot_storage import (
+from bot.bot_storage import (
     get_all_users,
     get_last_sent_id,
     get_news_after_id,
     save_last_sent_news_id,
 )
+from bot.sender import send_news
 from config import setup_logger
-from db_async import AsyncSessionLocal
+from db.db_async import AsyncSessionLocal
 from dotenv import load_dotenv
 from telegram import (
     BotCommand,
@@ -23,50 +24,6 @@
 logger = logging.getLogger(__name__)
 
 
-MAX_CAPTION_LENGTH = 1024
-
-
-def format_message(title, text):
-    message = f"*{title}*\n\n{text}\n"
-    if len(message) > MAX_CAPTION_LENGTH:
-        message = f"*{title}*\n\n{text[:MAX_CAPTION_LENGTH]} ...✂️\n"
-    return message
-
-
-async def send_news(
-    chat_id: int, context: ContextTypes.DEFAULT_TYPE, title, image, text, url
-):
-    message = format_message(title, text)
-    keyboard = [
-        [InlineKeyboardButton("Читать полную версию на сайте", url=url)]
-    ]
-    reply_markup = InlineKeyboardMarkup(keyboard)
-    if image:
-        try:
-            await context.bot.send_photo(
-                chat_id=chat_id,
-                photo=image,
-                caption=message,
-                parse_mode="Markdown",
-                reply_markup=reply_markup,
-            )
-            logger.info("Новость отправлена с картинкой")
-            return
-        except Exception as e:
-            logger.error(
-                f"Не удалось отправить фото по ссылке, ошибка: {e}",
-            )
-    try:
-        await context.bot.send_message(
-            chat_id, message, parse_mode="Markdown", reply_markup=reply_markup
-        )
-        logger.info(f"Новость '{title[:25]}' отправлена без картинки")
-    except Exception as e:
-        logger.error(
-            f"Не удалось отправить сообщение с новостью '{title[:25]}', ошибка: {e}"
-        )
-
-
 async def auto_send_news(context: ContextTypes.DEFAULT_TYPE):
     async with AsyncSessionLocal() as session:
         users = await get_all_users(session)

bot.py bot/handlers.pybot.py renamed to bot/handlers.py

@@ -0,0 +1,55 @@
+import logging
+
+from config import setup_logger
+from dotenv import load_dotenv
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup
+from telegram.ext import ContextTypes
+
+load_dotenv()
+
+setup_logger()
+logger = logging.getLogger(__name__)
+
+
+MAX_CAPTION_LENGTH = 1024
+
+
+def format_message(title, text):
+    message = f"*{title}*\n\n{text}\n"
+    if len(message) > MAX_CAPTION_LENGTH:
+        message = f"*{title}*\n\n{text[:MAX_CAPTION_LENGTH]} ...✂️\n"
+    return message
+
+
+async def send_news(
+    chat_id: int, context: ContextTypes.DEFAULT_TYPE, title, image, text, url
+):
+    message = format_message(title, text)
+    keyboard = [
+        [InlineKeyboardButton("Читать полную версию на сайте", url=url)]
+    ]
+    reply_markup = InlineKeyboardMarkup(keyboard)
+    if image:
+        try:
+            await context.bot.send_photo(
+                chat_id=chat_id,
+                photo=image,
+                caption=message,
+                parse_mode="Markdown",
+                reply_markup=reply_markup,
+            )
+            logger.info("Новость отправлена с картинкой")
+            return
+        except Exception as e:
+            logger.error(
+                f"Не удалось отправить фото по ссылке, ошибка: {e}",
+            )
+    try:
+        await context.bot.send_message(
+            chat_id, message, parse_mode="Markdown", reply_markup=reply_markup
+        )
+        logger.info(f"Новость '{title[:25]}' отправлена без картинки")
+    except Exception as e:
+        logger.error(
+            f"Не удалось отправить сообщение с новостью '{title[:25]}', ошибка: {e}"
+        )

bot/sender.py

@@ -6,7 +6,10 @@
 load_dotenv()
 
 
-engine = create_async_engine(os.getenv("DATABASE_URL_ASYNC"), echo=False)
+DATABASE_URL_ASYNC = os.getenv("DATABASE_URL_ASYNC")
+
+
+engine = create_async_engine(DATABASE_URL_ASYNC, echo=False)
 
 
 AsyncSessionLocal = async_sessionmaker(engine, expire_on_commit=False)

db/__init__.py

@@ -7,6 +7,9 @@
 load_dotenv()
 
 
-engine = create_engine(os.getenv("DATABASE_URL_SYNC"))
+DATABASE_URL_SYNC = os.getenv("DATABASE_URL_SYNC")
+
+
+engine = create_engine(DATABASE_URL_SYNC)
 
 SessionLocal = sessionmaker(engine)

db_async.py db/db_async.pydb_async.py renamed to db/db_async.py

@@ -2,9 +2,9 @@
 import logging
 
 from config import setup_logger
-from db_async import engine
+from db.db_async import engine
+from db.models import Base, LastSentNews, News  # noqa
 from dotenv import load_dotenv
-from models import Base, LastSentNews, News  # noqa
 
 load_dotenv()