Added ticker command
This commit is contained in:
parent
cd0e4111bc
commit
46838c888c
2 changed files with 25 additions and 2 deletions
17
main.py
17
main.py
|
@ -25,6 +25,23 @@ async def ping(ctx):
|
||||||
await ctx.send("pong!")
|
await ctx.send("pong!")
|
||||||
|
|
||||||
|
|
||||||
|
@bot.command()
|
||||||
|
async def ticker(ctx, stock):
|
||||||
|
user = str(ctx.message.author)
|
||||||
|
stock = stock.upper()
|
||||||
|
|
||||||
|
data = yfi.stock_info(stock)
|
||||||
|
return await ctx.send(
|
||||||
|
"""**{ticker}**
|
||||||
|
|
||||||
|
Bid: {bid}
|
||||||
|
Ask: {ask}
|
||||||
|
""".format(
|
||||||
|
ticker=stock, bid=data["bid"], ask=data["ask"]
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@bot.command(aliases=["pft"])
|
@bot.command(aliases=["pft"])
|
||||||
async def portfolio_text(ctx):
|
async def portfolio_text(ctx):
|
||||||
user = str(ctx.message.author)
|
user = str(ctx.message.author)
|
||||||
|
|
10
yfi.py
10
yfi.py
|
@ -3,12 +3,18 @@ import yfinance as yf
|
||||||
|
|
||||||
def stock_exists(ticker):
|
def stock_exists(ticker):
|
||||||
try:
|
try:
|
||||||
yf.Ticker(ticker).info
|
return yf.Ticker(ticker).info
|
||||||
return True
|
|
||||||
except KeyError:
|
except KeyError:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def stock_info(ticker):
|
||||||
|
if not (stock := stock_exists(ticker)):
|
||||||
|
return False
|
||||||
|
|
||||||
|
return stock
|
||||||
|
|
||||||
|
|
||||||
def get_current_price(ticker):
|
def get_current_price(ticker):
|
||||||
yf_obj = yf.Ticker(ticker)
|
yf_obj = yf.Ticker(ticker)
|
||||||
todays_data = yf_obj.history(period="1d")
|
todays_data = yf_obj.history(period="1d")
|
||||||
|
|
Loading…
Add table
Reference in a new issue