debug_nifty50.py

#!/usr/bin/env python3
"""
Debug script to test Nifty 50 data loading
"""

import sys
import os
sys.path.append(os.path.join(os.path.dirname(__file__), 'src'))

from nifty50_loader import get_nifty50_data
import pandas as pd

def debug_nifty50_data():
    """Debug the Nifty 50 data loading"""
    print("=== DEBUGGING NIFTY 50 DATA LOADING ===")
    
    # Test with different day ranges
    for days in [7, 30, 90]:
        print(f"\nTesting with {days} days:")
        
        try:
            stock_data = get_nifty50_data(days=days)
            print(f"Total stocks returned: {len(stock_data)}")
            
            # Check which stocks have data
            valid_stocks = []
            empty_stocks = []
            
            for symbol, data in stock_data.items():
                if data.empty:
                    empty_stocks.append(symbol)
                else:
                    valid_stocks.append(symbol)
                    print(f"  {symbol}: {data.shape[0]} rows, columns: {list(data.columns)[:5]}...")
            
            print(f"Valid stocks: {len(valid_stocks)}")
            print(f"Empty stocks: {len(empty_stocks)}")
            
            if valid_stocks:
                print(f"First valid stock ({valid_stocks[0]}) sample:")
                sample_data = stock_data[valid_stocks[0]]
                print(f"  Shape: {sample_data.shape}")
                print(f"  Columns: {list(sample_data.columns)}")
                print(f"  Date range: {sample_data.index[0]} to {sample_data.index[-1]}")
                print(f"  Close price range: {sample_data['Close'].min():.2f} to {sample_data['Close'].max():.2f}")
            
        except Exception as e:
            print(f"Error with {days} days: {str(e)}")
            import traceback
            traceback.print_exc()

if __name__ == "__main__":
    debug_nifty50_data()