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
126
127
128
129
130
131
132
133
134
135
136
137
|
"""
This script uses argparse to parse command line arguments.
It imports the required modules and sets up a parser with basic options for demonstration purposes.
"""
import argparse
from yoshi import crypto, database, process
def yoshi():
"""
CLI entry point logic, used to parse user inputs.
"""
parser = argparse.ArgumentParser(
description='Manage your username and passwords via a convenient CLI vault.'
)
# Top-level arguments
group_one = parser.add_mutually_exclusive_group()
group_one.add_argument(
'-n', '--new',
help='Create a new account.',
action='store_true'
)
group_one.add_argument(
'-l', '--list',
help='List all saved accounts.',
action='store_true'
)
group_one.add_argument(
'-e', '--edit',
help='Edit a saved account.',
action='store_true'
)
group_one.add_argument(
'-d', '--delete',
help='Delete a saved account.',
action='store_true'
)
group_one.add_argument(
'--purge',
help=(
'Purge all accounts and delete the vault. '
'(Caution: this will irreversibly destroy your data.)'
),
action='store_true'
)
group_one.add_argument(
'--encrypt',
help='Encrypt the vault.',
action='store_true'
)
group_one.add_argument(
'--decrypt',
help='Decrypt the vault.',
action='store_true'
)
# Encryption flags
group_two = parser.add_mutually_exclusive_group()
group_two.add_argument(
'-g', '--generate',
help=(
'When using the --encrypt option, generate a new encryption key.'
),
action='store_true'
)
group_two.add_argument(
'-k', '--keyfile',
help='Path to existing key file.',
action='store',
nargs=1,
type=str
)
# Edit flags
group_three = parser.add_argument_group()
group_three.add_argument(
'-u', '--uuid',
help=(
'When using the --edit or --delete options, provide the account UUID.'
),
action='store',
nargs=1,
type=str
)
group_three.add_argument(
'-f', '--field',
help=(
'When using the --edit option, specify the field to edit (integer index).'
),
action='store',
nargs=1,
type=int
)
args = parser.parse_args()
if args.decrypt:
if args.keyfile:
key = crypto.load_key(args.keyfile[0])
else:
key = input('Please enter your decryption key: ')
crypto.decrypt(key)
elif args.encrypt:
if args.generate:
key = crypto.generate_key()
print(
'WRITE THIS KEY DOWN SOMEWHERE SAFE. YOU WILL NOT BE ABLE TO DECRYPT '
'YOUR DATA WITHOUT IT!'
)
print(key.decode())
print('\n')
else:
if args.keyfile:
key = crypto.load_key(args.keyfile[0])
else:
key = input('Please enter your encryption key: ')
crypto.encrypt(key)
elif database.check_table():
if args.new:
process.create_account()
elif args.list:
process.list_accounts()
elif args.edit:
process.edit_account(args.uuid[0], args.field[0])
elif args.delete:
process.delete_account(args.uuid[0])
elif args.purge:
process.purge_accounts()
else:
raise TypeError(
'Please specify a command or use the --help flag for more information.'
)
if __name__ == "__main__":
yoshi()
|
