Added initial files

Author: BKPrograms

config.py

@@ -0,0 +1,6 @@
+MASTER_PWD_HASH = "YOUR MASTER PWD HASH GOES HERE"
+DB_NAME = "YOUR POSTGRESQL DB NAME GOES HERE"
+DB_UNAME = "YOUR POSTGRESQL USERNAME GOES HERE"  # "postgres" by default
+DB_PWD = "YOUR POSTGRESQL DB PASSWORD GOES HERE"
+DB_TABLE_NAME = 'YOUR DB TABLE NAME GOES HERE'  # Ex: 'public."Table"'
+SALT = b""  # Feel free to put any binary string here Ex: b".Y\x20\xBF\x045\xAC"

encrypt_funcs.py

@@ -0,0 +1,38 @@
+from Crypto.Cipher import AES
+from pbkdf2 import PBKDF2
+from base64 import *
+from config import SALT
+
+
+def encrypt(pass_to_encrypt, master_hash):
+    derived_key = PBKDF2(str(master_hash), SALT).read(32)
+
+    convert = pass_to_encrypt.encode()
+
+    cipher = AES.new(derived_key, AES.MODE_EAX)
+
+    nonce = cipher.nonce  # Unique to every transaction
+
+    ciphertext, tag = cipher.encrypt_and_digest(convert)
+
+    nonce_added = ciphertext + nonce
+
+    encoded_ciphertext = b64encode(nonce_added).decode()
+
+    return encoded_ciphertext
+
+
+def decrypt(pass_to_dec, master_hash):
+    pass_to_dec += "=" * (4 - len(pass_to_dec) % 4)
+
+    convert = b64decode(pass_to_dec)
+
+    key = PBKDF2(str(master_hash), SALT).read(32)
+
+    nonce = convert[-16:]
+
+    cipher = AES.new(key, AES.MODE_EAX, nonce=nonce)
+
+    plaintext = cipher.decrypt(convert[:-16])  # Excluding nonce
+
+    return plaintext.decode()

main.py

@@ -0,0 +1,88 @@
+import getpass
+import hashlib
+from config import *
+import psycopg2
+import encrypt_funcs
+
+
+def listfunc(decrypt, dbcursor):
+    dbcursor.execute('SELECT * FROM ' + DB_TABLE_NAME)
+
+    rows = dbcursor.fetchall()
+
+    if rows:
+        for row in rows:
+            print("-" * 20)
+            print("Website:", row[0])
+            print("Username:", row[1])
+
+            if decrypt:
+
+                print("Password:",
+                      encrypt_funcs.decrypt(row[2], MASTER_PWD_HASH))
+
+            else:
+                print("Password:", row[2])
+        print("-" * 20)
+
+    else:
+
+        print("No Records Found!")
+
+
+def list_encrypted_records(dbcursor):
+    listfunc(False, dbcursor)
+
+
+def list_decrypted_records(dbcursor):
+    user_pwd = getpass.getpass("\nEnter Master Password:").encode()
+    if hashlib.sha256(user_pwd).hexdigest() == MASTER_PWD_HASH:
+        listfunc(True, dbcursor)
+    else:
+        print("Sorry, wrong password!")
+
+
+def add_record(dbcursor):
+    website = input("\nWebsite: ")
+    name = input("Name: ")
+    password = input("Password: ")
+
+    dbcursor.execute(f'INSERT INTO {DB_TABLE_NAME} VALUES (%s, %s, %s)',
+                     (website, name,
+                      encrypt_funcs.encrypt(password, MASTER_PWD_HASH)))
+
+    print("Record Successfully Added!\n")
+
+
+if __name__ == '__main__':
+
+    print("\nSQL Password Manager")
+
+    user_master = getpass.getpass("\nEnter Master Password:").encode()
+
+    if hashlib.sha256(user_master).hexdigest() == MASTER_PWD_HASH:
+
+        connection = psycopg2.connect(dbname=DB_NAME, user=DB_UNAME,
+                                      password=DB_PWD)
+
+        cursor = connection.cursor()
+
+        print("1. List Encrypted Records")
+        print("2. List Decrypted Records")
+        print("3. Add Record\n")
+
+        userChoice = input("Enter Option: ")
+        if userChoice == "1":
+            list_encrypted_records(cursor)
+        elif userChoice == "2":
+            list_decrypted_records(cursor)
+        elif userChoice == "3":
+            add_record(cursor)
+        else:
+            print("Invalid Option!")
+
+        connection.commit()
+        connection.close()
+
+    else:
+        print("Wrong Master Password!")