diff --git a/setup.py b/setup.py index cfa8c05..fe02ef0 100644 --- a/setup.py +++ b/setup.py @@ -5,9 +5,9 @@ setup( name="turingquant", - version="0.1.3", + version="0.1.4", packages=find_packages(), - install_requires=["pandas", "pandas_datareader", "numpy", "matplotlib", "alpha_vantage", "bs4", "plotly"], + install_requires=["pandas", "pandas_datareader", "numpy", "matplotlib", "alpha_vantage", "bs4", "plotly", "yfinance"], author="Grupo Turing", author_email="turing.usp@gmail.com", diff --git a/turingquant/benchmark.py b/turingquant/benchmark.py index 24986b2..eeb2454 100644 --- a/turingquant/benchmark.py +++ b/turingquant/benchmark.py @@ -1,19 +1,20 @@ import pandas as pd import matplotlib.pyplot as plt +from datetime import datetime from matplotlib.ticker import PercentFormatter from pandas_datareader import data -def benchmark(ticker, start: pd.datetime, end: pd.datetime, source='yahoo', plot=True): +def benchmark(ticker, start: datetime, end: datetime, source='yahoo', plot=True): """ Essa função fornece um plot de retorno acumulado de um ativo ao longo de um dado intervalo de tempo, definido pelos parâmetros start e end. Os dados são coletados da API do yahoo, caso haja dados faltantes, os retornos são contabilizados como nulos. ticker[str]: recebe o ticker do papel que será obtido - start[pd.datetime]: início do intervalo + start[datetime]: início do intervalo - end[pd.datetime]: final do interval + end[datetime]: final do interval plot[bool]: (default = True) se True, realiza o plot """ @@ -26,13 +27,13 @@ def benchmark(ticker, start: pd.datetime, end: pd.datetime, source='yahoo', plot return cumulative -def benchmark_ibov(start: pd.datetime, end: pd.datetime, source='yahoo', plot=True): +def benchmark_ibov(start: datetime, end: datetime, source='yahoo', plot=True): """ Essa função produz um plot da evolução do Índice Bovespa ao longo de um dado intervalo, definido pelos parâmetros start e end. - start[pd.datetime]: início do intervalo + start[datetime]: início do intervalo - end[pd.datetime]: final do interval + end[datetime]: final do interval plot[bool]: (default = True) se True, realiza o plot """ @@ -40,13 +41,13 @@ def benchmark_ibov(start: pd.datetime, end: pd.datetime, source='yahoo', plot=Tr return benchmark('^BVSP', start=start, end=end, source=source, plot=plot) -def benchmark_sp500(start: pd.datetime, end: pd.datetime, source='yahoo', plot=True): +def benchmark_sp500(start: datetime, end: datetime, source='yahoo', plot=True): """ Essa função produz um plot da evolução do Índice S&P500 ao longo de um dado intervalo, definido pelos parâmetros start e end. - start[pd.datetime]: início do intervalo + start[datetime]: início do intervalo - end[pd.datetime]: final do interval + end[datetime]: final do interval plot[bool]: (default = True) se True, realiza o plot """