stock_GUI.py

# Summary: This module contains the user interface and logic for a graphical user interface version of the stock manager program.

from datetime import datetime
from os import path
from tkinter import *
from tkinter import ttk
from tkinter import messagebox, simpledialog, filedialog
import csv
import stock_data
from stock_class import Stock, DailyData
from utilities import clear_screen, display_stock_chart, sortStocks, sortDailyData
import os
os.environ['TK_SILENCE_DEPRECATION'] = '1'

class StockApp:
    def __init__(self):
        self.stock_list = []
        # Check for database, create if not exists
        if path.exists("stocks.db") == False:
            stock_data.create_database()

    # CREATE USER INTERFACE 
    
    # Create Window
        self.root = Tk()
        self.root.title("Stock Manager - Lab 2")  
        self.root.geometry("900x700")
        
    # MENUBAR
    
        self.menubar = Menu(self.root)
        
        # Add File Menu
        self.filemenu = Menu(self.menubar, tearoff=0)
        self.filemenu.add_command(label="Load Data", command=self.load)
        self.filemenu.add_command(label="Save Data", command=self.save)
        self.menubar.add_cascade(label="File", menu=self.filemenu)
        
        # Add Web Menu
        self.webmenu = Menu(self.menubar, tearoff=0)
        self.webmenu.add_command(label="Scrape Data from Yahoo! Finance...", 
                                command=self.scrape_web_data)
        self.webmenu.add_command(label="Import CSV From Yahoo! Finance...", 
                                command=self.importCSV_web_data)
        self.menubar.add_cascade(label="Web", menu=self.webmenu)
        
        # Add Chart Menu
        self.chartmenu = Menu(self.menubar, tearoff=0)
        self.chartmenu.add_command(label="Display Chart", command=self.display_chart)
        self.menubar.add_cascade(label="Chart", menu=self.chartmenu)
        
        # Add menus to window
        self.root.config(menu=self.menubar)
        
        # HEADING INFORMATION 
        
        self.headingLabel = Label(self.root, text="Stock Manager", 
                                font=("Arial", 16, "bold"))
        self.headingLabel.grid(row=0, column=0, columnspan=3, padx=10, pady=10)
        
        # STOCK LIST 
        
        # Label for stock list
        Label(self.root, text="Stocks:").grid(row=1, column=0, sticky=W, padx=10)
        
        # Listbox with scrollbar
        self.stockListScrollbar = Scrollbar(self.root)
        self.stockListScrollbar.grid(row=2, column=0, sticky=W+E+N+S, padx=(10,0))
        
        self.stockList = Listbox(self.root, height=20, 
                                yscrollcommand=self.stockListScrollbar.set)
        self.stockList.grid(row=2, column=0, sticky=W+E+N+S, padx=(10,0))
        self.stockListScrollbar.config(command=self.stockList.yview)
        
        # Bind selection event
        self.stockList.bind('<<ListboxSelect>>', self.update_data)
        
        #  TABS 
        
        # Create notebook (tab container)
        self.notebook = ttk.Notebook(self.root)
        self.notebook.grid(row=2, column=1, columnspan=2, sticky=W+E+N+S, 
                        padx=10, pady=5)
        
        # MAIN TAB
        
        self.mainTab = Frame(self.notebook)
        self.notebook.add(self.mainTab, text="Main")
        
        # Add Stock Section
        Label(self.mainTab, text="Add Stock", 
            font=("Arial", 12, "bold")).grid(row=0, column=0, columnspan=3, pady=10)
        
        Label(self.mainTab, text="Symbol:").grid(row=1, column=0, sticky=E, padx=5)
        self.addSymbolEntry = Entry(self.mainTab, width=15)
        self.addSymbolEntry.grid(row=1, column=1, padx=5)
        
        Label(self.mainTab, text="Name:").grid(row=2, column=0, sticky=E, padx=5)
        self.addNameEntry = Entry(self.mainTab, width=30)
        self.addNameEntry.grid(row=2, column=1, padx=5)
        
        Label(self.mainTab, text="Shares:").grid(row=3, column=0, sticky=E, padx=5)
        self.addSharesEntry = Entry(self.mainTab, width=15)
        self.addSharesEntry.grid(row=3, column=1, padx=5)
        
        Button(self.mainTab, text="Add Stock", 
            command=self.add_stock).grid(row=4, column=1, pady=10)
        
        # Update Shares Section
        Label(self.mainTab, text="Update Shares", 
            font=("Arial", 12, "bold")).grid(row=5, column=0, columnspan=3, pady=10)
        
        Label(self.mainTab, text="Shares:").grid(row=6, column=0, sticky=E, padx=5)
        self.updateSharesEntry = Entry(self.mainTab, width=15)
        self.updateSharesEntry.grid(row=6, column=1, padx=5)
        
        Button(self.mainTab, text="Buy", 
            command=self.buy_shares).grid(row=7, column=0, pady=5, padx=5)
        Button(self.mainTab, text="Sell", 
            command=self.sell_shares).grid(row=7, column=1, pady=5, padx=5)
        
        # Delete Stock Button
        Button(self.mainTab, text="Delete Stock", 
            command=self.delete_stock).grid(row=8, column=1, pady=10)
        
        # HISTORY TAB 
        
        self.historyTab = Frame(self.notebook)
        self.notebook.add(self.historyTab, text="History")
        
        # Add scrollbar
        self.dailyDataScrollbar = Scrollbar(self.historyTab)
        self.dailyDataScrollbar.grid(row=0, column=1, sticky=N+S)
        
        # Add text widget for displaying history
        self.dailyDataList = Text(self.historyTab, width=60, height=30,
                                yscrollcommand=self.dailyDataScrollbar.set)
        self.dailyDataList.grid(row=0, column=0, padx=10, pady=10)
        self.dailyDataScrollbar.config(command=self.dailyDataList.yview)
        
        #  REPORT TAB 
        
        self.reportTab = Frame(self.notebook)
        self.notebook.add(self.reportTab, text="Report")
        
        # Add scrollbar
        self.stockReportScrollbar = Scrollbar(self.reportTab)
        self.stockReportScrollbar.grid(row=0, column=1, sticky=N+S)
        
        # Add text widget for displaying report
        self.stockReport = Text(self.reportTab, width=60, height=30,
                            yscrollcommand=self.stockReportScrollbar.set)
        self.stockReport.grid(row=0, column=0, padx=10, pady=10)
        self.stockReportScrollbar.config(command=self.stockReport.yview)
        
        #  CONFIGURE GRID WEIGHTS 
        
        self.root.grid_rowconfigure(2, weight=1)
        self.root.grid_columnconfigure(1, weight=1)
        
        # START GUI 
        
        # Call MainLoop
        self.root.mainloop()

