Shirino/main.py

106 lines
3.4 KiB
Python
Raw Normal View History

from aiogram import Bot, Dispatcher, types
import asyncio
import yaml
2024-08-25 22:16:23 +03:00
import hashlib
2024-08-25 22:16:23 +03:00
import json
from function.convert import Converter
from function.format_number import format_number
2023-03-08 13:31:45 +03:00
config = yaml.safe_load(open("config.yaml"))
dp = Dispatcher()
2023-03-08 13:31:45 +03:00
@dp.message()
@dp.inline_query()
async def currency(query: types.Message | types.InlineQuery) -> None:
global result, from_currency_alias, conv_currency_alias
2023-03-08 13:31:45 +03:00
try:
currency_json = json.load(open('currency.json', 'r', encoding='utf-8'))
text = query.query if isinstance(query, types.InlineQuery) else query.text
args = text.split()
result_id = hashlib.md5(text.encode()).hexdigest()
conv = Converter()
if len(args) <= 1:
if query.chat.type not in ['supergroup', 'group']:
return await reply(result_id,
"2 or 3 arguments are required.",
"@shirino_bot USD RUB "
"\n@shirino_bot 12 USD RUB",
query)
if len(args) == 4:
conv.amount = float(args[0])
from_currency_alias = args[1].lower()
conv_currency_alias = args[3].lower()
elif len(args) == 3:
conv.amount = float(args[0])
from_currency_alias = args[1].lower()
conv_currency_alias = args[2].lower()
elif len(args) == 2:
from_currency_alias = args[0].lower()
conv_currency_alias = args[1].lower()
else:
if query.chat.type not in ['supergroup', 'group']:
return await reply(result_id, 'The source and target currency could not be determined.', None, query)
from_currency, conv_currency = None, None
for currency_code, aliases in currency_json.items():
if from_currency_alias in aliases:
from_currency = currency_code
if conv_currency_alias in aliases:
conv_currency = currency_code
if from_currency and conv_currency:
break
if not from_currency or not conv_currency:
return await reply(result_id,
'The currency exchange rate could not be found.',
None,
query)
conv.from_currency = from_currency.upper()
conv.conv_currency = conv_currency.upper()
conv.convert()
result = f'{format_number(conv.amount)} {conv.from_currency} = {conv.conv_amount} {conv.conv_currency}'
return await reply(result_id, result, None, query)
except Exception as e:
print(e)
async def reply(result_id: str | None, title: str | None, desc, query: types.InlineQuery | types.Message) -> None:
if isinstance(query, types.InlineQuery):
article = [None]
article[0] = types.InlineQueryResultArticle(
id=result_id,
title=title,
description=desc,
input_message_content=types.InputTextMessageContent(
message_text=title
)
)
await query.answer(
article,
cache_time=1,
is_personal=True,
)
else:
await query.answer(title)
2023-03-08 13:31:45 +03:00
async def main() -> None:
bot = Bot(config['telegram_token'])
await dp.start_polling(bot)
if __name__ == '__main__':
asyncio.run(main())