from clr import AddReference
AddReference("System.Core")
AddReference("QuantConnect.Common")
AddReference("QuantConnect.Algorithm")

from System import *
from QuantConnect import *
from QuantConnect.Algorithm import QCAlgorithm
from QuantConnect.Data import SubscriptionDataSource
from QuantConnect.Python import PythonData

from datetime import datetime, timedelta
import json

class ExuberAlgorithm(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2021, 5, 1)
        
        self.AddEquity('SPY', Resolution.Hour)
        CustomDataSource.set_algo(self)
        
        self.symbol = self.AddData(CustomDataSource, "CustomDataSource", Resolution.Hour).Symbol

    def OnData(self, data):
        if data.ContainsKey(self.symbol):
            self.Log(f'Data Received. Value: {data[self.symbol].Value}')

class CustomDataSource(PythonData):
    algo = None
    
    @staticmethod
    def set_algo(algo):
        CustomDataSource.algo = algo
    

    def GetSource(self, config, date, isLive):
        CustomDataSource.algo.Log(f"GetSource at {CustomDataSource.algo.Time} {date}")
        source = "https://www.dropbox.com/s/93f6dz8yqh8vr62/custom-data-test.csv?dl=1"
        return SubscriptionDataSource(source, SubscriptionTransportMedium.RemoteFile)


    def Reader(self, config, line, date, isLive):
        index = CustomDataSource()
        index.Symbol = config.Symbol
        
        data = line.split(',')
        index.Time = datetime.strptime(data[0], '%m/%d/%Y %H:%M')
        CustomDataSource.algo.Log(f"Reader at {CustomDataSource.algo.Time}")
        index.Value = 1.0  # Random value
        return index