from urllib.request import Request, urlopen
import pandas as pd
import os
import matplotlib.pyplot as plt
import seaborn as sns
if(os.path.isfile('ipca.csv')):
df = pd.read_csv('ipca.csv', encoding = "utf-8", delimiter=';', index_col=0)
print()
print('Dados extraídos do arquivo local.')
print()
else:
req = Request('https://ihack.com.br/dados/ipca.html', headers={'User-Agent': 'Mozilla/5.0'})
pagina = urlopen(req).read()
ipca = pd.read_html(pagina)
ipca = ipca[0]
ipca = ipca[[0,3]]
ipca = ipca.iloc[1:]
ipca[3] = ipca[3].str.replace(',', '.')
df = ipca[3].astype(float) / 10000
df = pd.concat([ipca, df], axis=1)
df.columns = ['Data', 'IPCA_OLD', 'IPCA']
df = df[['Data','IPCA']]
df.set_index(['Data'], inplace=True)
del(ipca)
# Reverse order
df = df.iloc[::-1]
df.to_csv('ipca.csv', sep=';', encoding='utf-8')
print()
print('Dados extraídos da Internet.')
print()
print(df)
sns.set_style('whitegrid')
df['IPCA'].plot()
plt.legend(loc=7, bbox_to_anchor=(1.0, 0.5))
plt.xlabel('Date')
plt.ylabel('IPCA')
plt.title('IPCA Mensal')
plt.show()