53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
from password import Password
|
|
from pwdfile import Pwdfile
|
|
|
|
|
|
class Credentials:
|
|
def __init__(self):
|
|
self.creds = Pwdfile().read()
|
|
|
|
def get_creds(self):
|
|
return self.creds
|
|
|
|
def read_cred(self, service):
|
|
return self.creds[service]
|
|
|
|
def expose_cred(self, service):
|
|
decrypted = {
|
|
"username": self.creds[service]["username"],
|
|
"password": Password().decrypt(
|
|
self.creds[service]["password"].encode()
|
|
)
|
|
}
|
|
return decrypted
|
|
|
|
def create_cred(self, service, username, password=None):
|
|
if password is None:
|
|
password = Password.generate(mn=64)
|
|
|
|
new_entry = {
|
|
"username": username,
|
|
"password": Password().encrypt(password).decode()
|
|
}
|
|
self.creds[service] = new_entry
|
|
Pwdfile().write(self.creds)
|
|
|
|
def update_cred(self, service, username=None, password=None):
|
|
current_entry = self.creds[service]
|
|
if username is None and password is not None:
|
|
self.creds[service] = {
|
|
"username": current_entry["username"],
|
|
"password": Password().encrypt(password).decode()
|
|
}
|
|
elif username is not None and password is None:
|
|
self.creds[service] = {
|
|
"username": username,
|
|
"password": current_entry["password"]
|
|
}
|
|
Pwdfile().write(self.creds)
|
|
|
|
def delete_cred(self, service):
|
|
del self.creds[service]
|
|
Pwdfile().write(self.creds)
|
|
|