AllStreet / scripts /download_data.py
kauabarros-24
CHORE: Generate most datas
43056e4
import yfinance as yf
import pandas as pd
import os
def download_stock_data(ticker, start, end, save_path):
os.makedirs('data', exist_ok=True)
print(f"Baixando dados de {ticker}...")
df = yf.download(ticker, start=start, end=end, progress=False)
df.to_csv(save_path)
print(f"Dados salvos em {save_path}")
print(f"Shape: {df.shape}")
print(f"Colunas: {list(df.columns)}")
return df
if __name__ == "__main__":
download_stock_data("PETR4.SA", "2000-01-01", "2025-01-01", "data/raw_PETR4_SA.csv")