summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--main.py17
1 files changed, 15 insertions, 2 deletions
diff --git a/main.py b/main.py
index 5ed51c5..8bd80dd 100644
--- a/main.py
+++ b/main.py
@@ -44,15 +44,27 @@ class CompanySelector:
def show_companies(self):
self.select_file_button.pack_forget()
- self.company_listbox = tk.Listbox(self.master, selectmode=tk.MULTIPLE, exportselection=False)
- self.company_listbox.pack()
+ self.company_listbox = tk.Listbox(self.master, selectmode=tk.MULTIPLE, exportselection=False, height=20)
+ self.company_listbox.pack(expand=True)
for company in self.df["Symbol"]:
self.company_listbox.insert(tk.END, company)
+
+ # bind the listbox to an onselect event
+ self.company_listbox.bind('<<ListboxSelect>>', self.onselect)
+
+ self.num_companies_selected_label = tk.Label(self.master, text="0 companies selected")
+ self.num_companies_selected_label.pack()
+
self.create_pyramid_button = tk.Button(self.master, text="Create Pyramid", command=self.create_pyramid)
self.create_pyramid_button.pack()
+ def onselect(self, evt):
+ w = evt.widget
+ num_selected = len(w.curselection())
+ self.num_companies_selected_label.config(text=f"{num_selected} companies selected")
+
def create_pyramid(self):
selected_companies = [self.company_listbox.get(index) for index in self.company_listbox.curselection()]
selected_df = self.df[self.df["Symbol"].isin(selected_companies)].sort_values(by="Weighting", ascending=False)
@@ -220,4 +232,5 @@ class CompanySelector:
if __name__ == "__main__":
root = tk.Tk()
app = CompanySelector(root)
+ #root.geometry("500x200")
root.mainloop()