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
113
114
115
116
117
118
119
120
121
122
123
124
125
|
"""
This module provides a basic interface for connecting to and interacting with a SQLite database.
It includes functions for creating connections, executing queries, and retrieving results.
"""
import sqlite3
import sys
import os
VAULT_DECRYPTED = 'vault.sqlite'
VAULT_ENCRYPTED = 'vault.sqlite.aes'
def create_table() -> None:
"""Create the accounts table within the vault database."""
db_connection = sqlite3.connect(VAULT_DECRYPTED)
cursor = db_connection.cursor()
cursor.execute(
''' CREATE TABLE IF NOT EXISTS accounts (uuid text, application text,
username text, password text, url text) '''
)
db_connection.commit()
db_connection.close()
def check_table() -> bool:
"""Check if the 'accounts' table exists within the vault database."""
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.lower() == 'y':
create_table()
check = True
else:
sys.exit('Program aborted upon user request.')
else:
check = True
db_connection.commit()
db_connection.close()
return check
def add_account(uuid: str, application: str, username: str, password: str,
url: str) -> None:
"""Add a new account within the vault database."""
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()
def delete_account(uuid: str) -> None:
"""Delete an account within the vault database by its unique ID."""
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()
def find_account(uuid: str) -> list:
"""Find an account within the vault database by its unique ID."""
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
def find_accounts() -> list:
"""Return all accounts stored within the vault database."""
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:
"""Update an account within the vault database by its unique ID."""
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:
"""Purge the 'accounts' table within the vault database."""
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:
"""Purge the entire vault database."""
os.remove(VAULT_DECRYPTED)
|
