Exemplo n.º 1
0
	def add(self, ticker, quantity):
		print "added " + ticker
		self.portfolio.addStock(ticker, quantity)
		self.portfolio.updateStatistics()
		risk = self.portfolio.annualVol
		self.calc = ttk.Label(lf, text=risk)
		self.fig1 = summaryVis.getGraph(self.portfolio, self.f1)
		recommendations = mpt.knn(mpt.trainingSet, self.portfolio, 5, self.riskVal, 1, self.d)
		self.recommendDict = dict()
		
		for beforeDist, ticker, quantity in recommendations:
			newPort = mpt.copyPortfolio(self.portfolio, mpt.testSet)
			newPort.addStock(ticker, quantity)
			newPort.updateStatistics()
			self.recommendDict[ticker] = newPort
		self.fig2 = interactivePlot.getInteractiveGraph(self.f2, self.portfolio, self.recommendDict, "<Button-1>", self.riskVal)
Exemplo n.º 2
0
    def add(self, ticker, quantity):
        print "added " + ticker
        self.portfolio.addStock(ticker, quantity)
        self.portfolio.updateStatistics()
        risk = self.portfolio.annualVol
        self.calc = ttk.Label(lf, text=risk)
        self.fig1 = summaryVis.getGraph(self.portfolio, self.f1)
        recommendations = mpt.knn(mpt.trainingSet, self.portfolio, 5,
                                  self.riskVal, 1, self.d)
        self.recommendDict = dict()

        for beforeDist, ticker, quantity in recommendations:
            newPort = mpt.copyPortfolio(self.portfolio, mpt.testSet)
            newPort.addStock(ticker, quantity)
            newPort.updateStatistics()
            self.recommendDict[ticker] = newPort
        self.fig2 = interactivePlot.getInteractiveGraph(
            self.f2, self.portfolio, self.recommendDict, "<Button-1>",
            self.riskVal)
Exemplo n.º 3
0
	def __init__(self, riskVal, d, f):
		self.riskVal = (float(riskVal) * 20 - 10)/100
		self.d = d
		portfolioDict = self.readPortfolio(f)	
		
		path = 'data/'
		for infile in glob.glob( os.path.join(path, '*.csv') ):
			ticker = infile.split('/')[1].split('.')[0]
			stockModel = mpt.StockModel(ticker)
			
			years = len(stockModel.historicalPrices)/float(252)
			if years < 5:
				os.remove(infile)
				print 'Removed due to low data: ' + ticker
				
			else:
				# number of trading days in a year * 4 years (fudge factor)
				divisionPoint = len(stockModel.historicalPrices) - 252*4
				
				trainingModel = mpt.StockModel(ticker, 0, divisionPoint)
				mpt.trainingSet[ticker] = trainingModel
		
		self.portfolio = mpt.PortfolioModel(mpt.trainingSet)
		
		
		for k, v in portfolioDict.iteritems():
			self.portfolio.addStock(k, int(v))
		self.portfolio.updateStatistics()	
		risk = self.portfolio.annualVol
	
		window = Tk()
		window.title("Test GUI")
		#window.geometry('1100x600+150+130')
		window.resizable(FALSE,FALSE)
		
		frame = ttk.Frame(window, padding="10 10 10 10")
		frame.grid(column=0, row=0, sticky=(N, W, E, S))
		
		# Tell the frame to expand automatically if resized by the user. DOESNT WORK?
		#frame.columnconfigure(0, weight=1)
		#frame.rowconfigure(0, weight=1)
		
		tabs = ttk.Notebook(frame)
		tabs.grid(column=1, row=1, sticky=(N, W))
		
		self.f1 = ttk.Frame(tabs); # first page, which would get widgets gridded into it
		tabs.add(self.f1, text='Portfolio Dashboard')
		self.fig1 = summaryVis.getGraph(self.portfolio, self.f1)
		canvas1 = FigureCanvasTkAgg(self.fig1, master=self.f1)
		canvas1.get_tk_widget().grid(column=0, row=0, sticky=(N, W, E, S))
		canvas1.get_tk_widget().pack(side=TOP, fill=BOTH, expand=1)
		canvas1._tkcanvas.pack(side=TOP, fill=BOTH, expand=1)
		canvas1.show()
		
		self.f2 = ttk.Frame(tabs); # second page
		tabs.add(self.f2, text='Stock Recommender')
