32 lines
978 B
Python
32 lines
978 B
Python
import os
|
|
|
|
import yaml
|
|
|
|
from ..config import Config, Setting
|
|
from ..exceptions import BackupPasswordKeyInvalid
|
|
from ..logger import getLogger
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
|
|
class Password():
|
|
def __init__(self, config: Config):
|
|
self.config = config
|
|
|
|
def resolve(self, password=None):
|
|
if password is None:
|
|
password = self.config.get(Setting.BACKUP_PASSWORD)
|
|
if len(password) == 0:
|
|
return None
|
|
if password.startswith("!secret "):
|
|
if not os.path.isfile(self.config.get(Setting.SECRETS_FILE_PATH)):
|
|
raise BackupPasswordKeyInvalid()
|
|
with open(self.config.get(Setting.SECRETS_FILE_PATH)) as f:
|
|
secrets_yaml = yaml.load(f, Loader=yaml.SafeLoader)
|
|
key = password[len("!secret "):]
|
|
if key not in secrets_yaml:
|
|
raise BackupPasswordKeyInvalid()
|
|
return str(secrets_yaml[key])
|
|
else:
|
|
return password
|