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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
# Import Python modules
import sqlite3
import sys
import os
# Specify the name of the vault database
vault_decrypted = 'vault.sqlite'
vault_encrypted = 'vault.sqlite.aes'
# Create the accounts table inside the vault database
def create_table() -> None:
db_connection = sqlite3.connect(vault_decrypted)
cursor = db_connection.cursor()
cursor.execute(
''' CREATE TABLE accounts (uuid text, application text, username text, password text, url text) ''')
db_connection.commit()
db_connection.close()
# Check if the account table exists within the database
def table_check() -> bool:
check = False
db_connection = sqlite3.connect(vault_decrypted)
cursor = db_connection.cursor()
cursor.execute(
''' SELECT count(name) FROM sqlite_master WHERE type='table' AND name='accounts' ''')
if cursor.fetchone()[0] != 1:
user_choice = input(
'Password vault does not exist. Would you like to create it now? (y/n): ')
if user_choice == 'y' or user_choice == 'Y':
create_table()
check = True
else:
sys.exit('Program aborted upon user request.')
else:
check = True
db_connection.commit()
db_connection.close()
return check
# Add a new account to the database
def create_account(uuid: str, application: str, username: str, password: str,
url: str) -> None:
db_connection = sqlite3.connect(vault_decrypted)
cursor = db_connection.cursor()
cursor.execute(
''' INSERT INTO accounts VALUES (:uuid,:application,:username,:password,:url) ''',
{'uuid': uuid, 'application': application, 'username': username,
'password': password, 'url': url})
db_connection.commit()
db_connection.close()
# Delete an account with a specified UUID
def delete_account(uuid: str) -> None:
db_connection = sqlite3.connect(vault_decrypted)
cursor = db_connection.cursor()
cursor.execute(
''' DELETE FROM accounts WHERE uuid = :uuid ''', {'uuid': uuid})
db_connection.commit()
db_connection.close()
# Find a specific account by UUID
def find_account(uuid: str) -> list:
db_connection = sqlite3.connect(vault_decrypted)
cursor = db_connection.cursor()
cursor.execute(
''' SELECT * FROM accounts WHERE uuid = :uuid ''', {'uuid': uuid})
account = cursor.fetchall()
db_connection.close()
return account
# Return all accounts found within the database table
# The `accounts` variable is a list of tuples
def find_accounts() -> list:
db_connection = sqlite3.connect(vault_decrypted)
cursor = db_connection.cursor()
cursor.execute(''' SELECT * FROM accounts ''')
accounts = cursor.fetchall()
db_connection.close()
return accounts
def update_account(field_name: str, new_value: str, uuid: str) -> None:
queries = {
'application': 'UPDATE accounts SET application = :new_value WHERE uuid = :uuid',
'username': 'UPDATE accounts SET username = :new_value WHERE uuid = :uuid',
'password': 'UPDATE accounts SET password = :new_value WHERE uuid = :uuid',
'url': 'UPDATE accounts SET url = :new_value WHERE uuid = :uuid'
}
db_connection = sqlite3.connect(vault_decrypted)
cursor = db_connection.cursor()
cursor.execute(queries[field_name],
{'new_value': new_value, 'uuid': uuid})
db_connection.commit()
db_connection.close()
def purge_table() -> None:
db_connection = sqlite3.connect(vault_decrypted)
cursor = db_connection.cursor()
cursor.execute(''' DROP TABLE accounts ''')
db_connection.commit()
db_connection.close()
def purge_database() -> None:
os.remove(vault_decrypted)
|
