class Summary(): def __init__(self, filename='stocks.txt', **kwargs): self.c_dir = os.path.dirname(os.path.abspath(__file__)) self.base_dir = os.path.join(self.c_dir, '..') self.data_dir = os.path.join(self.base_dir, 'data') self.stock_list = os.path.join(self.data_dir, filename) self.aggregator = Aggregator(self.stock_list, self.data_dir) def aggregate(self, filename="stocks.txt", range=1, sortkey='Ratio', screening_key=None, ascending=False, history=False): result = self.aggregator.summarize(range=range, screening_key=screening_key, sortkey=sortkey, ascending=ascending) p = os.path.join( os.path.dirname(os.path.abspath(__file__)), '..', 'data', filename) result.to_csv(p, sep="\t", index_label="Code") if history: today = datetime.datetime.now().strftime('%Y%m%d') p = os.path.join( os.path.dirname(os.path.abspath(__file__)), '..', 'data', 'history', "".join([filename, '.', today, '.csv'])) result.to_csv(p, sep="\t", index_label="Code")
class Summary(): def __init__(self, filename='stocks.txt', **kwargs): self.c_dir = os.path.dirname(os.path.abspath(__file__)) self.base_dir = os.path.join(self.c_dir, '..') self.data_dir = os.path.join(self.base_dir, 'data') self.stock_list = os.path.join(self.data_dir, filename) self.aggregator = Aggregator(self.stock_list, self.data_dir) def aggregate(self, filename="stocks.txt", range=1, sortkey='Ratio', screening_key=None, ascending=False, history=False): result = self.aggregator.summarize(range=range, screening_key=screening_key, sortkey=sortkey, ascending=ascending) p = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', 'data', filename) result.to_csv(p, sep="\t", index_label="Code") if history: today = datetime.datetime.now().strftime('%Y%m%d') p = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', 'data', 'history', "".join([filename, '.', today, '.csv'])) result.to_csv(p, sep="\t", index_label="Code")
def test_summarize(): c_dir = os.path.dirname(os.path.abspath(__file__)) base_dir = os.path.join(c_dir, '..') stock_list = os.path.join(base_dir, 'data', 'stocks.txt') data_dir = os.path.join(base_dir, 'test') aggregator = Aggregator(stock_list, data_dir) result = aggregator.summarize() eq_('stocks.txt', os.path.basename(aggregator.stock_list)) eq_('test', os.path.basename(aggregator.data_dir)) eq_(True, result.empty) return result