managerGui.py

import tkinter as tk
from functools import partial
import pyperclip

from PMUtils import *
from PMDB import PasswordDatabase

#used for login when the masterpassword is already created
class loginGui:
    def __init__(self) -> None:
        self.window = tk.Tk()
        self.window.title("LOGIN")
        self.window.geometry('250x100')

        self.Greeting = tk.Label(master=self.window, text="Welcome")
        self.Greeting.pack()
        
        self.Text = tk.Label(master=self.window,text ="Enter Master Password")
        self.Text.pack()

        self.pwd_var = tk.StringVar()
        self.Entry = tk.Entry(master=self.window,show="*",textvariable=self.pwd_var)
        self.Entry.focus_set()
        self.Entry.pack()

        #button on the login window calls the login function
        self.Button = tk.Button(master=self.window,text="Login", command=self.login)
        self.Button.pack()

        self.noMatch = tk.Label(master=self.window, text = "password is incorrect, try again")

        #also pressing enter with the text entry selected call the login function
        self.Entry.bind("<Return>", self.login)
        self.masterHash = getMaster() #gets the stored hash of the master password
        self.window.mainloop()

    #takes the entered password, hashes it, and compares it to the hash retrieved from file
    #prompt the user to re-enter if the hashes do not match
    def login(self, event=None):
        self.userpass = self.pwd_var.get()
        self.userhash = getUserHash(self.userpass)
        if(self.userhash != self.masterHash):
            self.noMatch.pack()
            self.Entry.delete(0,tk.END)
        else:
            self.noMatch.pack_forget()
            tk.Label(master=self.window,text = "success").pack()
            self.window.after(500,self.window.destroy)

#used to set the master password the first time 
#prompts user to enter the password twice to verify
class setMasterGui:
    def __init__(self) -> None:
        self.window = tk.Tk()
        self.window.title("New Passoword")
        self.window.geometry('250x100')

        self.Greeting = tk.Label(master= self.window, text = "Welcome")
        self.Greeting.pack()

        self.Text = tk.Label(master=self.window, text="Enter new Master password",)
        self.Text.pack()

        self.noMatch = tk.Label(master=self.window, text = "passwords did not match, please try again")
        
        self.pwd_var = tk.StringVar()
        self.Entry = tk.Entry(master=self.window,show="*",textvariable=self.pwd_var)
        self.Entry.focus_set()
        self.Entry.pack()

        self.Button = tk.Button(master=self.window,text="Submit", command=self.get_first)
        self.Button.pack()

        self.Entry.bind("<Return>", self.get_first)
        self.window.mainloop()
    
    #used to get the first entry from the text 
    def get_first(self,event=None):
        self.first = self.pwd_var.get()
        self.Entry.delete(0,tk.END)
        self.Button["command"] = self.get_second
        self.Entry.bind("<Return>", self.get_second)
        self.Text["text"] = "Confirm password"

    #after second password is entered, if the two entries match the password is hashed and saved
    #otherwise the user is promted to re-enter
    def get_second(self,event=None):
        self.second = self.pwd_var.get()
        if(self.second != self.first):
            self.noMatch.pack()
            self.Entry.delete(0,tk.END)
            self.Button["command"] = self.get_first
            self.Entry.bind("<Return>", self.get_first)
            self.Text["text"] = "Enter new Master password"
        else:
            self.noMatch.pack_forget()
            tk.Label(master=self.window,text = "new password saved").pack()
            setMaster(self.second)
            self.userpass = self.second
            self.window.after(500,self.window.destroy)

