def __init__(self, frame, db):
self.frame = frame
self.__db = db
self.__allStocks = {}
self.__selectedStock = None
self.graph_stock = StockGraph(self.frame, height=100)
self.graph_stock.bind(sequence="<ButtonPress-1>", func=self.graph_stock_mouseSelect)
self.graph_stock.pack(side=BOTTOM, anchor=S, fill=X)
self.label_graph_stock = Label(self.frame, text="StockEntry:")
self.label_graph_stock.pack(side=BOTTOM, anchor=SW)
self.listbox_stocks = Pmw.ScrolledListBox(self.frame,
labelpos='nw',
label_text='Stocks:',
usehullsize=1,
hull_width=200,
vscrollmode='static',
selectioncommand=self.listbox_stocks_SingleClick)
self.listbox_stocks.pack(side=LEFT, anchor=W, fill=Y, padx=5, pady=5)
self.listbox_stocks_populate()
#tabs
self.tabs = Pmw.NoteBook(self.frame)
self.tabs.pack(fill = 'both', expand = 1, padx = 5, pady = 5)
self.tabs.add('word-frequency')
self.tabs.tab('word-frequency').focus_set()
self.tabWordFreq = TabPredictWordFreq(self.tabs.page('word-frequency'), self.__db)
def __init__(self, frame, db):
self.frame = frame
self.__db = db
self._histogramRange = (10*365)
self.histogram_stock = Pmw.Blt.Graph(self.frame, height=100)
self.histogram_stock.xaxis_configure(stepsize=self._histogramRange/20, command=StockHistogram_formatXaxisLabel)
self.histogram_stock.grid_on()
self.histogram_stock.pack(side=BOTTOM, anchor=S, fill=X)
self.graph_stock = StockGraph(self.frame, height=200)
self.graph_stock.bind(sequence="<ButtonPress-1>", func=self.graph_stock_mouseSelect)
self.graph_stock.pack(side=BOTTOM, anchor=S, fill=X)
self.label_graph_stock = Label(self.frame, text="StockEntry:")
self.label_graph_stock.pack(side=BOTTOM, anchor=SW)
self.listbox_stocks = Pmw.ScrolledListBox(self.frame,
labelpos='nw',
label_text='Stocks:',
usehullsize=1,
hull_width=200,
vscrollmode='static',
selectioncommand=self.listbox_stocks_SingleClick)
self.listbox_stocks.pack(side=LEFT, anchor=W, fill=Y, padx=5, pady=5)
self.listBox_webpages = MultiListbox(self.frame, 'Webpages:', (('Date', 10), ('Title', 40)))
self.listBox_webpages.bind("<Double-Button-1>", self.listBox_webpagesDoubleClick)
self.listBox_webpages.pack(side=LEFT,anchor=W, fill=BOTH, expand=YES)
self.listbox_stocks_populate()
class TabStockPredictor():
def __init__(self, frame, db):
self.frame = frame
self.__db = db
self.__allStocks = {}
self.__selectedStock = None
self.graph_stock = StockGraph(self.frame, height=100)
self.graph_stock.bind(sequence="<ButtonPress-1>", func=self.graph_stock_mouseSelect)
self.graph_stock.pack(side=BOTTOM, anchor=S, fill=X)
self.label_graph_stock = Label(self.frame, text="StockEntry:")
self.label_graph_stock.pack(side=BOTTOM, anchor=SW)
self.listbox_stocks = Pmw.ScrolledListBox(self.frame,
labelpos='nw',
label_text='Stocks:',
usehullsize=1,
hull_width=200,
vscrollmode='static',
selectioncommand=self.listbox_stocks_SingleClick)
self.listbox_stocks.pack(side=LEFT, anchor=W, fill=Y, padx=5, pady=5)
self.listbox_stocks_populate()
#tabs
self.tabs = Pmw.NoteBook(self.frame)
self.tabs.pack(fill = 'both', expand = 1, padx = 5, pady = 5)
self.tabs.add('word-frequency')
self.tabs.tab('word-frequency').focus_set()
self.tabWordFreq = TabPredictWordFreq(self.tabs.page('word-frequency'), self.__db)
def listbox_stocks_populate(self):
self._allStockNames = self.__db.getAllStockNames()
self.listbox_stocks.setlist(self._allStockNames)
def listbox_stocks_SingleClick(self):
self.__selectedStock = self._allStockNames[int(self.listbox_stocks.curselection()[0])]
self.tabWordFreq.setSelectedStock(self.__selectedStock)
self.graph_stock_populate(self.__selectedStock)
def graph_stock_populate(self, stockname):
self.__allStocks = self.__db.getAllStocks(stockname)
vector_x = []
vector_y = []
for stock in sorted(self.__allStocks.itervalues()):
vector_x.append(stock['date'].toordinal())
vector_y.append(int(stock['shareprice_closing']*100))
self.graph_stock.line_create(stockname, xdata=vector_x, ydata=vector_y, symbol='')
self.graph_stock.legend_configure(self.graph_stock.legend_get(0), hide=True)
def graph_stock_mouseSelect(self, event):
dicto = self.graph_stock.element_closest(event.x, event.y, halo=1000, interpolate=0)
if dicto != None:
stock = self.__allStocks[date.fromordinal(int(dicto['x']))]
labelText = "StockEntry: " + str(stock['date']) + " value: " + str(stock['shareprice_closing']) + " SEK"
self.label_graph_stock.config(text=labelText)
(Y0, Y1) = self.graph_stock.yaxis_limits()
self.graph_stock.marker_create("line", name="marking line", dashes="", linewidth=2,
coords=(dicto['x'], Y0, dicto['x'], Y1))
class TabStockBrowser():
def __init__(self, frame, db):
self.frame = frame
self.__db = db
self._histogramRange = (10*365)
self.histogram_stock = Pmw.Blt.Graph(self.frame, height=100)
self.histogram_stock.xaxis_configure(stepsize=self._histogramRange/20, command=StockHistogram_formatXaxisLabel)
self.histogram_stock.grid_on()
self.histogram_stock.pack(side=BOTTOM, anchor=S, fill=X)
self.graph_stock = StockGraph(self.frame, height=200)
self.graph_stock.bind(sequence="<ButtonPress-1>", func=self.graph_stock_mouseSelect)
self.graph_stock.pack(side=BOTTOM, anchor=S, fill=X)
self.label_graph_stock = Label(self.frame, text="StockEntry:")
self.label_graph_stock.pack(side=BOTTOM, anchor=SW)
self.listbox_stocks = Pmw.ScrolledListBox(self.frame,
labelpos='nw',
label_text='Stocks:',
usehullsize=1,
hull_width=200,
vscrollmode='static',
selectioncommand=self.listbox_stocks_SingleClick)
self.listbox_stocks.pack(side=LEFT, anchor=W, fill=Y, padx=5, pady=5)
self.listBox_webpages = MultiListbox(self.frame, 'Webpages:', (('Date', 10), ('Title', 40)))
self.listBox_webpages.bind("<Double-Button-1>", self.listBox_webpagesDoubleClick)
self.listBox_webpages.pack(side=LEFT,anchor=W, fill=BOTH, expand=YES)
self.listbox_stocks_populate()
def listbox_stocks_populate(self):
self._allStockNames = self.__db.getAllStockNames()
self.listbox_stocks.setlist(self._allStockNames)
def graph_stock_populate(self, stockname):
self.__allStocks = self.__db.getAllStocks(stockname)
vector_x = []
vector_y = []
for stock in sorted(self.__allStocks.itervalues()):
vector_x.append(stock['date'].toordinal())
vector_y.append(int(stock['shareprice_closing']*100))
self.graph_stock.line_create(stockname, xdata=vector_x, ydata=vector_y, symbol='')
self.graph_stock.legend_configure(self.graph_stock.legend_get(0), hide=True)
(nValues, bins) = numpy.histogram(vector_y, bins=range(self._histogramRange))
# filter out small numbers so that trimming is fairly efficient
numpy.putmask(nValues, nValues<=3, 0)
trimmedN = numpy.trim_zeros(nValues)
histvector_x = bins.astype(int)[0:len(trimmedN)].tolist()
histvector_y = trimmedN.astype(int).tolist()
self.histogram_stock.bar_create(stockname, xdata=histvector_x, ydata=histvector_y, barwidth=3.0)
self.histogram_stock.legend_configure(self.histogram_stock.legend_get(0), hide=True)
def populate_webpages(self, stock):
self.listBox_webpages.delete(0, END)
self._selectedWebpages = self.__db.getWebpages(stock['date']-timedelta(3), stock['date'])
for webpage in self._selectedWebpages:
self.listBox_webpages.insert(END, (str(webpage.date), str(webpage.title)))
def listbox_stocks_SingleClick(self):
stockname = self._allStockNames[int(self.listbox_stocks.curselection()[0])]
self.graph_stock_populate(stockname)
def listBox_webpagesDoubleClick(self, event):
webpage = self._selectedWebpages[int(self.listBox_webpages.curselection()[0])]
webbrowser.open(webpage.url, new=0, autoraise=True)
def graph_stock_mouseSelect(self, event):
dicto = self.graph_stock.element_closest(event.x, event.y, halo=1000, interpolate=0)
if dicto != None:
stock = self.__allStocks[date.fromordinal(int(dicto['x']))]
labelText = "StockEntry: " + str(stock['date']) + " value: " + str(stock['shareprice_closing']) + " SEK"
self.label_graph_stock.config(text=labelText)
(Y0, Y1) = self.graph_stock.yaxis_limits()
self.graph_stock.marker_create("line", name="marking line", dashes="", linewidth=2,
coords=(dicto['x'], Y0, dicto['x'], Y1))
self.populate_webpages(stock)
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(1103, 509)
# Create StockGraph and _stockList object
self.shadowList = []
self.usedList = []
for i in range(0, 10):
stockGraph = StockGraph(numofplot=3, name=["close", "high", "low"])
self.shadowList.append(stockGraph)
self._stockList = StockList(self.shadowList, self.usedList)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setEnabled(True)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Maximum,
QtWidgets.QSizePolicy.Maximum)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.centralwidget.sizePolicy().hasHeightForWidth())
self.centralwidget.setSizePolicy(sizePolicy)
self.centralwidget.setMaximumSize(QtCore.QSize(1800, 700))
self.centralwidget.setObjectName("centralwidget")
self.mainBox = QtWidgets.QVBoxLayout(self.centralwidget)
self.mainBox.setObjectName("mainBox")
self.Container = QtWidgets.QHBoxLayout()
self.Container.setObjectName("Container")
'''Create Tab widget and settig'''
self.tabWidget = QtWidgets.QTabWidget(self.centralwidget)
self.tabWidget.setObjectName("tabWidget")
self.tabWidget.setTabsClosable(True)
self.tabWidget.tabCloseRequested.connect(self.tabRemove)
self.tabWidget.tabCloseRequested.connect(self.tabRemove)
# self.tabWidget.tabCloseRequested.connect(lambda index: self.tabWidget.removeTab(index))
#We fix the width of QToolBox, so that the width of Tabwidget will be not expanded when windows is enlarged
self.Setting = QtWidgets.QToolBox(self.centralwidget)
sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed,
QtWidgets.QSizePolicy.Preferred)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setHeightForWidth(
self.Setting.sizePolicy().hasHeightForWidth())
self.Setting.setSizePolicy(sizePolicy)
# self.Setting.setMinimumSize(QtCore.QSize(300, 520))
self.Setting.setObjectName("Setting")
self.ProxyPanel = ProxyPanel()
self.Setting.addItem(self.ProxyPanel.ret_widget(), "Proxy")
self.BasicPanel = BasicPanel(self.tabWidget)
self.Setting.addItem(self.BasicPanel.ret_widget(), "Basic")
self.BasicPanel.append_graphList(self._stockList)
self.OverviewPanel = OverviewPanel(self.tabWidget)
self.Setting.addItem(self.OverviewPanel.ret_widget(), "Overview")
self.RevenuePanel = RevenuePanel(self.tabWidget)
self.Setting.addItem(self.RevenuePanel.ret_widget(), "Revenue")
# #first Tab
# self.stockTab1 = QtWidgets.QWidget()
# self.stockTab1.setObjectName("stockTab1")
# self.tabWidget.addTab(self.stockTab1, "")
# #set Tab layout
# self.stockTab1.layout = QtWidgets.QHBoxLayout()
# self.StockGraph = StockGraph(numofplot = 3, name= ["close","high","low"])
# #Create a graph widget
# self.stockTab1.layout.addWidget(self.StockGraph.ret_GraphicsLayoutWidget())
# self.stockTab1.setLayout(self.stockTab1.layout)
# #second Tab
# self.stockTab2 = QtWidgets.QWidget()
# self.stockTab2.setObjectName("tab_4")
# #set Tab layout
# self.stockTab2.layout = QtWidgets.QHBoxLayout()
# self.Graph = Graphexample()
# #layout add to TAB2
# self.stockTab2.layout.addWidget(self.Graph.ret_GraphicsLayoutWidget())
# self.stockTab2.setLayout(self.stockTab2.layout)
# self.tabWidget.addTab(self.stockTab2, "")
self.Container.addWidget(self.tabWidget)
self.Container.addWidget(self.Setting)
self.mainBox.addLayout(self.Container)
MainWindow.setCentralWidget(self.centralwidget)
self.retranslateUi(MainWindow)
self.tabWidget.setCurrentIndex(0)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
