tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-10 13:07:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: support parse secret

Browse Source
This commit is contained in:
arkohut 2024-09-09 22:26:26 +08:00
parent f9e2b2261b
commit 6107c22def
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
memos/config.py
View File

@ -98,6 +98,21 @@ def dict_representer(dumper, data):
yaml.add_representer(OrderedDict, dict_representer) yaml.add_representer(OrderedDict, dict_representer)
# Custom representer for SecretStr
def secret_str_representer(dumper, data):
return dumper.represent_scalar("tag:yaml.org,2002:str", data.get_secret_value())
# Custom constructor for SecretStr
def secret_str_constructor(loader, node):
value = loader.construct_scalar(node)
return SecretStr(value)
yaml.add_representer(SecretStr, secret_str_representer)
yaml.add_constructor("tag:yaml.org,2002:str", secret_str_constructor)
def create_default_config(): def create_default_config():
config_path = Path.home() / ".memos" / "config.yaml" config_path = Path.home() / ".memos" / "config.yaml"
if not config_path.exists(): if not config_path.exists():