#popup window for entering in new credentials 
#gives user the option to use a randomly generated passwords 
class entryPopup:
    def __init__(self,Master) -> None:
        self.master= Master
        self.pop= tk.Toplevel(master=self.master)
        self.pop.title("Add New Credentials")
        self.pop.geometry("250x120")
        
        tk.Label(master=self.pop,text="Website:").grid(row=0,column=0,sticky= 'E')
        self.site = tk.StringVar()
        self.siteEntry = tk.Entry(master=self.pop,textvariable=self.site)
        self.siteEntry.grid(row=0,column=1)


        tk.Label(master=self.pop,text="User:").grid(row=1,column=0,sticky='E')
        self.user = tk.StringVar()
        self.userEntry = tk.Entry(master=self.pop,textvariable=self.user)
        self.userEntry.grid(row=1,column=1)

        tk.Label(master=self.pop,text="Password:").grid(row=2,column=0,sticky='E')
        self.password = tk.StringVar()
        self.passwordEntry = tk.Entry(master=self.pop,show='*',textvariable=self.password)
        self.passwordEntry.grid(row=2,column=1)

        tk.Button(master=self.pop,text="generate",command=self.genRandomPass).grid(row=2,column=2)

        tk.Button(master=self.pop,text="confirm",command=self.saveCred).grid(row=3,column=0,columnspan=3)

        self.error = tk.Label(master=self.pop, text="An entry was left blank")

        self.pop.bind("<Return>", self.saveCred)
        self.pop.wait_window()
        #self.u = None
        #self.s = None
        #self.p = None

    #verifies that none of the entries were left blank when the confirm button is pressed
    #promts user if any entry is blank
    def saveCred(self,event=None):
        self.s = self.siteEntry.get()
        self.u = self.userEntry.get()
        self.p = self.passwordEntry.get()
        if(len(self.s )==0 or len(self.u )==0 or len(self.p )==0):
            self.error.grid(row=4,column=0,columnspan=3)
        else:
            self.error.grid_forget()
            self.pop.after(500,self.pop.destroy)

    def genRandomPass(self):
        self.password.set(genRandPw()) #see PMUtils.py

#passwords are not displayed on the main screen, popup allows user to view
class showPasswordPopup:
    def __init__(self,Master,pw) -> None:
        self.master= Master
        self.pop= tk.Toplevel(master=self.master)
        self.pop.title("show Password")
        self.pop.geometry("200x100")

        password = tk.StringVar()
        password.set(pw)
        tk.Entry(master=self.pop,textvariable=password, state="readonly" ).pack()
        self.pop.wait_window()

#popup for deletion of credential - prompts user to confimrm before deletion
class confirmDeletePop:
    def __init__(self,Master,site,db:PasswordDatabase) -> None:
        self.master= Master
        self.site =site
        self.db = db
        self.pop= tk.Toplevel(master=self.master)
        self.pop.title("Delete Credential")
        self.pop.geometry("200x100")
        
        tk.Label(master=self.pop,text="Are you sure?").pack()
        tk.Button(master=self.pop,text="Delete Credential",command=self.deleteCred).pack()
        self.pop.wait_window()

    def deleteCred(self):
        self.db.delCred(self.site)
        self.pop.destroy()

