tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-06 03:05:25 +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

17
memos/config.py
View File

@ -50,7 +50,7 @@ class Settings(BaseSettings):
database_path: str = os.path.join(base_dir, "database.db")
default_library: str = "screenshots"
screenshots_dir: str = os.path.join(base_dir, "screenshots")
typesense_host: str = "localhost"
typesense_port: str = "8108"
typesense_protocol: str = "http"
@ -98,6 +98,21 @@ def dict_representer(dumper, data):
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():
config_path = Path.home() / ".memos" / "config.yaml"
if not config_path.exists():