128 lines
3.5 KiB
Python
128 lines
3.5 KiB
Python
import asyncio
|
|
import logging
|
|
import os
|
|
|
|
import requests
|
|
import telegram.constants
|
|
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
|
from telegram.ext import ApplicationBuilder, ContextTypes, CommandHandler
|
|
from telegram.helpers import escape_markdown
|
|
|
|
logging.basicConfig(
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.INFO
|
|
)
|
|
|
|
supported_sites = {
|
|
"百度": "baidu",
|
|
"少数派": "shaoshupai",
|
|
"微博": "weibo",
|
|
"知乎": "zhihu",
|
|
"36氪": "36kr",
|
|
"吾爱破解": "52pojie",
|
|
"哔哩哔哩": "bilibili",
|
|
"豆瓣": "douban",
|
|
"虎扑": "hupu",
|
|
"贴吧": "tieba",
|
|
"掘金": "juejin",
|
|
"抖音": "douyin",
|
|
"V2EX": "v2ex",
|
|
"今日头条": "jinritoutiao"
|
|
}
|
|
|
|
|
|
async def news(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""
|
|
Send a message when the command /news is issued.
|
|
"""
|
|
|
|
user = update.effective_user
|
|
response_message = f"哈喽！ {user.first_name} {user.last_name if user.last_name else ''}, 我是一个新闻机器人。 请选择想要了解的站点热榜:\n\n"
|
|
|
|
# Create the inline keyboard layout
|
|
reply_markup = build_reply_makeup()
|
|
|
|
await context.bot.send_message(
|
|
chat_id=update.effective_chat.id,
|
|
text=response_message,
|
|
reply_markup=reply_markup
|
|
)
|
|
|
|
|
|
async def selected_news(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""
|
|
Return the selected site's news.
|
|
"""
|
|
query = update.callback_query
|
|
await query.answer()
|
|
|
|
news_url = f"https://orz.ai/api/v1/dailynews/?platform={supported_sites[query.data]}"
|
|
|
|
resp = requests.get(news_url)
|
|
|
|
if resp.status_code != 200:
|
|
await context.bot.send_message(
|
|
chat_id=update.effective_chat.id,
|
|
text=f"Failed to get news from {query.data}, please try again later.",
|
|
)
|
|
return
|
|
|
|
response_message = f"下面是来自*{query.data}*的最新热榜：\n\n"
|
|
|
|
news_datas = resp.json()['data']
|
|
for i, news_data in enumerate(news_datas):
|
|
if i >= 10:
|
|
break
|
|
|
|
response_message += f"{i + 1}\. [{escape_markdown(news_data['title'], version=2)}]({escape_markdown(news_data['url'], version=2)})\n\n"
|
|
|
|
reply_markup = build_reply_makeup()
|
|
|
|
await context.bot.send_message(
|
|
chat_id=update.effective_chat.id,
|
|
text=response_message,
|
|
parse_mode=telegram.constants.ParseMode.MARKDOWN_V2,
|
|
reply_markup=reply_markup
|
|
)
|
|
|
|
|
|
def build_reply_makeup():
|
|
"""
|
|
Build the inline keyboard markup.
|
|
"""
|
|
keyboard = []
|
|
row = []
|
|
i = 0
|
|
for k in supported_sites.keys():
|
|
row.append(InlineKeyboardButton(k, callback_data=k))
|
|
if (i + 1) % 3 == 0:
|
|
keyboard.append(row)
|
|
row = []
|
|
i += 1
|
|
|
|
if row:
|
|
keyboard.append(row)
|
|
return InlineKeyboardMarkup(keyboard)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
token = os.getenv('TG_BOT_TOKEN')
|
|
if not token:
|
|
logging.error("Please set the environment variable 'TG_BOT_TOKEN'")
|
|
exit(1)
|
|
|
|
# if you want to use proxy
|
|
# proxy_url = 'http://127.0.0.1:7890'
|
|
# app = ApplicationBuilder().token(token).proxy(proxy_url).get_updates_proxy_url(proxy_url).build()
|
|
|
|
app = ApplicationBuilder().token(token).build()
|
|
|
|
app.add_handler(CommandHandler(['start', 'news'], news))
|
|
app.add_handler(telegram.ext.CallbackQueryHandler(selected_news))
|
|
|
|
loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(loop)
|
|
|
|
# use polling mode
|
|
app.run_polling()
|