#main window:
#options to add new credentials, and dynamic filtering of the display list
#users can select to delete, view and copy to clipboard individual passwords
class mainView:
    def __init__(self,db:PasswordDatabase,key) -> None:
        self.db = db
        self.key = key

        self.window =tk.Tk()
        self.window.minsize(400,400)
        self.window.title("My password Manager")

        self.topFrame = tk.Frame(master=self.window)
        self.topFrame.pack(side="top")
        self.searchStr = tk.StringVar()
        self.searchStr.trace_add("write",self.filterView)
        tk.Button(master=self.topFrame,text="Add New",command=self.addCred).grid(row=0,column=0,padx=2)
        tk.Entry(self.topFrame, textvariable=self.searchStr).grid(row=0,column=2,padx=2)
        #tk.Button(master=self.window,text="Find",command=self.filterView).grid(row=0,column=3,padx=2)
    
        self.viewFrame = tk.Frame(master=self.window)
        self.viewFrame.pack(side="top",fill='y')

        #self.scroll = tk.Scrollbar(self.viewFrame)
        #self.scroll.pack(side="right",fill="y")

        tk.Label(master=self.viewFrame,text="Website:").grid(row=0,column=0,padx=2,sticky='W')
        tk.Label(master=self.viewFrame,text="username:" ).grid(row=0,column=1,padx=2,sticky='W')
        tk.Label(master=self.viewFrame,text="password:" ).grid(row=0,column=2,padx=2,columnspan=2,sticky="W")
        self.viewList = []
        self.viewListSite:list[tk.Text] = []
        self.viewListUser:list[tk.Entry] = []
        self.viewListPassShow:list[tk.Button] = []
        self.viewListPassCopy:list[tk.Button] = []
        self.viewListDel:list[tk.Button] = []

        self.searchStr.set("")
        self.window.mainloop()

    #dynamically changes the list of credentials based on the entry box contents
    def filterView(self,var=None, index=None, mode=None):
        search = "{}".format(self.searchStr.get())

        #clears the lists containing the text boxs and buttons from the window
        for x in self.viewListSite:
            x.destroy()
        self.viewListSite.clear()
        for x in self.viewListUser:
            x.destroy()
        self.viewListUser.clear()
        for x in self.viewListPassCopy:
            x.destroy()
        self.viewListPassCopy.clear()
        for x in self.viewListPassShow:
            x.destroy()
        self.viewListPassShow.clear()
        for x in self.viewListDel:
            x.destroy()
        self.viewListDel.clear()

        #retrieve list of credentials from db
        self.viewList = self.db.getFilteredList(search) 
        #if no values are returned for a search string then empty entry is displayed
        if(len(self.viewList) == 0):
            self.viewList.append(tuple(("","","")))
        
        #repopulate window with credentials matching the search, and the associated buttons
        for x in range(len(self.viewList)):
            site = tk.StringVar()
            site.set(self.viewList[x][0])
            self.viewListSite.append(tk.Entry(master=self.viewFrame,textvariable=site,state="readonly"))
            self.viewListSite[x].grid(row=x+1,column=0,padx=2)

            user = tk.StringVar()
            user.set(self.viewList[x][1])
            self.viewListUser.append(tk.Entry(master=self.viewFrame,textvariable=user,state="readonly"))
            self.viewListUser[x].grid(row=x+1,column=1,padx=2)

            self.viewListPassShow.append(tk.Button(master=self.viewFrame,text="Show",command=partial(self.showPassword,cp=self.viewList[x][2]))) 
            self.viewListPassShow[x].grid(row=x+1,column=2,padx=2)

            self.viewListPassCopy.append(tk.Button(master=self.viewFrame,text="Copy",command=partial(self.copyToClip,cp=self.viewList[x][2]))) 
            self.viewListPassCopy[x].grid(row=x+1,column=3,padx=2)

            self.viewListDel.append(tk.Button(master=self.viewFrame,text="X",command=partial(self.deleteCred,site=self.viewList[x][0]))) 
            self.viewListDel[x].grid(row=x+1,column=4,padx=2)

    #creates add credential popup, and verifies the values returned are not empty(for the case where popup was closed by the X button, not the confirm button)
    def addCred(self):
        addform = entryPopup(self.window)
        if(len(addform.s)==0 or len(addform.u)==0 or len(addform.p)==0):
            errorpop = tk.Toplevel(self.window)
            tk.Message(master=errorpop,text="Form was incomplete, entry was not saved.").pack()
            errorpop.wait_window()
        else:
            ep = encryptPasswords(addform.p,self.key)        #password is encrypted before storage
            self.db.newCred(addform.s,addform.u,ep)
            del addform
            self.filterView()
 
    def deleteCred(self, site):
        confirmDeletePop(self.window, site, self.db)
        self.filterView()

    def copyToClip(self, cp):
        pp = decryptPasswords(cp,self.key)
        pyperclip.copy(pp)

    def showPassword(self,cp):
        pp = decryptPasswords(cp,self.key)
        showPasswordPopup(self.window,pp)


#driver code
login = None
if(masterExists()):
    login = loginGui()
else:
    login = setMasterGui()
userpass = login.userpass
del login
mydb = PasswordDatabase()
mydb.connect_or_create()
key = genKeyFromMaster(userpass)
app = mainView(mydb,key)
mydb.closeConn()