from pandas_datareader.data import YahooDailyReader, GoogleDailyReader
import time, datetime


class YahooDailyReader_all(YahooDailyReader):

    def _get_params(self, symbol):
        self.start = datetime.datetime(1970, 1, 2)
        unix_start = int(time.mktime(self.start.timetuple()))
        day_end = self.end.replace(hour=23, minute=59, second=59)
        unix_end = int(time.mktime(day_end.timetuple()))

        params = {
            'period1': unix_start,
            'period2': unix_end,
            'interval': self.interval,
            'events': self.service,
            'crumb': self.crumb
        }
        return params


class GoogleDailyReader_all(GoogleDailyReader):

    def _get_params(self, symbol):
        self.start = datetime.datetime(1980, 1, 2)
        params = {
            'q': symbol,
            'startdate': self.start.strftime('%b %d, %Y'),
            'enddate': self.end.strftime('%b %d, %Y'),
            'output': "csv"
        }
        return params

def get_alldata_yahoo(*args, **kwargs):
    df  = YahooDailyReader_all(*args, **kwargs).read()
    if df is not None:
        df.drop('Adj Close',axis=1, inplace=True)
    return df

def get_alldata_google(*args, **kwargs):
    return GoogleDailyReader_all(*args, **kwargs).read()