def test_df_to_csv(self): df_original = get_operations_dataframe() df_to_csv(df_original, 'df_to_csv_testing.txt') df_lido = get_operations_dataframe('df_to_csv_testing.txt') assert df_lido.equals(df_original)
def test_calcula_precos_medios_do_dropbox(self): from src.dropbox_files import download_dropbox_file download_dropbox_file() df = get_operations_dataframe() precos_medios_de_compra = calcula_precos_medio_de_compra(df) assert type(precos_medios_de_compra) is dict
def do_busca_trades_e_faz_merge_operacoes(): from src.crawler_cei import CrawlerCei crawler_cei = CrawlerCei(headless=True) df_cei = crawler_cei.busca_trades() from src.dropbox_files import download_dropbox_file download_dropbox_file() df = get_operations_dataframe() df = merge_operacoes(df, df_cei) df_to_csv(df, OPERATIONS_FILEPATH) upload_dropbox_file(OPERATIONS_FILEPATH, os.environ['DROPBOX_FILE_LOCATION'])
def test_relatorio_txt(self): from src.dropbox_files import download_dropbox_file download_dropbox_file() df = get_operations_dataframe() df = df.tail(150) calcula_custodia(df) calculo_ir = CalculoIr(df=df) calculo_ir.calcula() with open("relatorio.txt", "w") as relatorio: relatorio.write(relatorio_txt(calculo_ir))
def test_relatorio_html(self): from src.dropbox_files import download_dropbox_file download_dropbox_file() df = get_operations_dataframe() df = df.tail(150) calcula_custodia(df) calculo_ir = CalculoIr(df=df) calculo_ir.calcula() from py_w3c.validators.html.validator import HTMLValidator assert HTMLValidator().validate_fragment(relatorio_html(calculo_ir))
def do_calculo_ir(): from src.dropbox_files import download_dropbox_file download_dropbox_file() df = get_operations_dataframe() from src.stuff import calcula_custodia calcula_custodia(df) calculo_ir = CalculoIr(df=df) calculo_ir.calcula() print(relatorio_txt(calculo_ir)) envia_relatorio_html_por_email(assunto(calculo_ir), relatorio_html(calculo_ir))
def do_calculo_ir(): from src.dropbox_files import download_dropbox_file download_dropbox_file() df = get_operations_dataframe() from src.stuff import calcula_custodia calcula_custodia(df) calculo_ir = CalculoIr(df=df) calculo_ir.calcula() print(relatorio_txt(calculo_ir)) envia_relatorio_html_por_email( 'Calculo de IR - ' + calculo_ir.mes_do_relatorio + ' - CPF: ' + os.environ['CPF'] + ' - ' + datetime.now().strftime("%H:%M:%S"), relatorio_html(calculo_ir))
def test_calcula_custodia(self): from src.dropbox_files import download_dropbox_file download_dropbox_file() df = get_operations_dataframe() data = datetime.datetime.now().date() custodia = calcula_custodia(df, data) assert type(custodia) is pd.DataFrame assert 'preco_medio_compra' in custodia.columns assert 'ticker' in custodia.columns assert 'qtd' in custodia.columns assert 'preco_atual' in custodia.columns assert 'valorizacao' in custodia.columns assert 'valor' in custodia.columns assert 'tipo' in custodia.columns assert 'data_primeira_compra' in custodia.columns
def test_deve_criar_dataframe_vazio_se_arquivo_vazio(self): df = get_operations_dataframe('arquivo_invalido.txt') self.assertListEqual(list(df.columns), todas_as_colunas())