sql-password-manager/main.py at main · BKPrograms/sql-password-manager · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
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!")