# This section provides the functionality
       
    # Load stocks and history from database.
    def load(self):
        self.stockList.delete(0,END)
        stock_data.load_stock_data(self.stock_list)
        sortStocks(self.stock_list)
        for stock in self.stock_list:
            self.stockList.insert(END,stock.symbol)
        messagebox.showinfo("Load Data","Data Loaded")

    # Save stocks and history to database.
    def save(self):
        stock_data.save_stock_data(self.stock_list)
        messagebox.showinfo("Save Data","Data Saved")

    # Refresh history and report tabs
    def update_data(self, evt):
        self.display_stock_data()

    # Display stock price and volume history.
    def display_stock_data(self):
        symbol = self.stockList.get(self.stockList.curselection())
        for stock in self.stock_list:
            if stock.symbol == symbol:
                self.headingLabel['text'] = stock.name + " - " + str(stock.shares) + " Shares"
                self.dailyDataList.delete("1.0",END)
                self.stockReport.delete("1.0",END)
                self.dailyDataList.insert(END,"- Date -   - Price -   - Volume -\n")
                self.dailyDataList.insert(END,"=================================\n")
                for daily_data in stock.DataList:
                    row = daily_data.date.strftime("%m/%d/%y") + "   " +  '${:0,.2f}'.format(daily_data.close) + "   " + str(daily_data.volume) + "\n"
                    self.dailyDataList.insert(END,row)

                #display report


                    
    
    # Add new stock to track.
    def add_stock(self):
        new_stock = Stock(self.addSymbolEntry.get(),self.addNameEntry.get(),float(str(self.addSharesEntry.get())))
        self.stock_list.append(new_stock)
        self.stockList.insert(END,self.addSymbolEntry.get())
        self.addSymbolEntry.delete(0,END)
        self.addNameEntry.delete(0,END)
        self.addSharesEntry.delete(0,END)

    # Buy shares of stock.
    def buy_shares(self):
        symbol = self.stockList.get(self.stockList.curselection())
        for stock in self.stock_list:
            if stock.symbol == symbol:
                stock.buy(float(self.updateSharesEntry.get()))
                self.headingLabel['text'] = stock.name + " - " + str(stock.shares) + " Shares"
        messagebox.showinfo("Buy Shares","Shares Purchased")
        self.updateSharesEntry.delete(0,END)

    # Sell shares of stock.
    def sell_shares(self):
        symbol = self.stockList.get(self.stockList.curselection())
        for stock in self.stock_list:
            if stock.symbol == symbol:
                stock.sell(float(self.updateSharesEntry.get()))
                self.headingLabel['text'] = stock.name + " - " + str(stock.shares) + " Shares"
        messagebox.showinfo("Sell Shares","Shares Sold")
        self.updateSharesEntry.delete(0,END)

    # Remove stock and all history from being tracked.
    def delete_stock(self):
       pass

    # Get data from web scraping.
    def scrape_web_data(self):
        dateFrom = simpledialog.askstring("Starting Date","Enter Starting Date (m/d/yy)")
        dateTo = simpledialog.askstring("Ending Date","Enter Ending Date (m/d/yy")
        try:
            stock_data.retrieve_stock_web(dateFrom,dateTo,self.stock_list)
        except:
            messagebox.showerror("Cannot Get Data from Web","Check Path for Chrome Driver")
            return
        self.display_stock_data()
        messagebox.showinfo("Get Data From Web","Data Retrieved")

    # Import CSV stock history file.
    def importCSV_web_data(self):
        symbol = self.stockList.get(self.stockList.curselection())
        filename = filedialog.askopenfilename(title="Select " + symbol + " File to Import",filetypes=[('Yahoo Finance! CSV','*.csv')])
        if filename != "":
            stock_data.import_stock_web_csv(self.stock_list,symbol,filename)
            self.display_stock_data()
            messagebox.showinfo("Import Complete",symbol + "Import Complete")   
    
    # Display stock price chart.
    def display_chart(self):
        symbol = self.stockList.get(self.stockList.curselection())
        display_stock_chart(self.stock_list,symbol)


def main():
        app = StockApp()
        

if __name__ == "__main__":
    # execute only if run as a script
    main()