Files
OpenBB/gamestonk_terminal/common/behavioural_analysis/finbrain_model.py
LBolte29 6426b76771 decorated functions (#1282)
* decorated functions and removed assertion tests that are now caught by decorator logs
* modified functions with @property
* fixed bc of coingecko
* fixed outedated tests
2022-01-31 20:37:16 +01:00

39 lines
1.0 KiB
Python

"""FinBrain Model"""
__docformat__ = "numpy"
import logging
import pandas as pd
import requests
from gamestonk_terminal.decorators import log_start_end
logger = logging.getLogger(__name__)
@log_start_end(log=logger)
def get_sentiment(ticker: str) -> pd.DataFrame:
"""Gets Sentiment analysis provided by FinBrain's API [Source: finbrain]
Parameters
----------
ticker : str
Ticker to get the sentiment analysis from
Returns
-------
DataFrame()
Empty if there was an issue with data retrieval
"""
result = requests.get(f"https://api.finbrain.tech/v0/sentiments/{ticker}")
sentiment = pd.DataFrame()
if result.status_code == 200:
if "sentimentAnalysis" in result.json():
sentiment = pd.DataFrame.from_dict(
result.json()["sentimentAnalysis"], orient="index"
)
sentiment.index = pd.to_datetime(sentiment.index).to_pydatetime()
sentiment.index.name = "date"
sentiment.columns = ["Sentiment Analysis"]
return sentiment