PagerMaid_Plugins/bc.py

""" PagerMaid Plugin Coin by Pentacene """
#   ______          _
#   | ___ \        | |
#   | |_/ /__ _ __ | |_ __ _  ___ ___ _ __   ___
#   |  __/ _ \ '_ \| __/ _` |/ __/ _ \ '_ \ / _ \
#   | | |  __/ | | | || (_| | (_|  __/ | | |  __/
#   \_|  \___|_| |_|\__\__,_|\___\___|_| |_|\___|
#

from asyncio import sleep
from sys import executable
import urllib.request
from telethon.tl.custom.message import Message
from pagermaid.listener import listener
from pagermaid.utils import execute, alias_command

imported = True
try:
    from binance.client import Client
    import xmltodict
except ImportError:
    imported = False

API = "https://www.ecb.europa.eu/stats/eurofxref/eurofxref-daily.xml"
CURRENCIES = []
DATA = {}
BINANCE_API_KEY = '8PDfQ2lSIyHPWdNAHNIaIoNy3MiiMuvgwYADbmtsKo867B0xnIhIGjPULsOtvMRk'
BINANCE_API_SECRET = 'tbUiyZ94l0zpYOlKs3eO1dvLNMOSbOb2T1T0eT0I1eogH9Fh8Htvli05eZ1iDvra'


def init() -> None:
    """ INIT """
    with urllib.request.urlopen(API) as response:
        result = response.read()
        try:
            global CURRENCIES, DATA
            rate_data = xmltodict.parse(result)
            rate_data = rate_data['gesmes:Envelope']['Cube']['Cube']['Cube']
            for i in rate_data:
                CURRENCIES.append(i['@currency'])
                DATA[i['@currency']] = float(i['@rate'])
            CURRENCIES.sort()
        except Exception as e:
            raise e


@listener(is_plugin=True, outgoing=True, command=alias_command("bc"),
          description="coins",
          parameters="<num> <coin1> <coin2>")
async def coin(context: Message) -> None:
    """ coin change """
    if not imported:
        await context.edit("支持库 `python-binance` `xmltodict` 未安装...\n正在尝试自动安装...")
        await execute(f'{executable} -m pip install python-binance')
        await execute(f'{executable} -m pip install xmltodict')
        await sleep(10)
        result = await execute(f'{executable} -m pip show python-binance')
        result1 = await execute(f'{executable} -m pip show xmltodict')
        if len(result) > 0 and len(result1) > 0:
            await context.edit('支持库 `python-binance` `xmltodict` 安装成功...\n正在尝试自动重启...')
            await context.client.disconnect()
        else:
            await context.edit(f"自动安装失败..请尝试手动安装 `{executable} -m pip install python-binance`\n\n`"
                               f"{executable} -m pip install xmltodict`\n随后，请重启 PagerMaid")
        return
    init()
    action = context.arguments.split()
    binanceclient = Client(BINANCE_API_KEY, BINANCE_API_SECRET)
    if len(action) < 3:
        await context.edit('输入错误.\n-bc 数量 币种1 币种2')
        return
    else:
        prices = binanceclient.get_all_tickers()
        try:
            number = float(action[0])
        except ValueError:
            await context.edit('输入错误.\n-bc 数量 币种1 币种2')
            return
        _from = action[1].upper().strip()
        _to = action[2].upper().strip()
        front_text = ''
        text = ''
        rear_text = ''
        price = 0.0
        _to_USD_rate = 0.0

        if (CURRENCIES.count(_from) != 0) and (CURRENCIES.count(_to) != 0):
            # both are real currency
            text = f'{action[0]} {action[1].upper().strip()} = {float(action[0])*DATA[_to]/DATA[_from]:.2f} {action[2].upper().strip()}'

        else:
            if CURRENCIES.count(_from) != 0:
                # from virtual currency to real currency
                number = number * DATA["USD"] / DATA[_from]
                _from = 'USDT'
                front_text = f'{action[0]} {action[1]} = \n'

            if CURRENCIES.count(_to) != 0:
                # from real currency to virtual currency
                _to_USD_rate = DATA[_to] / DATA["USD"]
                _to = 'USDT'

            for _a in prices:
                if _a['symbol'] == str(f'{_from}{_to}'):
                    price = _a['price']
                    if _to == 'USDT':
                        if action[2].upper().strip() == 'USDT':
                            rear_text = f'\n= {number * float(price) * DATA["CNY"]/DATA["USD"]:.2f} CNY'
                        else:
                            rear_text = f'\n= {number * float(price) * _to_USD_rate:.2f} {action[2].upper().strip()}'
                    if float(price) < 1:
                        text = f'{number} {_from} = {number * float(price):.8f} {_to}'
                    else:
                        text = f'{number} {_from} = {number * float(price):.2f} {_to}'
                    break
                elif _a['symbol'] == str(f'{_to}{_from}'):
                    price = 1 / float(_a['price'])
                    text = f'{number} {_from} = {number * float(price):.8f} {_to}'
                    break
                else:
                    price = None

        if price is None:
            text = f'Cannot find coinpair {action[1].upper().strip()}{action[2].upper().strip()} or {action[2].upper().strip()}{action[1].upper().strip()}'

    await context.edit(f'{front_text}{text}{rear_text}')