usman_pwmng.py

#!/usr/bin/env python
# coding: utf-8

# In[3]:


import sqlite3, hashlib                #For database
from tkinter import *
from tkinter import simpledialog
from functools import partial
import uuid                            #For recovery key
import pyperclip                       #To copy the recovery key
import base64                          #To encrypt the data
import os
from cryptography.hazmat.primitives import hashes
from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
from cryptography.hazmat.backends import default_backend
from cryptography.fernet import Fernet


backend = default_backend()
salt = b'2444'

kdf = PBKDF2HMAC(
    algorithm=hashes.SHA256(),
    length = 32,
    salt = salt,
    iterations=100000,
    backend=backend
)

encryptionKey = 0

def encrypt(message: bytes, key: bytes) -> bytes:
    return Fernet(key).encrypt(message)

def decrypt(message: bytes, token: bytes) -> bytes:
    return Fernet(token).decrypt(message)



#Creating a database
with sqlite3.connect("My_Password_Manager.db") as db:
    cursor = db.cursor()

#creating a table
cursor.execute("""
CREATE TABLE IF NOT EXISTS masterpassword(
id INTEGER  PRIMARY KEY,
password TEXT NOT NULL,
recoveryKey TEXT NOT NULL);
""")

cursor.execute("""
CREATE TABLE IF NOT EXISTS vault(
id INTEGER  PRIMARY KEY,
application TEXT NOT NULL,
username TEXT NOT NULL,
password TEXT NOT NULL);
""")

#Create PopUps
def popUp(text):
    answer = simpledialog.askstring("input string", text)
    return(answer)


#Initializing the window
window = Tk()
window.update()
window.title("Password Vault")

def Hash_Password(input):
    hash1 = hashlib.sha256(input)
    hash1 = hash1.hexdigest()

    return hash1

def Home_Screen():

    for widget in window.winfo_children():
        widget.destroy()

    window.geometry("500x250")

    lbl = Label(window, text="Create a Master Password")
    lbl.config(anchor=CENTER)
    lbl.pack()

    txt = Entry(window, width=50, show = "*")
    txt.pack()
    txt.focus()

    lbl1 = Label(window, text="Re-enter Password")
    lbl1.config(anchor=CENTER)
    lbl1.pack()

    txt1 = Entry(window, width=50, show = "*")
    txt1.pack()
    txt1.focus()


    def Save_Password():
        if txt.get() == txt1.get():

            sql = "DELETE FROM masterpassword WHERE ID = 1"
            cursor.execute(sql)
    
            hashedpassword = Hash_Password(txt.get().encode('utf-8'))
            key = str(uuid.uuid4().hex)
            recoveryKey = Hash_Password(key.encode('utf-8'))

            global encryptionKey
            encryptionKey = base64.urlsafe_b64encode(kdf.derive(txt.get().encode()))


            insert_password = """INSERT INTO masterpassword(password, recoveryKey)
            VALUES(?, ?) """
            cursor.execute(insert_password, ((hashedpassword), (recoveryKey)))
            db.commit()

            Recovery_Screen(key)

        else:
            lbl1.config(text="Passwords do not match :(")
        

    btn = Button(window, text = "Save", command=Save_Password)
    btn.pack(pady=5)

def Recovery_Screen(key):
    for widget in window.winfo_children():
        widget.destroy()


    window.geometry("500x250")

    lbl = Label(window, text="Please save this key for you to recover your Vault.")
    lbl.config(anchor=CENTER)
    lbl.pack()

   
    lbl1 = Label(window, text=key)
    lbl1.config(anchor=CENTER)
    lbl1.pack()

    def Copy_Key():
        pyperclip.copy(lbl1.cget("text"))        
        
    btn = Button(window, text = "Copy Key", command=Copy_Key)
    btn.pack(pady=5)

    def Done():
        Password_Vault()   

    btn = Button(window, text = "Done", command=Done)
    btn.pack(pady=5)

