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")