5e01cc6d8f
Блоки обработки articles и posts в !nw были почти идентичны. Вынесено в _format_feed_section() с параметризацией лейблов, сообщений об ошибках и заголовков RSS-источников. Файлы: commands/news.py
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
import asyncio
|
||
import logging
|
||
from typing import Optional
|
||
|
||
from discord.ext import commands
|
||
from utils.news import (
|
||
fetch_rss,
|
||
format_articles,
|
||
RSS_URL_ARTICLES,
|
||
RSS_URL_POSTS,
|
||
truncate_message,
|
||
)
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def _format_feed_section(
|
||
data: Optional[list[dict]],
|
||
author: str,
|
||
label_warning: str,
|
||
label_info: str,
|
||
msg_error: str,
|
||
msg_empty: str,
|
||
title: str,
|
||
link: str,
|
||
) -> str:
|
||
"""Форматировать один RSS-раздел (статьи или посты)."""
|
||
if data is None:
|
||
logger.warning(
|
||
"%s: !nw — не удалось получить %s (API вернул None)", author, label_warning
|
||
)
|
||
return msg_error
|
||
elif data:
|
||
return "\n".join(format_articles(data, title, link))
|
||
else:
|
||
logger.info("%s: !nw — %s нет в RSS", author, label_info)
|
||
return msg_empty
|
||
|
||
|
||
class News(commands.Cog):
|
||
"""Команда !news — свежие статьи и новости по AI с Habr"""
|
||
|
||
@commands.command(name="nw")
|
||
async def nw(self, ctx: commands.Context) -> None:
|
||
"""Топ-5 свежих статей и новостей по AI с Habr"""
|
||
articles, posts = await asyncio.gather(
|
||
fetch_rss(RSS_URL_ARTICLES),
|
||
fetch_rss(RSS_URL_POSTS),
|
||
)
|
||
|
||
parts: list[str] = [
|
||
_format_feed_section(
|
||
articles,
|
||
str(ctx.author),
|
||
"статьи",
|
||
"статей",
|
||
"Не удалось получить статьи.",
|
||
"Статей пока нет.",
|
||
"Лучшие статьи за сутки / Искусственный интеллект / Хабr",
|
||
"https://habr.com/ru/hubs/artificial_intelligence/articles/top/daily/",
|
||
),
|
||
_format_feed_section(
|
||
posts,
|
||
str(ctx.author),
|
||
"посты",
|
||
"постов",
|
||
"Не удалось получить новости.",
|
||
"Новостей пока нет.",
|
||
"Лучшие новости за сутки / Искусственный интеллект / Хабr",
|
||
"https://habr.com/ru/hubs/artificial_intelligence/news/top/daily/",
|
||
),
|
||
]
|
||
|
||
message = truncate_message("\n".join(parts))
|
||
await ctx.send(message)
|
||
logger.info(
|
||
"%s: !nw выполнена (статей: %d, постов: %d)",
|
||
ctx.author,
|
||
len(articles) if articles else 0,
|
||
len(posts) if posts else 0,
|
||
)
|