mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 05:21:48 +00:00
2d14890c41
* improve screener on discordbot * improve screener code of discordbot * linter * line length
130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
import difflib
|
|
import discord
|
|
import pandas as pd
|
|
|
|
from gamestonk_terminal.stocks.screener.finviz_model import get_screener_data
|
|
import discordbot.config_discordbot as cfg
|
|
from discordbot.helpers import pagination
|
|
from discordbot.stocks.screener import screener_options as so
|
|
|
|
|
|
async def technical_command(ctx, preset="template", sort="", limit="5", ascend="False"):
|
|
"""Displays stocks according to chosen preset, sorting by technical factors [Finviz]"""
|
|
try:
|
|
# Check for argument
|
|
if preset == "template" or preset not in so.all_presets:
|
|
raise Exception("Invalid preset selected!")
|
|
|
|
# Debug
|
|
if cfg.DEBUG:
|
|
print(f"!stocks.scr.technical {preset} {sort} {limit} {ascend}")
|
|
|
|
# Check for argument
|
|
if not limit.lstrip("-").isnumeric():
|
|
raise Exception("Number has to be an integer")
|
|
|
|
limit = int(limit)
|
|
|
|
if limit < 0:
|
|
raise Exception("Number has to be above 0")
|
|
|
|
if ascend.lower() == "false":
|
|
ascend = False
|
|
elif ascend.lower() == "true":
|
|
ascend = True
|
|
else:
|
|
raise Exception("ascend argument has to be true or false")
|
|
|
|
# Output Data
|
|
df_screen = get_screener_data(
|
|
preset,
|
|
"technical",
|
|
limit,
|
|
ascend,
|
|
)
|
|
|
|
description = ""
|
|
|
|
if isinstance(df_screen, pd.DataFrame):
|
|
if df_screen.empty:
|
|
return []
|
|
|
|
df_screen = df_screen.dropna(axis="columns", how="all")
|
|
|
|
if sort:
|
|
if " ".join(sort) in so.d_cols_to_sort["technical"]:
|
|
df_screen = df_screen.sort_values(
|
|
by=[" ".join(sort)],
|
|
ascending=ascend,
|
|
na_position="last",
|
|
)
|
|
else:
|
|
similar_cmd = difflib.get_close_matches(
|
|
" ".join(sort),
|
|
so.d_cols_to_sort["technical"],
|
|
n=1,
|
|
cutoff=0.7,
|
|
)
|
|
if similar_cmd:
|
|
description = f"Replacing '{' '.join(sort)}' by '{similar_cmd[0]}' so table can be sorted.\n\n"
|
|
df_screen = df_screen.sort_values(
|
|
by=[similar_cmd[0]],
|
|
ascending=ascend,
|
|
na_position="last",
|
|
)
|
|
else:
|
|
raise ValueError(
|
|
f"Wrong sort column provided! Select from: {', '.join(so.d_cols_to_sort['technical'])}"
|
|
)
|
|
|
|
df_screen = df_screen.fillna("")
|
|
future_column_name = df_screen["Ticker"]
|
|
df_screen = df_screen.head(n=limit).transpose()
|
|
df_screen.columns = future_column_name
|
|
df_screen.drop("Ticker")
|
|
|
|
columns = []
|
|
initial_str = description + "Page 0: Overview"
|
|
i = 1
|
|
for column in df_screen.columns.values:
|
|
initial_str = initial_str + "\nPage " + str(i) + ": " + column
|
|
i += 1
|
|
columns.append(
|
|
discord.Embed(
|
|
title="Stocks: [Finviz] Technical Screener",
|
|
description=initial_str,
|
|
colour=cfg.COLOR,
|
|
).set_author(
|
|
name=cfg.AUTHOR_NAME,
|
|
icon_url=cfg.AUTHOR_ICON_URL,
|
|
)
|
|
)
|
|
for column in df_screen.columns.values:
|
|
columns.append(
|
|
discord.Embed(
|
|
title="Stocks: [Finviz] Technical Screener",
|
|
description="```"
|
|
+ df_screen[column].fillna("").to_string()
|
|
+ "```",
|
|
colour=cfg.COLOR,
|
|
).set_author(
|
|
name=cfg.AUTHOR_NAME,
|
|
icon_url=cfg.AUTHOR_ICON_URL,
|
|
)
|
|
)
|
|
|
|
await pagination(columns, ctx)
|
|
|
|
except Exception as e:
|
|
embed = discord.Embed(
|
|
title="ERROR Stocks: [Finviz] Technical Screener",
|
|
colour=cfg.COLOR,
|
|
description=e,
|
|
)
|
|
embed.set_author(
|
|
name=cfg.AUTHOR_NAME,
|
|
icon_url=cfg.AUTHOR_ICON_URL,
|
|
)
|
|
|
|
await ctx.send(embed=embed)
|