OpenBB/bots/cmds/economy_cmds.py

from __future__ import annotations

import disnake
from disnake.ext import commands

from bots.common import commands_dict
from bots.economy.cpi import cpi_command
from bots.economy.currencies import currencies_command
from bots.economy.energy import energy_command
from bots.economy.futures import futures_command
from bots.economy.futures_coms import futures_coms_command
from bots.economy.glbonds import glbonds_command
from bots.economy.grains import grains_command
from bots.economy.indices import indices_command
from bots.economy.meats import meats_command
from bots.economy.metals import metals_command
from bots.economy.overview import overview_command
from bots.economy.performance import performance_command
from bots.economy.reverse_repo import reverse_repo_command
from bots.economy.softs import softs_command
from bots.economy.usbonds import usbonds_command
from bots.economy.valuation import valuation_command
from bots.helpers import ShowView


class EconomyCommands(commands.Cog):
    """Economy Commands menu"""

    def __init__(self, bot: commands.Bot):
        self.bot = bot

    @commands.slash_command(name="econ")
    async def econ(self, inter):
        pass

    @econ.sub_command()
    async def overview(self, inter: disnake.AppCmdInter):
        """Market data overview [Wall St. Journal]"""
        await ShowView().discord(overview_command, inter, "econ overview")

    @econ.sub_command()
    async def indices(self, inter: disnake.AppCmdInter):
        """US indices overview [Wall St. Journal]"""
        await ShowView().discord(indices_command, inter, "econ indices")

    @econ.sub_command()
    async def futures(self, inter: disnake.AppCmdInter):
        """Futures and commodities overview [Wall St. Journal]"""
        await ShowView().discord(futures_coms_command, inter, "econ futures")

    @commands.slash_command(name="futures")
    async def futs(self, inter: disnake.AppCmdInter):
        """Futures [Yahoo Finance]"""
        await ShowView().discord(futures_command, inter, "futures")

    @econ.sub_command()
    async def usbonds(self, inter: disnake.AppCmdInter):
        """US bonds overview [Wall St. Journal]"""
        await ShowView().discord(usbonds_command, inter, "econ usbonds")

    @econ.sub_command()
    async def glbonds(self, inter: disnake.AppCmdInter):
        """Global bonds overview [Wall St. Journal]"""
        await ShowView().discord(glbonds_command, inter, "econ glbonds")

    @econ.sub_command()
    async def cpi(self, inter: disnake.AppCmdInter, start: str = ""):
        """Displays Consumer Prices Index (CPI)
        Parameters
        ----------
        start: YYYY-MM-DD format
        """
        await ShowView().discord(cpi_command, inter, "econ energy", start)

    @econ.sub_command()
    async def energy(self, inter: disnake.AppCmdInter):
        """Displays energy futures data [Finviz]"""
        await ShowView().discord(energy_command, inter, "econ energy")

    @econ.sub_command()
    async def metals(self, inter: disnake.AppCmdInter):
        """Displays metals futures data [Finviz]"""
        await ShowView().discord(metals_command, inter, "econ metals")

    @econ.sub_command()
    async def meats(self, inter: disnake.AppCmdInter):
        """Displays meats futures data [Finviz]"""
        await ShowView().discord(meats_command, inter, "econ meats")

    @econ.sub_command()
    async def grains(self, inter: disnake.AppCmdInter):
        """Displays grains futures data [Finviz]"""
        await ShowView().discord(grains_command, inter, "econ grains")

    @econ.sub_command()
    async def softs(self, inter: disnake.AppCmdInter):
        """Displays softs futures data [Finviz]"""
        await ShowView().discord(softs_command, inter, "econ softs")

    @econ.sub_command()
    async def currencies(self, inter: disnake.AppCmdInter):
        """Currencies overview [Wall St. Journal]"""
        await ShowView().discord(currencies_command, inter, "econ currencies")

    @econ.sub_command()
    async def valuation(
        self,
        inter: disnake.AppCmdInter,
        economy_group: str = commands.Param(choices=commands_dict.econ_group),
    ):
        """Valuation of sectors, industry, country [Finviz]

        Parameters
        -----------
        economy_group: Economy Group
        """
        await ShowView().discord(
            valuation_command, inter, "econ valuation", economy_group
        )

    @econ.sub_command()
    async def repo(
        self,
        inter: disnake.AppCmdInter,
        days: int = 50,
    ):
        """Displays Reverse Repo [Stocksera]

        Parameters
        -----------
        days: Number of days to display. Default: 50
        """
        await ShowView().discord(reverse_repo_command, inter, "econ repo", days)

    @econ.sub_command()
    async def performance(
        self,
        inter: disnake.AppCmdInter,
        economy_group: str = commands.Param(choices=commands_dict.econ_group),
    ):
        """Performance of sectors, industry, country [Finviz]

        Parameters
        -----------
        economy_group: Economy Group
        """
        await ShowView().discord(
            performance_command, inter, "econ performance", economy_group
        )


def setup(bot: commands.Bot):
    bot.add_cog(EconomyCommands(bot))