mirror of
https://github.com/Xtao-Labs/iShotaBot.git
synced 2024-12-02 20:05:52 +00:00
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
from pyrogram import Client, filters
|
|
from pyrogram.enums import ChatType
|
|
from pyrogram.types import Message, InlineQuery, InlineQueryResultArticle, InputTextMessageContent
|
|
|
|
from defs.exchange import exchange_client
|
|
from scheduler import scheduler
|
|
from init import bot
|
|
|
|
|
|
@scheduler.scheduled_job("cron", hour="8", id="exchange.refresh")
|
|
async def exchange_refresh() -> None:
|
|
await exchange_client.refresh()
|
|
|
|
|
|
async def get_text(text: str, inline: bool):
|
|
if not exchange_client.inited:
|
|
return "获取汇率数据出现错误！", False
|
|
text = await exchange_client.check_ex(text)
|
|
if text == "help":
|
|
prefix = "" if inline else "/"
|
|
text = (
|
|
f"该指令可用于查询汇率。\n使用方式举例:\n{prefix}exchange USD CNY - 1 USD 等于多少 CNY\n"
|
|
f"{prefix}exchange 11 CNY USD - 11 CNY 等于多少 USD"
|
|
)
|
|
elif text == "ValueError":
|
|
text = "金额不合法"
|
|
elif text == "ValueBig":
|
|
text = "我寻思你也没这么多钱啊。"
|
|
elif text == "ValueSmall":
|
|
text = "小数点后的数字咋这么多？"
|
|
elif text == "FromError":
|
|
text = "不支持的起源币种。"
|
|
elif text == "ToError":
|
|
text = "不支持的目标币种。"
|
|
else:
|
|
return text, True
|
|
return text, False
|
|
|
|
|
|
@bot.on_message(
|
|
filters.incoming & filters.command(["exchange", f"exchange@{bot.me.username}"])
|
|
)
|
|
async def exchange_command(_: Client, message: Message):
|
|
if not exchange_client.inited:
|
|
await exchange_client.refresh()
|
|
text, success = await get_text(message.text, False)
|
|
reply_ = await message.reply(text)
|
|
if not success and message.chat.type == ChatType.PRIVATE:
|
|
await reply_.reply(
|
|
"支持货币： <code>" + ", ".join(exchange_client.currencies) + "</code>"
|
|
)
|
|
|
|
|
|
@bot.on_inline_query(filters.regex("^exchange"))
|
|
async def exchange_inline(_: Client, inline_query: InlineQuery):
|
|
text, success = await get_text(inline_query.query, True)
|
|
results = [
|
|
InlineQueryResultArticle(
|
|
title="查询汇率数据成功" if success else "查询汇率数据失败",
|
|
input_message_content=InputTextMessageContent(message_text=text),
|
|
# reply_markup=InlineKeyboardMarkup(
|
|
# [[InlineKeyboardButton(text="重试", callback_data="dc")]]
|
|
# ),
|
|
)
|
|
]
|
|
await inline_query.answer(
|
|
results=results,
|
|
cache_time=0,
|
|
)
|