def __init__(self, *args, **kw): self.symbols.sort() super(PortfolioModel, self).__init__(*args, **kw) symbol_list = price_utils.load_symbols_from_table(dbfilename=self.dbfilename)['symbol'].tolist() self.symbol_pool = [Symbol(symbol=symb) for symb in symbol_list] #self.symbols2 = [Symbol(symbol=s) for s in init_symbols] self.plot = self._create_plot_component()
def _dbfilename_changed(self, event): #reload symbols # TODO: what's the right way to do this? symbol_list = price_utils.load_symbols_from_table(dbfilename=self.dbfilename)['symbol'].tolist() #print self.dbfilename, symbol_list if len(symbol_list) > 2: #print "New symbols: ", symbol_list self.trait_view('symbols_item').editor.values = symbol_list #print "Symbol list setting: ", self.trait_view('symbols_item').editor.values self.trait_view('traits_view').updated = True self.symbols = symbol_list[:2] #self.trait_view( 'symbols' ).updated = True return
(0.8, 0.4, 0.5, 0.7), (0.5, 0.8, 0.8, 0.7), (0.1, 0.8, 0.0, 0.7), (0.4, 0.7, 0.8, 0.7), (0.0, 0.5, 0.8, 0.7), (0.6, 0.5, 0.8, 0.7), (0.2, 0.1, 0.8, 0.7), (0.8, 0.4, 0.5, 0.7), (0.7, 0.0, 0.1, 0.7)] #db = "data/stocks.db" db = "data/bonds.db" # TODO: Get rid of "tolist" requirement .. seems messy all_symbols = price_utils.load_symbols_from_table(dbfilename=db)['symbol'].tolist() print all_symbols # Some other ways to do symbols ... #init_symbols = ["CSCO", "AAPL", "IBM", "MSFT", "GE", "WFC", "RIG", "T", "AA", "CAT"] #init_symbols = ["AAPL", "GOOG", "EOG", "YUM", "AA", "BA", "COP"] init_symbols = ["FSTYX", "LALDX", "PIMSX", "USSBX", "BSV"] init_symbols.sort() def color_tuple_to_int(rgb_tuple): # TODO: This is a major hack to return a hex string so that the bg_color for the rows # will work properly. return "#"+"".join(map(chr, (int(x*255) for x in rgb_tuple))).encode('hex') class Symbol(HasTraits): symbol = Str