Source code for tt.plugins.default_plugins.cex_exchange_plugin

from cefi import CexTrader
from findmyorder import FindMyOrder

from tt.config import logger, settings
from tt.plugins.plugin_manager import BasePlugin
from tt.utils import send_notification




class CexExchangePlugin(BasePlugin):
    """
    Class CexExchangePlugin
    to support CEX Exchange
    built via Cefi lib
    More info: https://github.com/mraniki/cefi
    Order are identified and parsed
    using Findmyorder lib
    More info: https://github.com/mraniki/findmyorder

    Args:
        None

    Returns:
        None

    """

    # name = os.path.splitext(os.path.basename(__file__))[0]



    def __init__(self):
        super().__init__()
        self.enabled = settings.cex_enabled
        if self.enabled:
            self.fmo = FindMyOrder()
            self.exchange = CexTrader()




    async def send_notification(self, message):
        """Sends a notification"""
        if self.enabled:
            await send_notification(message)




    async def handle_message(self, msg):
        """
        Handles incoming messages
        to route to the respective function

        Args:
            msg (str): The incoming message

        Returns:
            None

        """
        if not self.should_handle(msg):
            return
        logger.debug("settings.bot_ignore: {}", settings.bot_ignore)
        if (settings.bot_ignore not in msg or settings.bot_prefix not in msg) and (
            await self.fmo.search(msg) and self.should_handle_timeframe()
        ):
            order = await self.fmo.get_order(msg)
            if order and settings.trading_enabled:
                trade = await self.exchange.submit_order(order)
                if trade:
                    await send_notification(trade)

        if msg.startswith(settings.bot_prefix):
            command, *args = msg.split(" ")
            command = command[1:]

            command_mapping = {
                settings.bot_command_info: self.exchange.get_info,
                settings.bot_command_quote: lambda: self.exchange.get_quotes(args[0]),
                settings.bot_command_bal: self.exchange.get_balances,
                settings.bot_command_pos: self.exchange.get_positions,
            }

            if command in command_mapping:
                function = command_mapping[command]
                await self.send_notification(f"{await function()}")