import pandas as pd

from app.entity.product.ChartDataBean import *

class StrategyCalculatorDataBean:
    
    def __init__(self, code='StrategyCalculatorDataBean', entityType='StrategyCalculatorDataBean',
                 name='None', marketInfo=None, scenario=None, 
                 portfolio=[], scenarioResults=[], 
                 axisMin=0., axisMax=0., chartDataBean = None):
        self.code = code
        self.entityType = 'StrategyCalculatorDataBean'
        
        self.name = name
        
        self.marketInfo = marketInfo
        self.scenario = scenario
        self.portfolio = portfolio
        self.scenarioResults = scenarioResults
        self.axisMin = float(axisMin)
        self.axisMax = float(axisMax)
        if chartDataBean is not None:
            self.chartDataBean = chartDataBean
        else:
            self.chartDataBean = ChartDataBean()
        

    def serialize(self):
        return {
                "code": self.code,
                "entityType": self.entityType,
                "name": self.name,
                "marketInfo": self.marketInfo,
                "scenario": self.scenario,
                "portfolio": self.portfolio,
                "scenarioResults": self.scenarioResults,
                "axisMin": self.axisMin,
                "axisMax": self.axisMax,
                "chartDataBean": self.chartDataBean,
                }