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