#		self.canvas2 = Canvas(self.f2)
#		self.canvas2.grid(column=0, row=0, sticky=(N, W, E, S))
#		self.canvas2.bind("<Button-1>", self.xy)
#		self.canvas2.bind("<B1-Motion>", self.addLine)
		
		recommendations = mpt.knn(mpt.trainingSet, self.portfolio, 5, self.riskVal, 1, self.d)
		self.recommendDict = dict()
		
		for beforeDist, ticker, quantity in recommendations:
			newPort = mpt.copyPortfolio(self.portfolio, mpt.testSet)
			newPort.addStock(ticker, quantity)
			newPort.updateStatistics()
			self.recommendDict[ticker] = newPort
        	
		self.fig2 = interactivePlot.getInteractiveGraph(self.f2, self.portfolio, recommendDict, "<Button-1>", self.riskVal)
#		recommendedFrame = ttk.Frame(self.f2)
#		recommendedFrame.grid(column=0, row=1, sticky=(N, W, E, S))
		canvas2 = FigureCanvasTkAgg(self.fig2, master=self.f2)
		canvas2.get_tk_widget().grid(column=0, row=0, sticky=(N, W, E, S))
#		canvas2.get_tk_widget().pack(side=TOP, fill=BOTH, expand=1)
#		canvas2._tkcanvas.pack(side=TOP, fill=BOTH, expand=1)
		canvas2.show()
				
		detail_frame = ttk.Frame(self.f2, padding="10 10 10 10")
		detail_frame.grid(column=0, row=1, sticky=(N, W, E, S))
		details = ttk.Label(detail_frame, text='Some Details about the selected stock...')
		details.grid(column=0, row=0, sticky=(N, W, E, S))

		m = 3
		for beforeDist, ticker, quantity in recommendations:
			ttk.Button(detail_frame, text="Add "+ticker+" to Portfolio", command=lambda n=ticker, q=quantity: self.add(n, q)).grid(column=0, row=m, sticky=SE)
			m+=1
		
		stat_frame = ttk.Frame(window, padding="10 10 10 10")
		stat_frame.grid(column=1, row=0, sticky=(N, W, E, S))
		lf = ttk.Labelframe(stat_frame, text='Key Statistics', padding="10 10 10 10")
		lf.grid(column=0, row=0, sticky=(N, W, E, S))
		
		ttk.Label(lf, text="Target Risk: ").grid(column=0, row=0, sticky=W)
		ttk.Label(lf, text=self.riskVal).grid(column=1, row=1, sticky=E)
		ttk.Label(lf, text="Calculated Risk: ").grid(column=0, row=2, sticky=W)
		self.calc = ttk.Label(lf, text=risk)
		self.calc.grid(column=0, row=3, sticky=E)
		ttk.Label(lf, text="File: ").grid(column=0, row=4, sticky=W)
		ttk.Label(lf, text=f).grid(column=0, row=5, sticky=E)
		
		# iterate over widgets that are children of frame, add padding around each
		for child in frame.winfo_children(): child.grid_configure(padx=5, pady=5)
		
		# Enter event loop (ie run the GUI)
		window.mainloop()
