Source code for portfolio_toolkit.asset.market.market_asset

from dataclasses import dataclass, field
from typing import Any, Dict, Optional

import pandas as pd




@dataclass
class MarketAsset:
    ticker: str
    prices: pd.Series
    info: Dict
    currency: Optional[str] = None

    # Campos derivados que no se pasan al constructor
    sector: str = field(init=False)
    country: str = field(init=False)

    def __post_init__(self):
        self.sector = self.info.get("sector", "Unknown")
        self.country = self.info.get("country", "Unknown")
        self.currency = self.currency or self.info.get("currency", "Unknown")



    def to_dict(self) -> Dict[str, Any]:
        """Convert to a dictionary representation."""
        return {
            "ticker": self.ticker,
            "sector": self.sector,
            "country": self.country,
            "currency": self.currency,
        }


    def __repr__(self):
        return (
            f"MarketAsset(ticker={self.ticker}, sector={self.sector}, currency={self.currency}, "
            f"prices_length={len(self.prices)}, info_keys={list(self.info.keys())})"
        )