mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-09 10:51:57 +00:00
02901f1473
* first round of removing spaces * remove more print statemnts * remove more spaces * bullbear color * spacc color * improve stalker readability * delete more spaces * reddit colors * rever hood * flake8 * forgot this one * fix econometrics tests * fix econometrics tests * fix more tests Co-authored-by: James Maslek <jmaslek11@gmail.com>
104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
"""NASDAQ DataLink View"""
|
|
__docformat__ = "numpy"
|
|
|
|
from datetime import datetime
|
|
import logging
|
|
import os
|
|
|
|
from openbb_terminal.decorators import log_start_end
|
|
from openbb_terminal.helper_funcs import export_data, print_rich_table
|
|
from openbb_terminal.rich_config import console
|
|
from openbb_terminal.stocks.discovery import nasdaq_model
|
|
from openbb_terminal.decorators import check_api_key
|
|
|
|
# pylint: disable=E1123
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
@check_api_key(["API_KEY_QUANDL"])
|
|
def display_top_retail(limit: int = 3, export: str = ""):
|
|
"""Display the top 10 retail traded stocks for last days
|
|
|
|
Parameters
|
|
----------
|
|
limit : int, optional
|
|
Number of days to show by default 3
|
|
export : str, optional
|
|
Format to export data, by default ""
|
|
"""
|
|
retails = nasdaq_model.get_retail_tickers()
|
|
|
|
if retails.empty:
|
|
return
|
|
|
|
for date, df in retails.head(limit * 10).groupby("Date"):
|
|
df = df.drop(columns=["Date"])
|
|
df = df.reset_index(drop=True)
|
|
print_rich_table(
|
|
df,
|
|
headers=[x.title() for x in df.columns],
|
|
show_index=False,
|
|
title=f"[bold]{date} Top Retail:[/bold]",
|
|
)
|
|
console.print("")
|
|
|
|
export_data(export, os.path.dirname(os.path.abspath(__file__)), "rtat", retails)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def display_dividend_calendar(
|
|
date: str = None,
|
|
sortby: str = "Dividend",
|
|
ascend: bool = False,
|
|
limit: int = 10,
|
|
export: str = "",
|
|
):
|
|
"""Display NASDAQ dividend calendar
|
|
|
|
Parameters
|
|
----------
|
|
date: str
|
|
Date to get dividend calendar for, format YYYY-MM-DD
|
|
sortby: str
|
|
Column to sort data for
|
|
ascend: bool
|
|
Flag to sort in ascending order
|
|
limit: int
|
|
Number of results to show
|
|
export: str
|
|
Format to export data
|
|
"""
|
|
|
|
if date is None:
|
|
date = datetime.today().strftime("%Y-%m-%d")
|
|
|
|
div_map = {
|
|
"symbol": "Symbol",
|
|
"companyName": "Name",
|
|
"dividend_Ex_Date": "Ex-Dividend Date",
|
|
"payment_Date": "Payment Date",
|
|
"record_Date": "Record Date",
|
|
"dividend_Rate": "Dividend",
|
|
"indicated_Annual_Dividend": "Annual Dividend",
|
|
"announcement_Date": "Announcement Date",
|
|
}
|
|
calendar = nasdaq_model.get_dividend_cal(date)
|
|
if calendar.empty:
|
|
console.print(
|
|
"No data found. Check that the date provided is a market day. If it is then try this function"
|
|
" again as the request may have not gone through.\n"
|
|
)
|
|
return
|
|
calendar = calendar.drop(columns=["announcement_Date"])
|
|
calendar.columns = calendar.columns.map(div_map)
|
|
calendar = calendar.sort_values(by=sortby, ascending=ascend)
|
|
print_rich_table(
|
|
calendar.head(limit),
|
|
headers=[x.title() for x in calendar.columns],
|
|
title=f"[bold]Dividend Calendar for {date}[/bold]",
|
|
)
|
|
export_data(export, os.path.dirname(os.path.abspath(__file__)), "divcal", calendar)
|