def Reset_Screen():
    for widget in window.winfo_children():
        widget.destroy()


    window.geometry("500x250")

    lbl = Label(window, text="Please Enter Recovery Key")
    lbl.config(anchor=CENTER)
    lbl.pack()

    txt = Entry(window, width=20)
    txt.pack()
    txt.focus()

    lbl1 = Label(window)
    lbl1.config(anchor=CENTER)
    lbl1.pack()

    def Get_Recovery_Key():
        recoveryKeyCheck = Hash_Password(str(txt.get()).encode('utf-8'))
        cursor.execute('SELECT * FROM masterpassword WHERE id = 1 AND recoveryKey = ?', [(recoveryKeyCheck)])
        return cursor.fetchall()  

    def Check_Recovery_Key():
        checked = Get_Recovery_Key()

        if checked:
            Home_Screen()
        else:
            txt.delete(0, 'end')
            lbl1.config(text='You have entered the wrong key')

        
    btn = Button(window, text = "Check Key", command=Check_Recovery_Key)
    btn.pack(pady=5)



def Login_Screen():
    window.geometry("350x170")

    lbl = Label(window, text="Please Enter Master Password")
    lbl.config(anchor=CENTER)
    lbl.pack()

    txt = Entry(window, width=20, show="*")
    txt.pack()
    txt.focus()

    lbl1 = Label(window)
    lbl1.config(anchor=CENTER)
    lbl1.pack(side=TOP)

    def getMasterPassword():
       checkhashedpassword =  Hash_Password(txt.get().encode('utf-8')) 


       global encryptionKey
       encryptionKey = base64.urlsafe_b64encode(kdf.derive(txt.get().encode()))


       cursor.execute("SELECT * FROM masterpassword WHERE id = 1 AND password = ?", [(checkhashedpassword)])
       print(checkhashedpassword)
       return cursor. fetchall()

    def Check_Password():
        match = getMasterPassword()

        print(match)
       
        
        if match:
            Password_Vault()
        else:
            txt.delete(0, 'end')
            lbl1.config(text="You have entered the wrong password!")

    def Reset_Password():
        Reset_Screen()



    btn = Button(window, text = "Submit", command=Check_Password)
    btn.pack(pady=1)

    btn = Button(window, text = "Reset Password", command=Reset_Password)
    btn.pack(pady=1)

def Password_Vault():
    for widget in window.winfo_children():
        widget.destroy()

    def addEntry():
        text1 = "Application"
        text2 = "Username"
        text3 = "Password"

        application = encrypt(popUp(text1).encode(), encryptionKey)
        username = encrypt(popUp(text2).encode(), encryptionKey)
        password = encrypt(popUp(text3).encode(), encryptionKey)

        insert_fields = """Insert INTO vault(application, username, password)
        VALUES(?, ?, ?)"""
        cursor.execute(insert_fields, (application, username, password))
        db.commit()

        Password_Vault()
        
    def removeEntry(input):
        cursor.execute("DELETE FROM vault WHERE id = ?", (input,))
        db.commit()

        Password_Vault()

    window.geometry("1050x700")
    window.resizable(height=None, width=None)
    lbl = Label(window, text="Welcome to Acid's Password Vault!")
    lbl.grid(column=1)

    btn = Button(window, text = "+", command = addEntry)
    btn.grid(column=1, pady=10)

    lbl = Label(window, text="Application")
    lbl.grid(row=2, column=0, padx=80)
    lbl = Label(window, text="Username")
    lbl.grid(row=2, column=1, padx=80)
    lbl = Label(window, text="Password")
    lbl.grid(row=2, column=2, padx=80)

    cursor.execute("SELECT * FROM vault")
    if(cursor.fetchall() != None):
        i = 0
        while True:
            cursor.execute('SELECT * FROM vault')
            array = cursor.fetchall()

            if (len(array) == 0):
                break

            lbl1 = Label(window, text=(decrypt(array[i][1], encryptionKey)))
            lbl1.grid(column=0, row=(i+3))
            lbl2 = Label(window, text=(decrypt(array[i][2], encryptionKey)))
            lbl2.grid(column=1, row=(i+3))
            lbl3 = Label(window, text=(decrypt(array[i][3], encryptionKey)))
            lbl3.grid(column=2, row=(i+3))

            btn = Button(window, text="Delete", command= partial(removeEntry, array[i][0]))
            btn.grid(column=3, row=i+3, pady=10)

            i=i+1

            cursor.execute("SELECT * FROM vault")
            if (len(cursor.fetchall()) <= i):
                break



cursor.execute("SELECT * FROM masterpassword")
if cursor.fetchall():
    Login_Screen()
else:
    Home_Screen()
window.mainloop()


# In[ ]: