mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +00:00
b71abcfbf4
* quick econ fix * black * keys and feature flags * terminal name 👀 * some more replacements * some more replacements * edit pyproject * gst -> openbb * add example portfolios back to git * Update api from gst * sorry. skipping some tests * another round of names * another round of test edits * Missed some .gst refs and update timezone * water mark stuff * Fixing Names in terminal.spec and name of GTFF_DEFAULTS to OBBFF_DEFAULTS * fix more GST to OpenBB Terminal * Logging : merge conflicts with main * Revert wrong files Co-authored-by: Andrew <andrew.kenreich@gmail.com> Co-authored-by: DidierRLopes <dro.lopes@campus.fct.unl.pt> Co-authored-by: Chavithra PARANA <chavithra@gmail.com>
89 lines
1.9 KiB
Python
89 lines
1.9 KiB
Python
"""Google Model"""
|
|
__docformat__ = "numpy"
|
|
|
|
import logging
|
|
|
|
import pandas as pd
|
|
from pytrends.request import TrendReq
|
|
|
|
from openbb_terminal.decorators import log_start_end
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def get_mentions(ticker: str) -> pd.DataFrame:
|
|
"""Get interest over time from google api [Source: google]
|
|
|
|
Parameters
|
|
----------
|
|
ticker : str
|
|
Stock ticker
|
|
|
|
Returns
|
|
-------
|
|
pd.DataFrame
|
|
Dataframe of interest over time
|
|
"""
|
|
pytrend = TrendReq()
|
|
pytrend.build_payload(kw_list=[ticker])
|
|
return pytrend.interest_over_time()
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def get_regions(ticker: str) -> pd.DataFrame:
|
|
"""Get interest by region from google api [Source: google]
|
|
|
|
Parameters
|
|
----------
|
|
ticker : str
|
|
Ticker to look at
|
|
|
|
Returns
|
|
-------
|
|
pd.DataFrame
|
|
Dataframe of interest by region
|
|
"""
|
|
pytrend = TrendReq()
|
|
pytrend.build_payload(kw_list=[ticker])
|
|
return pytrend.interest_by_region()
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def get_queries(ticker: str) -> pd.DataFrame:
|
|
"""Get related queries from google api [Source: google]
|
|
|
|
Parameters
|
|
----------
|
|
ticker : str
|
|
Stock ticker to compare
|
|
|
|
Returns
|
|
-------
|
|
pd.DataFrame
|
|
Dataframe of related queries
|
|
"""
|
|
pytrend = TrendReq()
|
|
pytrend.build_payload(kw_list=[ticker])
|
|
return pytrend.related_queries()
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def get_rise(ticker: str) -> pd.DataFrame:
|
|
"""Get top rising related queries with this stock's query [Source: google]
|
|
|
|
Parameters
|
|
----------
|
|
ticker : str
|
|
Stock ticker
|
|
|
|
Returns
|
|
-------
|
|
pd.DataFrame
|
|
Dataframe containing rising related queries
|
|
"""
|
|
pytrend = TrendReq()
|
|
pytrend.build_payload(kw_list=[ticker])
|
|
df_related_queries = pytrend.related_queries()
|
|
return df_related_queries[ticker]["rising"]
|