kudzues/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-05 01:52:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: deduplicate legacy section mapping in ConfigHelper (#32715)

Browse Source
This commit is contained in:
Olexandr88
2026-03-09 08:43:06 +03:00
committed by GitHub
parent f2d3feca66
commit 8b1ea3a8f5
1 changed files with 14 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
scripts/stress-test/common/config_helper.py
View File

@@ -6,6 +6,13 @@ from typing import Any
class ConfigHelper: class ConfigHelper:
_LEGACY_SECTION_MAP = {
"admin_config": "admin",
"token_config": "auth",
"app_config": "app",
"api_key_config": "api_key",
}
"""Helper class for reading and writing configuration files.""" """Helper class for reading and writing configuration files."""
def __init__(self, base_dir: Path | None = None): def __init__(self, base_dir: Path | None = None):
@@ -50,14 +57,8 @@ class ConfigHelper:
Dictionary containing config data, or None if file doesn't exist Dictionary containing config data, or None if file doesn't exist
""" """
# Provide backward compatibility for old config names # Provide backward compatibility for old config names
if filename in ["admin_config", "token_config", "app_config", "api_key_config"]: if filename in self._LEGACY_SECTION_MAP:
section_map = { return self.get_state_section(self._LEGACY_SECTION_MAP[filename])
"admin_config": "admin",
"token_config": "auth",
"app_config": "app",
"api_key_config": "api_key",
}
return self.get_state_section(section_map[filename])
config_path = self.get_config_path(filename) config_path = self.get_config_path(filename)
@@ -85,14 +86,11 @@ class ConfigHelper:
True if successful, False otherwise True if successful, False otherwise
""" """
# Provide backward compatibility for old config names # Provide backward compatibility for old config names
if filename in ["admin_config", "token_config", "app_config", "api_key_config"]: if filename in self._LEGACY_SECTION_MAP:
section_map = { return self.update_state_section(
"admin_config": "admin", self._LEGACY_SECTION_MAP[filename],
"token_config": "auth", data,
"app_config": "app", )
"api_key_config": "api_key",
}
return self.update_state_section(section_map[filename], data)
self.ensure_config_dir() self.ensure_config_dir()
config_path = self.get_config_path(filename) config_path = self.get_config_path(filename)