Add files via upload

Author: Apfirebolt

balance.py

@@ -0,0 +1,76 @@
+import pickle
+import tkinter as tk
+from tkinter import messagebox
+
+
+class Portfolio:
+    def __init__(self, username, balance):
+        self.username = username
+        self.balance = balance
+        self.stock_list = {}
+
+
+def add_balance_util(a, b):
+    s = int(b.get())
+    a.balance += s
+
+    filename = 'user_data/' + a.username + '.pkl'
+    pickle_out = open(filename, 'wb')
+    pickle.dump(a, pickle_out)
+    pickle_out.close()
+
+    messagebox.showinfo("Confirmation Message", str(s) + " was added to your account.")
+
+
+def add_balance(x):
+    add_balance_window = tk.Tk()
+    add_balance_window.title('Adding balance to your account!')
+    add_balance_window.geometry('400x200')
+
+    add_balance_frame = tk.Frame(add_balance_window)
+    add_balance_frame.pack(side='top', fill='x')
+
+    add_balance_label = tk.Label(add_balance_frame, text='Enter the amount to add ')
+    add_balance_label.pack(side='left')
+
+    add_balance_entry = tk.Entry(add_balance_frame)
+    add_balance_entry.pack(side='left')
+
+    add_balance_button = tk.Button(add_balance_window, text='Ok',
+                                   command=lambda a=x, b=add_balance_entry: add_balance_util(a, b))
+    add_balance_button.pack(side='bottom', fill='x')
+
+
+#Main function of this module starts from here hence the name start LOL. Username passed as user from home function
+def start(user):
+    temp = Portfolio('', 0)
+    filename = 'user_data/' + user + '.pkl'
+    file = open(filename, 'rb')
+    temp = pickle.load(file)
+
+    #closed the original loaded pickle user file.
+    file.close()
+
+    # temp has temporary variables extracted from pickled files. Successfully printed values.
+    # print(temp.username)
+    # print(temp.balance)
+    # print(temp.stock_list)
+
+    #Opening a new window layout
+    balance_window = tk.Tk()
+    balance_window.title('Your Account information!')
+    balance_window.geometry('300x300')
+
+    exit_button = tk.Button(balance_window, text='Exit', command=balance_window.destroy)
+    exit_button.pack(side='bottom', fill='x')
+
+    #Passing temp object as an argument (x) to permanently modify and write new balance value.
+    balance_button = tk.Button(balance_window, text='Add Balance', command=lambda x=temp: add_balance(x))
+    balance_button.pack(side='bottom', fill='x')
+
+    balance_label = tk.Label(balance_window, text='Your current balance is Rs ' + str(temp.balance))
+    balance_label.pack(side='bottom', fill='x')
+
+    balance_window.mainloop()
+
+

buy.py