Exemplo n.º 4
0
    def __init__(self, riskVal, d, f):
        self.riskVal = (float(riskVal) * 20 - 10) / 100
        self.d = d
        portfolioDict = self.readPortfolio(f)

        path = 'data/'
        for infile in glob.glob(os.path.join(path, '*.csv')):
            ticker = infile.split('/')[1].split('.')[0]
            stockModel = mpt.StockModel(ticker)

            years = len(stockModel.historicalPrices) / float(252)
            if years < 5:
                os.remove(infile)
                print 'Removed due to low data: ' + ticker

            else:
                # number of trading days in a year * 4 years (fudge factor)
                divisionPoint = len(stockModel.historicalPrices) - 252 * 4

                trainingModel = mpt.StockModel(ticker, 0, divisionPoint)
                mpt.trainingSet[ticker] = trainingModel

        self.portfolio = mpt.PortfolioModel(mpt.trainingSet)

        for k, v in portfolioDict.iteritems():
            self.portfolio.addStock(k, int(v))
        self.portfolio.updateStatistics()
        risk = self.portfolio.annualVol

        window = Tk()
        window.title("Test GUI")
        #window.geometry('1100x600+150+130')
        window.resizable(FALSE, FALSE)

        frame = ttk.Frame(window, padding="10 10 10 10")
        frame.grid(column=0, row=0, sticky=(N, W, E, S))

        # Tell the frame to expand automatically if resized by the user. DOESNT WORK?
        #frame.columnconfigure(0, weight=1)
        #frame.rowconfigure(0, weight=1)

        tabs = ttk.Notebook(frame)
        tabs.grid(column=1, row=1, sticky=(N, W))

        self.f1 = ttk.Frame(tabs)
        # first page, which would get widgets gridded into it
        tabs.add(self.f1, text='Portfolio Dashboard')
        self.fig1 = summaryVis.getGraph(self.portfolio, self.f1)
        canvas1 = FigureCanvasTkAgg(self.fig1, master=self.f1)
        canvas1.get_tk_widget().grid(column=0, row=0, sticky=(N, W, E, S))
        canvas1.get_tk_widget().pack(side=TOP, fill=BOTH, expand=1)
        canvas1._tkcanvas.pack(side=TOP, fill=BOTH, expand=1)
        canvas1.show()

        self.f2 = ttk.Frame(tabs)
        # second page
        tabs.add(self.f2, text='Stock Recommender')
        #		self.canvas2 = Canvas(self.f2)
        #		self.canvas2.grid(column=0, row=0, sticky=(N, W, E, S))
        #		self.canvas2.bind("<Button-1>", self.xy)
        #		self.canvas2.bind("<B1-Motion>", self.addLine)

        recommendations = mpt.knn(mpt.trainingSet, self.portfolio, 5,
                                  self.riskVal, 1, self.d)
        self.recommendDict = dict()

        for beforeDist, ticker, quantity in recommendations:
            newPort = mpt.copyPortfolio(self.portfolio, mpt.testSet)
            newPort.addStock(ticker, quantity)
            newPort.updateStatistics()
            self.recommendDict[ticker] = newPort

        self.fig2 = interactivePlot.getInteractiveGraph(
            self.f2, self.portfolio, recommendDict, "<Button-1>", self.riskVal)
        #		recommendedFrame = ttk.Frame(self.f2)
        #		recommendedFrame.grid(column=0, row=1, sticky=(N, W, E, S))
        canvas2 = FigureCanvasTkAgg(self.fig2, master=self.f2)
        canvas2.get_tk_widget().grid(column=0, row=0, sticky=(N, W, E, S))
        #		canvas2.get_tk_widget().pack(side=TOP, fill=BOTH, expand=1)
        #		canvas2._tkcanvas.pack(side=TOP, fill=BOTH, expand=1)
        canvas2.show()

        detail_frame = ttk.Frame(self.f2, padding="10 10 10 10")
        detail_frame.grid(column=0, row=1, sticky=(N, W, E, S))
        details = ttk.Label(detail_frame,
                            text='Some Details about the selected stock...')
        details.grid(column=0, row=0, sticky=(N, W, E, S))

        m = 3
        for beforeDist, ticker, quantity in recommendations:
            ttk.Button(
                detail_frame,
                text="Add " + ticker + " to Portfolio",
                command=lambda n=ticker, q=quantity: self.add(n, q)).grid(
                    column=0, row=m, sticky=SE)
            m += 1

        stat_frame = ttk.Frame(window, padding="10 10 10 10")
        stat_frame.grid(column=1, row=0, sticky=(N, W, E, S))
        lf = ttk.Labelframe(stat_frame,
                            text='Key Statistics',
                            padding="10 10 10 10")
        lf.grid(column=0, row=0, sticky=(N, W, E, S))

        ttk.Label(lf, text="Target Risk: ").grid(column=0, row=0, sticky=W)
        ttk.Label(lf, text=self.riskVal).grid(column=1, row=1, sticky=E)
        ttk.Label(lf, text="Calculated Risk: ").grid(column=0, row=2, sticky=W)
        self.calc = ttk.Label(lf, text=risk)
        self.calc.grid(column=0, row=3, sticky=E)
        ttk.Label(lf, text="File: ").grid(column=0, row=4, sticky=W)
        ttk.Label(lf, text=f).grid(column=0, row=5, sticky=E)

        # iterate over widgets that are children of frame, add padding around each
        for child in frame.winfo_children():
            child.grid_configure(padx=5, pady=5)

        # Enter event loop (ie run the GUI)
        window.mainloop()