# region imports
from AlgorithmImports import *
# endregion

class CrawlingBlackDogfish(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2020, 11, 12)  # Set Start Date
        self.SetCash(100000)  # Set Strategy Cash

        # Setup trading framework
        # Transaction and submit/execution rules will use IB models 
        self.SetBrokerageModel(BrokerageName.InteractiveBrokersBrokerage, 
            AccountType.Margin)

        # Configure all algorithm securities
# Uncomment this line and run to replicate runtime error:
# During the algorithm initialization, the following exception has occurred: 
#  Attempted to divide by zero. in BuyingPowerModel.cs:line 110 
        self.SetSecurityInitializer(self.CustomSecurityInitializer)

        self.AddEquity("SPY", Resolution.Minute)
        self.AddEquity("BND", Resolution.Minute)
        self.AddEquity("AAPL", Resolution.Minute)

    def OnData(self, data: Slice):
        if not self.Portfolio.Invested:
            self.SetHoldings("SPY", 0.33)
            self.SetHoldings("BND", 0.33)
            self.SetHoldings("AAPL", 0.33)

    def CustomSecurityInitializer(self, security):
        """
        Define models to be used for securities as they are added to the 
        algorithm's universe.
        """
        # Set the custom buying power model
        security.SetBuyingPowerModel(CustomBuyingPowerModel(self))

################################################################################
class CustomBuyingPowerModel(BuyingPowerModel):
    """
    Custom model to determine if the portfolio has sufficient buying power for 
    a new desired order.
    """
    def __init__(self, algorithm):
        """Initialize the custom buying power model."""
        super().__init__() # now required for 3.0 IDE
        self.algorithm = algorithm

    def HasSufficientBuyingPowerForOrder(self, parameters):
        """Determine if the portfolio has the buying power for an order."""
        # This model will assume that there is always enough buying power
        sufficient_buying_power = HasSufficientBuyingPowerForOrderResult(True)
        return sufficient_buying_power