def do_busca_carteira():
from src.crawler_cei import CrawlerCei
crawler_cei = CrawlerCei(headless=True)
df_carteira = crawler_cei.busca_carteira(data=data_referencia)
df_to_csv(df_carteira,
WORK_DIR + nome_com_referencia('carteira_cei.txt'),
colunas=df_carteira.columns,
header=True)
def test_busca_carteira(self):
# Permite separar os arquivos de testes por CPF
directory = os.getenv('DIR_TESTES_CEI','./temp/tests_cei_carteira/') + os.environ['CPF'] + '/'
if not os.path.exists(directory):
os.makedirs(directory)
crawler_cei = CrawlerCei(headless=True, directory=directory, debug=True)
carteira = crawler_cei.busca_carteira(datetime.date(2020,12,31))
verifica_carteira(carteira, directory)