data.py 文件源码

def load_data(self, pcontract, dt_start=None, dt_end=None):
        """ ??????????.

        Args:
            pcontract (PContract): ????

        Returns:
            DataFrame. 

        Raises:
            FileDoesNotExist
        """
        if pcontract.contract.exch_type == 'stock':
            import tushare as ts
            # ??tushare??
            print "load stock data with tushare..." 
            data = ts.get_hist_data(pcontract.contract.code)
            return process_tushare_data(data)
        else:
            # ????
            fname = ''.join([str(pcontract), ".csv"])
            try:
                data = pd.read_csv(fname, index_col=0, parse_dates=True)
                assert data.index.is_unique
            except Exception:
                #print u"**Warning: File \"%s\" doesn't exist!"%fname
                raise FileDoesNotExist(file=fname)
            else:
                return data