mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-03 16:02:04 +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>
107 lines
2.9 KiB
Python
107 lines
2.9 KiB
Python
"""NASDAQ DataLink Model"""
|
|
__docformat__ = "numpy"
|
|
|
|
from datetime import datetime
|
|
import logging
|
|
|
|
import pandas as pd
|
|
import requests
|
|
|
|
import openbb_terminal.config_terminal as cfg
|
|
from openbb_terminal.decorators import check_api_key, log_start_end
|
|
from openbb_terminal.helper_funcs import get_user_agent
|
|
from openbb_terminal.rich_config import console
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
@check_api_key(["API_KEY_QUANDL"])
|
|
def get_retail_tickers() -> pd.DataFrame:
|
|
"""Gets the top 10 retail stocks per day
|
|
|
|
Returns
|
|
-------
|
|
pd.DataFrame
|
|
Dataframe of tickers
|
|
"""
|
|
r = requests.get(
|
|
f"https://data.nasdaq.com/api/v3/datatables/NDAQ/RTAT10/?api_key={cfg.API_KEY_QUANDL}"
|
|
)
|
|
|
|
df = pd.DataFrame()
|
|
|
|
if r.status_code == 200:
|
|
df = pd.DataFrame(r.json()["datatable"]["data"])
|
|
df.columns = ["Date", "Ticker", "Activity", "Sentiment"]
|
|
# Wrong API Key
|
|
elif r.status_code == 400:
|
|
console.print(r.text)
|
|
# Premium Feature
|
|
elif r.status_code == 403:
|
|
console.print(r.text)
|
|
# Catching other exception
|
|
elif r.status_code != 200:
|
|
console.print(r.text)
|
|
|
|
return df
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def get_dividend_cal(date: str = None) -> pd.DataFrame:
|
|
"""Gets dividend calendar for given date. Date represents Ex-Dividend Date
|
|
|
|
Parameters
|
|
----------
|
|
date: datetime
|
|
Date to get for in format YYYY-MM-DD
|
|
|
|
Returns
|
|
-------
|
|
pd.DataFrame
|
|
Dataframe of dividend calendar
|
|
"""
|
|
# TODO: HELP WANTED:
|
|
# Nasdaq API doesn't like a lot of stuff. Your ISP or VPN, the specific user agent
|
|
# that you might be using, etc. More exploration is required to make this feature
|
|
# equally usable for all. In the time being we patch selection of the user agent and
|
|
# add a timeout for cases when the URL doesn't respond.
|
|
|
|
if date is None:
|
|
date = datetime.today().strftime("%Y-%m-%d")
|
|
|
|
ag = get_user_agent()
|
|
# Nasdaq API doesn't like this user agent, thus we always get other than this particular one
|
|
while (
|
|
ag
|
|
== "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.10; rv:82.1) Gecko/20100101 Firefox/82.1"
|
|
):
|
|
ag = get_user_agent()
|
|
try:
|
|
r = requests.get(
|
|
f"https://api.nasdaq.com/api/calendar/dividends?date={date}",
|
|
headers={"User-Agent": ag},
|
|
timeout=5,
|
|
)
|
|
|
|
df = pd.DataFrame()
|
|
|
|
if r.status_code == 200:
|
|
df = pd.DataFrame(r.json()["data"]["calendar"]["rows"])
|
|
|
|
# Wrong API Key
|
|
elif r.status_code == 400:
|
|
console.print(r.text)
|
|
# Premium Feature
|
|
elif r.status_code == 403:
|
|
console.print(r.text)
|
|
# Catching other exception
|
|
elif r.status_code != 200:
|
|
console.print(r.text)
|
|
|
|
except requests.exceptions.ReadTimeout:
|
|
logger.exception("Request timed out")
|
|
return pd.DataFrame()
|
|
return df
|