mirror of
https://github.com/Redume/Shirino.git
synced 2024-12-25 01:33:45 +03:00
feat: Added selection of currency rate with and without chart
This commit is contained in:
parent
6d9dfac41a
commit
75e7555e50
1 changed files with 72 additions and 68 deletions
140
main.py
140
main.py
|
|
@ -1,123 +1,127 @@
|
|||
from aiogram import Bot, Dispatcher, types
|
||||
|
||||
import asyncio
|
||||
import yaml
|
||||
import asyncio
|
||||
import requests
|
||||
|
||||
import hashlib
|
||||
import json
|
||||
|
||||
from http import HTTPStatus
|
||||
import re
|
||||
|
||||
from function.convert import Converter
|
||||
from function.format_number import format_number
|
||||
|
||||
config = yaml.safe_load(open("config.yaml"))
|
||||
dp = Dispatcher()
|
||||
config = yaml.safe_load(open('config.yaml'))
|
||||
|
||||
|
||||
@dp.message()
|
||||
@dp.inline_query()
|
||||
async def currency(query: types.Message | types.InlineQuery) -> None:
|
||||
async def currency(query: types.InlineQuery) -> None:
|
||||
global result, from_currency, conv_currency
|
||||
|
||||
try:
|
||||
text = query.query if isinstance(query, types.InlineQuery) else query.text
|
||||
text = query.query.lower()
|
||||
args = text.split()
|
||||
result_id = hashlib.md5(text.encode()).hexdigest()
|
||||
|
||||
conv = Converter()
|
||||
|
||||
if len(args) <= 1:
|
||||
try:
|
||||
if query.chat.type in ['supergroup', 'group']:
|
||||
return
|
||||
except:
|
||||
pass
|
||||
if len(args) < 2:
|
||||
return await reply(result_id,
|
||||
[("2 or 3 arguments are required.",
|
||||
'@shirino_bot USD RUB \n'
|
||||
'@shirino_bot 12 USD RUB')],
|
||||
query)
|
||||
|
||||
return await reply(result_id,
|
||||
"2 or 3 arguments are required.",
|
||||
"@shirino_bot USD RUB "
|
||||
"\n@shirino_bot 12 USD RUB",
|
||||
None,
|
||||
query)
|
||||
if len(args) == 4:
|
||||
if len(args) == 3:
|
||||
conv.amount = float(args[0])
|
||||
from_currency = args[1].lower()
|
||||
conv_currency = args[3].lower()
|
||||
elif len(args) == 3:
|
||||
conv.amount = float(args[0])
|
||||
from_currency = args[1].lower()
|
||||
conv_currency = args[2].lower()
|
||||
from_currency = args[1]
|
||||
conv_currency = args[2]
|
||||
elif len(args) == 2:
|
||||
from_currency = args[0].lower()
|
||||
conv_currency = args[1].lower()
|
||||
from_currency = args[0]
|
||||
conv_currency = args[1]
|
||||
else:
|
||||
try:
|
||||
if query.chat.type in ['supergroup', 'group']:
|
||||
return
|
||||
except:
|
||||
pass
|
||||
|
||||
return await reply(result_id, 'The source and target currency could not be determined.', None, None, query)
|
||||
return await reply(result_id, [('The source and target currency could not be determined.')], query)
|
||||
|
||||
if not from_currency or not conv_currency:
|
||||
return await reply(result_id,
|
||||
'The currency exchange rate could not be found.',
|
||||
None,
|
||||
None,
|
||||
query)
|
||||
return await reply(result_id, [('The currency exchange rate could not be found.')], query)
|
||||
|
||||
conv.from_currency = from_currency.upper()
|
||||
conv.conv_currency = conv_currency.upper()
|
||||
conv.convert()
|
||||
|
||||
res_chart = requests.get(f'{config['kekkai_instance']}/api/getChart/week/', {
|
||||
req_chart = requests.get(f'{config['kekkai_instance']}/api/getChart/week/', {
|
||||
'from_currency': from_currency,
|
||||
'conv_currency': conv_currency
|
||||
}, timeout=3)
|
||||
|
||||
if not HTTPStatus(res_chart.status_code).is_success:
|
||||
res_chart = None
|
||||
if not HTTPStatus(req_chart.status_code).is_success:
|
||||
req_chart = None
|
||||
else:
|
||||
res_chart = res_chart.json()['message']
|
||||
req_chart = req_chart.json().get('message', None)
|
||||
|
||||
result = f'{format_number(conv.amount)} {conv.from_currency} = {conv.conv_amount} {conv.conv_currency}' \
|
||||
f'\n{f'[График]({res_chart})' if res_chart else ''}'
|
||||
return await reply(result_id, result, None, res_chart, query)
|
||||
await reply(result_id,
|
||||
[
|
||||
(
|
||||
f'{format_number(conv.amount)} {conv.from_currency} = {conv.conv_amount} {conv.conv_currency}' \
|
||||
f'\n{f'[График]({req_chart})' if req_chart else ''}',
|
||||
None,
|
||||
req_chart
|
||||
),
|
||||
(
|
||||
f'{format_number(conv.amount)} {conv.from_currency} = {conv.conv_amount} {conv.conv_currency}',
|
||||
None,
|
||||
None
|
||||
)
|
||||
],
|
||||
query)
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
|
||||
async def reply(
|
||||
result_id: str | None,
|
||||
title: str,
|
||||
desc: str | None,
|
||||
img: str | None,
|
||||
query: types.InlineQuery | types.Message,
|
||||
) -> None:
|
||||
async def reply(result_id: str, args: list, query: types.InlineQuery) -> None:
|
||||
if not args:
|
||||
return
|
||||
|
||||
if isinstance(query, types.InlineQuery):
|
||||
article = [None]
|
||||
article[0] = types.InlineQueryResultArticle(
|
||||
id=result_id,
|
||||
title=title,
|
||||
thumbnail_url=img,
|
||||
description=desc,
|
||||
articles = []
|
||||
|
||||
for idx, arg in enumerate(args):
|
||||
title = arg[0]
|
||||
description = arg[1] if arg[1] else None
|
||||
img = arg[2] if arg[2] else None
|
||||
|
||||
|
||||
article = types.InlineQueryResultArticle(
|
||||
id=f"{result_id}_{idx}",
|
||||
title=remove_markdown(title).replace('График', ''),
|
||||
thumbnail_url=img,
|
||||
description=description,
|
||||
input_message_content=types.InputTextMessageContent(
|
||||
message_text=title,
|
||||
parse_mode='markdown'
|
||||
)
|
||||
)
|
||||
|
||||
await query.answer(
|
||||
article,
|
||||
parse_mode='markdown',
|
||||
cache_time=0,
|
||||
is_personal=True,
|
||||
)
|
||||
else:
|
||||
await query.answer(f'{title} \n{'' if desc else ''}')
|
||||
articles.append(article)
|
||||
|
||||
await query.answer(
|
||||
results=articles,
|
||||
parse_mode='markdown',
|
||||
cache_time=0,
|
||||
is_personal=True
|
||||
)
|
||||
|
||||
def remove_markdown(text: str) -> str:
|
||||
text = re.sub(r'\*([^\*]+)\*', r'\1', text)
|
||||
text = re.sub(r'\_([^\_]+)\_', r'\1', text)
|
||||
text = re.sub(r'\[([^\[]+)\]\([^\)]+\)', r'\1', text)
|
||||
text = re.sub(r'[`]+([^`]+)`+', r'\1', text)
|
||||
text = re.sub(r'~~([^~]+)~~', r'\1', text)
|
||||
text = re.sub(r'\[([^\[]+)\]\([^\)]+\)', '', text)
|
||||
return text
|
||||
|
||||
|
||||
async def main() -> None:
|
||||
bot = Bot(config['telegram_token'])
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue