def test_trendet():
"""
This function checks that main functions of trendet work properly.
"""
print(trendet.__author__)
print(trendet.__version__)
params = [
{
'stock': 'BBVA',
'country': 'Spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'window_size': 5,
'trend_limit': 2,
'labels': ['A', 'B'],
'identify': 'both'
},
{
'stock': 'BBVA',
'country': 'Spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'window_size': 5,
'trend_limit': 2,
'labels': None,
'identify': 'up',
},
{
'stock': 'BBVA',
'country': 'Spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'window_size': 5,
'trend_limit': 2,
'labels': ['A', 'B'],
'identify': 'up',
},
{
'stock': 'BBVA',
'country': 'Spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'window_size': 5,
'trend_limit': 2,
'labels': ['A', 'B'],
'identify': 'down',
},
{
'stock': 'BBVA',
'country': 'Spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'window_size': 5,
'trend_limit': 2,
'labels': None,
'identify': 'down',
}
]
stocks = get_stocks_list(country='Spain')
for stock in stocks[:25]:
obj = {
'stock': stock,
'country': 'Spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'window_size': 5,
'trend_limit': 5,
'labels': None,
'identify': 'both',
}
params.append(obj)
for param in params:
trendet.identify_trends(stock=param['stock'],
country=param['country'],
from_date=param['from_date'],
to_date=param['to_date'],
window_size=param['window_size'],
trend_limit=param['trend_limit'],
labels=param['labels'],
identify=param['identify'])
trendet.identify_all_trends(stock=param['stock'],
country=param['country'],
from_date=param['from_date'],
to_date=param['to_date'],
window_size=param['window_size'],
identify=param['identify'])
df = get_stock_historical_data(stock='REP',
country='Spain',
from_date='01/01/2018',
to_date='01/01/2019')
params = [
{
'column': 'Close',
'window_size': 5,
'identify': 'both'
},
{
'column': 'Close',
'window_size': 5,
'identify': 'up'
},
{
'column': 'Close',
'window_size': 5,
'identify': 'down'
},
]
for param in params:
trendet.identify_df_trends(df=df,
column=param['column'],
window_size=param['window_size'],
identify=param['identify'])
def test_stocks_errors():
"""
This function raises errors on stock retrieval functions
"""
try:
retrieve_stocks(test_mode=None)
except:
pass
try:
retrieve_stock_countries(test_mode=None)
except:
pass
params = [
{
'country': ['error']
},
{
'country': 'error'
},
]
for param in params:
try:
investpy.get_stocks(country=param['country'])
except:
pass
try:
investpy.get_stocks_list(country=param['country'])
except:
pass
params = [
{
'country': ['error'],
'columns': None,
'as_json': False
},
{
'country': 'spain',
'columns': None,
'as_json': 'error'
},
{
'country': 'spain',
'columns': 0,
'as_json': True
},
{
'country': 'spain',
'columns': ['error'],
'as_json': False
},
]
for param in params:
try:
investpy.get_stocks_dict(country=param['country'],
columns=param['columns'],
as_json=param['as_json'])
except:
pass
params = [
{
'stock': 'FERR',
'country': 'spain',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': None,
'country': 'spain',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': None,
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': ['error'],
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'greece',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'as_json': 'error',
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'as_json': True,
'order': 'error',
'debug': True
},
{
'stock': 'error',
'country': 'spain',
'as_json': True,
'order': 'ascending',
'debug': True
},
{
'stock': ['error'],
'country': 'spain',
'as_json': True,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'as_json': True,
'order': 'ascending',
'debug': 'error'
},
]
for param in params:
try:
investpy.get_stock_recent_data(stock=param['stock'],
country=param['country'],
as_json=param['as_json'],
order=param['order'],
debug=param['debug'])
except:
pass
params = [
{
'stock': 'FERR',
'country': 'spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': None,
'country': 'spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': None,
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': ['error'],
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'greece',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': 'error',
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': False,
'order': 'error',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': 'error',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': '01/01/2019',
'to_date': 'error',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'error',
'country': 'spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': ['error'],
'country': 'spain',
'from_date': '01/01/2018',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': '01/01/1999',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': '01/01/1900',
'to_date': '01/01/1950',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': '01/01/1950',
'to_date': '01/01/2019',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': '01/01/2019',
'to_date': '01/01/1999',
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'stock': 'BBVA',
'country': 'spain',
'from_date': '01/01/2019',
'to_date': '01/03/2019',
'as_json': False,
'order': 'ascending',
'debug': 'error'
},
]
for param in params:
try:
investpy.get_stock_historical_data(stock=param['stock'],
country=param['country'],
from_date=param['from_date'],
to_date=param['to_date'],
as_json=param['as_json'],
order=param['order'],
debug=param['debug'])
except:
pass
params = [
{
'stock': None,
'country': 'spain',
'language': 'spanish'
},
{
'stock': 'BBVA',
'country': None,
'language': 'spanish'
},
{
'stock': 'BBVA',
'country': 'greece',
'language': 'spanish'
},
{
'stock': 'BBVA',
'country': 'spain',
'language': 'error'
},
{
'stock': 'error',
'country': 'spain',
'language': 'spanish'
},
{
'stock': ['error'],
'country': 'spain',
'language': 'spanish'
},
]
for param in params:
try:
investpy.get_stock_company_profile(stock=param['stock'],
country=param['country'],
language=param['language'])
except:
pass
params = [
{
'stock': None,
'country': 'spain',
},
{
'stock': ['error'],
'country': 'spain',
},
{
'stock': 'bbva',
'country': ['error'],
},
{
'stock': 'bbva',
'country': 'error',
},
{
'stock': 'error',
'country': 'spain',
},
{
'stock': 'ALUA',
'country': 'argentina',
},
]
for param in params:
try:
investpy.get_stock_dividends(stock=param['stock'],
country=param['country'])
except:
pass
params = [
{
'by': None,
'value': 'BBVA',
},
{
'by': ['error'],
'value': 'BBVA',
},
{
'by': 'error',
'value': 'BBVA',
},
{
'by': 'name',
'value': None,
},
{
'by': 'name',
'value': ['error'],
},
{
'by': 'isin',
'value': 'BBVA',
},
]
for param in params:
try:
investpy.search_stocks(by=param['by'], value=param['value'])
except:
pass
def test_investpy_stocks():
"""
This function checks that stock data retrieval functions listed in investpy work properly.
"""
params = [
{
'country': 'spain',
},
{
'country': None,
},
]
for param in params:
investpy.get_stocks(country=param['country'])
investpy.get_stocks_list(country=param['country'])
params = [
{
'country': None,
'columns': ['full_name', 'name'],
'as_json': True
},
{
'country': None,
'columns': ['full_name', 'name'],
'as_json': False
},
{
'country': 'spain',
'columns': ['full_name', 'name'],
'as_json': True
},
{
'country': 'spain',
'columns': ['full_name', 'name'],
'as_json': False
},
{
'country': 'spain',
'columns': None,
'as_json': False
},
]
for param in params:
investpy.get_stocks_dict(country=param['country'],
columns=param['columns'],
as_json=param['as_json'])
investpy.get_stock_countries()
params = [
{
'as_json': True,
'order': 'ascending',
},
{
'as_json': False,
'order': 'ascending',
},
{
'as_json': True,
'order': 'descending',
},
{
'as_json': False,
'order': 'descending',
},
]
for param in params:
investpy.get_stock_recent_data(stock='BBVA',
country='spain',
as_json=param['as_json'],
order=param['order'],
interval='Daily')
investpy.get_stock_historical_data(stock='BBVA',
country='spain',
from_date='01/01/1990',
to_date='01/01/2019',
as_json=param['as_json'],
order=param['order'],
interval='Daily')
for value in ['spanish', 'english']:
investpy.get_stock_company_profile(stock='BBVA',
country='spain',
language=value)
params = [
{
'stock': 'bbva',
'country': 'spain',
'as_json': False
},
{
'stock': 'bbva',
'country': 'spain',
'as_json': True
},
{
'stock': 'HSBK',
'country': 'kazakhstan',
'as_json': False
}
]
for param in params:
investpy.get_stock_information(stock=param['stock'], country=param['country'], as_json=param['as_json'])
params = [
{
'country': 'spain',
'as_json': True,
'n_results': 50
},
{
'country': 'united states',
'as_json': False,
'n_results': 50
},
{
'country': 'bosnia',
'as_json': False,
'n_results': 50
},
{
'country': 'palestine',
'as_json': False,
'n_results': 50
},
{
'country': 'dubai',
'as_json': False,
'n_results': 50
},
{
'country': 'ivory coast',
'as_json': False,
'n_results': 50
}
]
for param in params:
investpy.get_stocks_overview(country=param['country'], as_json=param['as_json'], n_results=param['n_results'])
params = [
{
'stock': 'bbva',
'country': 'spain'
},
{
'stock': 'entel',
'country': 'chile'
}
]
for param in params:
investpy.get_stock_dividends(stock=param['stock'], country=param['country'])
params = [
{
'stock': 'bbva',
'country': 'spain',
'summary_type': 'balance_sheet',
'period': 'annual'
},
{
'stock': 'aapl',
'country': 'united states',
'summary_type': 'income_statement',
'period': 'quarterly'
},
{
'stock': 'barc',
'country': 'united kingdom',
'summary_type': 'cash_flow_statement',
'period': 'annual'
}
]
for param in params:
investpy.get_stock_financial_summary(stock=param['stock'],
country=param['country'],
summary_type=param['summary_type'],
period=param['period'])
investpy.search_stocks(by='name', value='BBVA')
def test_investpy_stocks():
"""
This function checks that stock data retrieval functions listed in investpy work properly.
"""
params = [
{
'country': 'spain',
},
{
'country': None,
},
]
for param in params:
investpy.get_stocks(country=param['country'])
investpy.get_stocks_list(country=param['country'])
params = [
{
'country': None,
'columns': ['full_name', 'name'],
'as_json': True
},
{
'country': None,
'columns': ['full_name', 'name'],
'as_json': False
},
{
'country': 'spain',
'columns': ['full_name', 'name'],
'as_json': True
},
{
'country': 'spain',
'columns': ['full_name', 'name'],
'as_json': False
},
{
'country': 'spain',
'columns': None,
'as_json': False
},
]
for param in params:
investpy.get_stocks_dict(country=param['country'],
columns=param['columns'],
as_json=param['as_json'])
investpy.get_stock_countries()
params = [
{
'as_json': True,
'order': 'ascending',
'debug': False
},
{
'as_json': False,
'order': 'ascending',
'debug': True
},
{
'as_json': True,
'order': 'descending',
'debug': False
},
{
'as_json': False,
'order': 'descending',
'debug': False
},
]
for param in params:
investpy.get_stock_recent_data(stock='BBVA',
country='spain',
as_json=param['as_json'],
order=param['order'],
debug=param['debug'])
investpy.get_stock_historical_data(stock='BBVA',
country='spain',
from_date='01/01/1990',
to_date='01/01/2019',
as_json=param['as_json'],
order=param['order'],
debug=param['debug'])
for value in ['spanish', 'english']:
investpy.get_stock_company_profile(stock='BBVA',
country='spain',
language=value)
investpy.get_stock_dividends(stock='BBVA', country='spain')
investpy.search_stocks(by='name', value='BBVA')
dfs.append(df)
metas.append(meta)
except Exception as e:
print(f'ticker {ticker} data retreive error...')
print(e)
else:
print(f'ticker {ticker} SUCCESS.....')
time.sleep(13)
#%%
import investpy
# https://investpy.readthedocs.io/_info/introduction.html
# Retrieve all the available stocks as a Python list
stocks = investpy.get_stocks_list()
funds = investpy.get_funds_list()
etfs = investpy.get_etfs_list()
# Retrieve the recent historical data (past month) of a stock as a pandas.DataFrame on ascending date order
df_bbva = investpy.get_stock_recent_data(stock='bbva',
country='spain',
as_json=False,
order='ascending')
# Retrieve the company profile of the introduced stock on english
profile = investpy.get_stock_company_profile(stock='bbva',
country='spain',
language='english')
df_aapl = investpy.get_stock_historical_data(stock='AAPL',
import investpy as inv
acoes_br = inv.get_stocks_list('brazil')
print(acoes_br)
# fazendo uma listagem de todas as ações no banco de dados
""" acoes_br = inv.get_stocks('brazil')
print(acoes_br) """
# buscando uma descrição de uma ação
""" acoes_br = inv.get_stock_company_profile('USIM5', country='brazil')
print(acoes_br) """
# baixando a contação hitorica
""" usim5 = inv.get_stock_historical_data('USIM5', country='brazil', from_date='01/01/1900', to_date='11/07/2020')
print(usim5) #posso acrecentar ['nome da coluna'] para exibir so a conluna """
data = [trace1]
layout = go.Layout()
fig = go.Figure(data=data, layout=layout)
return fig
# ## CRIANDO UMA BARRA LATERAL
barra_lateral = st.sidebar.empty()
st.sidebar.title('Mercado de Ações')
country_select = st.sidebar.selectbox("Selecione o país:", countries)
acoes = ip.get_stocks_list(country=country_select)
stock_select = st.sidebar.selectbox("Selecione o ativo:", acoes)
from_date = st.sidebar.date_input('De:', start_date)
to_date = st.sidebar.date_input('Para:', end_date)
interval_select = st.sidebar.selectbox("Selecione o interval:", intervals)
carregar_dados = st.sidebar.checkbox('Carregar dados')
### elementos centrais da página
st.subheader('Visualização gráfica')
grafico_candle = st.empty()