mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 20:42:26 +00:00
fix and improve options menu (#1090)
This commit is contained in:
Didier Rodrigues Lopes
committed by
GitHub
GitHub
parent
a0a22ee2cf
commit
e56cdcc49c
@@ -1,7 +1,6 @@
|
||||
import asyncio
|
||||
|
||||
import discord
|
||||
import discord_components
|
||||
|
||||
import discordbot.config_discordbot as cfg
|
||||
from discordbot.run_discordbot import gst_bot
|
||||
@@ -74,202 +73,95 @@ class OptionsCommands(discord.ext.commands.Cog):
|
||||
await oi_command(ctx, ticker, expiration)
|
||||
|
||||
@discord.ext.commands.command(name="stocks.opt")
|
||||
async def opt(self, ctx: discord.ext.commands.Context, ticker=""):
|
||||
"""Stocks Context - Shows Options Menu"""
|
||||
async def opt(self, ctx: discord.ext.commands.Context, ticker="", expiration=""):
|
||||
"""Stocks Context - Shows Options Menu
|
||||
|
||||
Returns
|
||||
-------
|
||||
Sends a message to the discord user with the commands from the stocks/options context.
|
||||
The user can then select a reaction to trigger a command.
|
||||
"""
|
||||
|
||||
if cfg.DEBUG:
|
||||
print(f"!stocks.opt {ticker}")
|
||||
cols_temp = []
|
||||
cols = []
|
||||
|
||||
if ticker == "":
|
||||
if not ticker:
|
||||
embed = discord.Embed(
|
||||
title="ERROR Stocks: Options (opt) Menu",
|
||||
description="Provide a ticker and expiration date with this menu, e.g. !stocks.opt TSLA 2021-06-04",
|
||||
colour=cfg.COLOR,
|
||||
description="A stock ticker is required",
|
||||
)
|
||||
embed.set_author(
|
||||
title="ERROR Stocks: Options Menu",
|
||||
).set_author(
|
||||
name=cfg.AUTHOR_NAME,
|
||||
icon_url=cfg.AUTHOR_ICON_URL,
|
||||
)
|
||||
|
||||
await ctx.send(embed=embed)
|
||||
return
|
||||
|
||||
text = (
|
||||
f"0️⃣ !stocks.opt.exp {ticker}\n"
|
||||
f"1️⃣ !stocks.opt.calls {ticker} <EXPIRATION> \n"
|
||||
)
|
||||
|
||||
cols_temp.append(text)
|
||||
for col in cols_temp:
|
||||
cols.append(
|
||||
discord.Embed(
|
||||
description=col,
|
||||
colour=cfg.COLOR,
|
||||
title="Stocks: Options (opt) Menu",
|
||||
).set_author(
|
||||
name=cfg.AUTHOR_NAME,
|
||||
icon_url=cfg.AUTHOR_ICON_URL,
|
||||
)
|
||||
if expiration:
|
||||
text = (
|
||||
f"0️⃣ !stocks.opt.exp {ticker}\n"
|
||||
f"1️⃣ !stocks.opt.calls {ticker} {expiration} \n"
|
||||
f"2️⃣ !stocks.opt.puts {ticker} {expiration} \n"
|
||||
f"3️⃣ !stocks.opt.oi {ticker} {expiration} \n"
|
||||
)
|
||||
else:
|
||||
text = f"0️⃣ !stocks.opt.exp {ticker}\n"
|
||||
|
||||
title = "Stocks: Options Menu"
|
||||
embed = discord.Embed(title=title, description=text, colour=cfg.COLOR)
|
||||
embed.set_author(
|
||||
name=cfg.AUTHOR_NAME,
|
||||
icon_url=cfg.AUTHOR_ICON_URL,
|
||||
)
|
||||
msg = await ctx.send(embed=embed)
|
||||
|
||||
if expiration:
|
||||
emoji_list = ["0️⃣", "1️⃣", "2️⃣", "3️⃣"]
|
||||
else:
|
||||
emoji_list = ["0️⃣"]
|
||||
|
||||
emoji_list = ["0️⃣", "1️⃣", "2️⃣", "3️⃣"]
|
||||
current = 0
|
||||
components = [
|
||||
[
|
||||
discord_components.Button(
|
||||
label="Prev", id="back", style=discord_components.ButtonStyle.red
|
||||
),
|
||||
discord_components.Button(
|
||||
label=f"Page {int(cols.index(cols[current]))}/{len(cols) - 1}",
|
||||
id="cur",
|
||||
style=discord_components.ButtonStyle.green,
|
||||
disabled=True,
|
||||
),
|
||||
discord_components.Button(
|
||||
label="Next", id="front", style=discord_components.ButtonStyle.green
|
||||
),
|
||||
]
|
||||
]
|
||||
main_message = await ctx.send(embed=cols[current], components=components)
|
||||
for emoji in emoji_list:
|
||||
await main_message.add_reaction(emoji)
|
||||
await msg.add_reaction(emoji)
|
||||
|
||||
while True:
|
||||
# Try and except blocks to catch timeout and break
|
||||
try:
|
||||
interaction = await gst_bot.wait_for(
|
||||
"button_click",
|
||||
check=lambda i: i.component.id
|
||||
in ["back", "front"], # You can add more
|
||||
timeout=cfg.MENU_TIMEOUT, # Some seconds of inactivity
|
||||
)
|
||||
def check(reaction, user):
|
||||
return user == ctx.message.author and str(reaction.emoji) in emoji_list
|
||||
|
||||
# Getting the right list index
|
||||
if interaction.component.id == "back":
|
||||
current -= 1
|
||||
elif interaction.component.id == "front":
|
||||
current += 1
|
||||
try:
|
||||
reaction, _ = await gst_bot.wait_for(
|
||||
"reaction_add", timeout=cfg.MENU_TIMEOUT, check=check
|
||||
)
|
||||
if reaction.emoji == "0️⃣":
|
||||
if cfg.DEBUG:
|
||||
print("Reaction selected: 0")
|
||||
await expirations_command(ctx, ticker)
|
||||
elif reaction.emoji == "1️⃣":
|
||||
if cfg.DEBUG:
|
||||
print("Reaction selected: 1")
|
||||
await calls_command(ctx, ticker, expiration)
|
||||
elif reaction.emoji == "2️⃣":
|
||||
if cfg.DEBUG:
|
||||
print("Reaction selected: 2")
|
||||
await puts_command(ctx, ticker, expiration)
|
||||
elif reaction.emoji == "3️⃣":
|
||||
if cfg.DEBUG:
|
||||
print("Reaction selected: 3")
|
||||
await oi_command(ctx, ticker, expiration)
|
||||
|
||||
# If its out of index, go back to start / end
|
||||
if current == len(cols):
|
||||
current = 0
|
||||
elif current < 0:
|
||||
current = len(cols) - 1
|
||||
for emoji in emoji_list:
|
||||
await msg.remove_reaction(emoji, ctx.bot.user)
|
||||
|
||||
# Edit to new page + the center counter changes
|
||||
components = [
|
||||
[
|
||||
discord_components.Button(
|
||||
label="Prev",
|
||||
id="back",
|
||||
style=discord_components.ButtonStyle.red,
|
||||
),
|
||||
discord_components.Button(
|
||||
label=f"Page {int(cols.index(cols[current]))}/{len(cols) - 1}",
|
||||
id="cur",
|
||||
style=discord_components.ButtonStyle.green,
|
||||
disabled=True,
|
||||
),
|
||||
discord_components.Button(
|
||||
label="Next",
|
||||
id="front",
|
||||
style=discord_components.ButtonStyle.green,
|
||||
),
|
||||
]
|
||||
]
|
||||
await interaction.edit_origin(
|
||||
embed=cols[current], components=components
|
||||
)
|
||||
|
||||
# pylint: disable=too-many-branches
|
||||
@gst_bot.event
|
||||
async def on_reaction_add(reaction, user):
|
||||
if user == ctx.message.author and str(reaction.emoji) in emoji_list:
|
||||
if reaction.emoji == "0️⃣":
|
||||
if cfg.DEBUG:
|
||||
print("Reaction selected: 0")
|
||||
if current == 0:
|
||||
await expirations_command(ctx, ticker)
|
||||
elif reaction.emoji == "1️⃣":
|
||||
if cfg.DEBUG:
|
||||
print("Reaction selected: 1")
|
||||
await calls_command(ctx, ticker)
|
||||
elif reaction.emoji == "2️⃣️":
|
||||
if cfg.DEBUG:
|
||||
print("Reaction selected: 2")
|
||||
await puts_command(ctx, ticker)
|
||||
elif reaction.emoji == "3️⃣":
|
||||
if cfg.DEBUG:
|
||||
print("Reaction selected: 3")
|
||||
if current == 0:
|
||||
await oi_command(ctx, ticker)
|
||||
|
||||
for emoji in emoji_list:
|
||||
await main_message.remove_reaction(emoji, ctx.bot.user)
|
||||
components = [
|
||||
[
|
||||
discord_components.Button(
|
||||
label="Prev",
|
||||
id="back",
|
||||
style=discord_components.ButtonStyle.green,
|
||||
disabled=True,
|
||||
),
|
||||
discord_components.Button(
|
||||
label=f"Page {int(cols.index(cols[current]))}/{len(cols) - 1}",
|
||||
id="cur",
|
||||
style=discord_components.ButtonStyle.grey,
|
||||
disabled=True,
|
||||
),
|
||||
discord_components.Button(
|
||||
label="Next",
|
||||
id="front",
|
||||
style=discord_components.ButtonStyle.green,
|
||||
disabled=True,
|
||||
),
|
||||
]
|
||||
]
|
||||
await main_message.edit(components=components)
|
||||
return
|
||||
|
||||
except asyncio.TimeoutError:
|
||||
# Disable and get outta here
|
||||
components = [
|
||||
[
|
||||
discord_components.Button(
|
||||
label="Prev",
|
||||
id="back",
|
||||
style=discord_components.ButtonStyle.green,
|
||||
disabled=True,
|
||||
),
|
||||
discord_components.Button(
|
||||
label=f"Page {int(cols.index(cols[current]))}/{len(cols) - 1}",
|
||||
id="cur",
|
||||
style=discord_components.ButtonStyle.grey,
|
||||
disabled=True,
|
||||
),
|
||||
discord_components.Button(
|
||||
label="Next",
|
||||
id="front",
|
||||
style=discord_components.ButtonStyle.green,
|
||||
disabled=True,
|
||||
),
|
||||
]
|
||||
]
|
||||
await main_message.edit(components=components)
|
||||
embed = discord.Embed(
|
||||
description="Error timeout - you snooze you lose! 😋",
|
||||
colour=cfg.COLOR,
|
||||
title="TIMEOUT Stocks: Options (opt) Menu",
|
||||
).set_author(
|
||||
name=cfg.AUTHOR_NAME,
|
||||
icon_url=cfg.AUTHOR_ICON_URL,
|
||||
)
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
for emoji in emoji_list:
|
||||
await main_message.remove_reaction(emoji, ctx.bot.user)
|
||||
break
|
||||
except asyncio.TimeoutError:
|
||||
for emoji in emoji_list:
|
||||
await msg.remove_reaction(emoji, ctx.bot.user)
|
||||
embed = discord.Embed(
|
||||
description="Error timeout - you snooze you lose! 😋",
|
||||
colour=cfg.COLOR,
|
||||
title="TIMEOUT Stocks: Government (GOV) Menu",
|
||||
).set_author(
|
||||
name=cfg.AUTHOR_NAME,
|
||||
icon_url=cfg.AUTHOR_ICON_URL,
|
||||
)
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
|
||||
def setup(bot: discord.ext.commands.Bot):
|
||||
|
||||
Reference in New Issue
Block a user