@@ -0,0 +1,127 @@
+import pickle
+import tkinter as tk
+import requests
+from bs4 import BeautifulSoup
+from tkinter import messagebox
+
+
+class Portfolio:
+    def __init__(self, username, balance):
+        self.username = username
+        self.balance = balance
+        self.stock_list = {}
+
+list_stock = ['ADANIPORTS', 'ASIANPAINT', 'AXISBANK', 'BAJAJ-AUTO', 'BAJFINANCE', 'BAJAJFINSV', 'BPCL',
+'BHARTIARTL', 'INFRATEL', 'CIPLA', 'COALINDIA', 'DRREDDY', 'EICHERMOT', 'GAIL', 'GRASIM',
+'HCLTECH', 'HDFCBANK', 'HEROMOTOCO', 'HINDALCO', 'HINDPETRO', 'HINDUNILVR', 'HDFC', 'ITC',
+'ICICIBANK', 'IBULHSGFIN', 'IOC', 'INDUSINDBK', 'INFY', 'KOTAKBANK', 'LT', 'LUPIN', 'M&M',
+'MARUTI', 'NTPC', 'ONGC', 'POWERGRID', 'RELIANCE', 'SBIN', 'SUNPHARMA', 'TCS', 'TATAMOTORS',
+'TATASTEEL', 'TECHM', 'TITAN', 'UPL', 'ULTRACEMCO', 'VEDL', 'WIPRO', 'YESBANK', 'ZEEL']
+
+
+def on_buy_util(a, b, c, d):
+    unit = int(b.get())
+
+    #Balance update
+    a.balance = a.balance - (unit * c)
+
+    #Portfolio update
+    a.stock_list[d] = unit
+
+    filename = 'user_data/' + a.username + '.pkl'
+    file = open(filename, 'wb')
+
+    #Storing updated values in pickled database.
+    pickle.dump(a, file)
+    file.close()
+
+    success_text = 'Congratulations! You have bought ' + str(unit) + ' of ' + d
+    messagebox.showinfo(success_text)
+
+
+def on_buy_func(user, y):
+    my_stock = y.get()
+
+    temp = Portfolio('', 0)
+    filename = 'user_data/' + user + '.pkl'
+    file = open(filename, 'rb')
+    temp = pickle.load(file)
+
+    print(temp.balance)
+
+    try:
+        url = 'https://www.screener.in/company/' + my_stock
+        response = requests.get(url)
+        data = response.text
+
+    except requests.exceptions.RequestException as exp:
+        print(exp)
+
+    soup = BeautifulSoup(data, 'lxml')
+
+    data_items = soup.findAll('b')[0:4]
+    stock_data = [item.text for item in data_items]
+
+    current_price = stock_data[1]
+    current_price = current_price.replace(',','')
+
+    current_price = int(float(current_price))
+
+    if current_price > temp.balance:
+        messagebox.showinfo("You cannot buy this stock due to low balance.")
+
+    else:
+        quantity = temp.balance // current_price
+        msg = 'You can buy ' + str(quantity) + ' of this stock.'
+
+        #So far so good. Now call the on_buy_util function, pass quantity, user
+        on_buy_window = tk.Tk()
+        on_buy_window.geometry('400x200')
+        on_buy_window.title('Confirmation for buying ' + my_stock)
+
+        t1 = 'The current price of ' + my_stock + ' is ' + str(current_price)
+        l1 = tk.Label(on_buy_window, text=t1)
+        l1.pack(side='top')
+
+        t2 = 'Your current balance is ' + str(temp.balance)
+        l2 = tk.Label(on_buy_window, text=t2)
+        l2.pack(side='top')
+
+        l3 = tk.Label(on_buy_window, text=msg)
+        l3.pack(side='top')
+
+        on_buy_frame = tk.Frame(on_buy_window)
+        quantity_label = tk.Label(on_buy_frame, text='Select quantity')
+        quantity_label.pack(side='left')
+
+        on_buy_entry = tk.Entry(on_buy_frame)
+        on_buy_entry.pack(side='left')
+
+        on_buy_frame.pack(side='top')
+
+        on_buy_button = tk.Button(on_buy_window, text='Buy now',
+                                  command=lambda a=temp, b=on_buy_entry, c=current_price, d=my_stock: on_buy_util(a, b, c, d))
+        on_buy_button.pack(side='bottom')
+
+        on_buy_window.mainloop()
+
+
+def start(user):
+    buy_window = tk.Tk()
+    buy_window.title('Buy a stock!')
+    buy_window.geometry('450x200')
+
+    buy_frame = tk.Frame(buy_window)
+    buy_label = tk.Label(buy_frame, text='Enter a stock to buy')
+    buy_label.pack(side='left')
+
+    buy_entry = tk.Entry(buy_frame)
+    buy_entry.pack(side='left')
+
+    buy_frame.pack(side='top')
+
+    buy_button = tk.Button(buy_window, text='Buy this stock', command=lambda x=user, y=buy_entry: on_buy_func(x, y))
+    buy_button.pack(side='bottom')
+
+    buy_window.mainloop()
+