mirror of
https://github.com/freqtrade/freqtrade.git
synced 2025-12-14 20:01:18 +00:00
68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
from copy import deepcopy
|
|
|
|
from freqtrade.constants import Config, ExchangeConfig
|
|
|
|
|
|
_SENSITIVE_KEYS = [
|
|
"exchange.key",
|
|
"exchange.api_key",
|
|
"exchange.apiKey",
|
|
"exchange.secret",
|
|
"exchange.password",
|
|
"exchange.uid",
|
|
"exchange.account_id",
|
|
"exchange.accountId",
|
|
"exchange.wallet_address",
|
|
"exchange.walletAddress",
|
|
"exchange.private_key",
|
|
"exchange.privateKey",
|
|
"telegram.token",
|
|
"telegram.chat_id",
|
|
"discord.webhook_url",
|
|
"api_server.password",
|
|
"webhook.url",
|
|
]
|
|
|
|
|
|
def sanitize_config(config: Config, *, show_sensitive: bool = False) -> Config:
|
|
"""
|
|
Remove sensitive information from the config.
|
|
:param config: Configuration
|
|
:param show_sensitive: Show sensitive information
|
|
:return: Configuration
|
|
"""
|
|
if show_sensitive:
|
|
return config
|
|
config = deepcopy(config)
|
|
for key in _SENSITIVE_KEYS:
|
|
if "." in key:
|
|
nested_keys = key.split(".")
|
|
nested_config = config
|
|
for nested_key in nested_keys[:-1]:
|
|
nested_config = nested_config.get(nested_key, {})
|
|
if nested_keys[-1] in nested_config:
|
|
nested_config[nested_keys[-1]] = "REDACTED"
|
|
else:
|
|
if key in config:
|
|
config[key] = "REDACTED"
|
|
|
|
return config
|
|
|
|
|
|
def remove_exchange_credentials(exchange_config: ExchangeConfig, dry_run: bool) -> None:
|
|
"""
|
|
Removes exchange keys from the configuration and specifies dry-run
|
|
Used for backtesting / hyperopt and utils.
|
|
Modifies the input dict!
|
|
:param exchange_config: Exchange configuration
|
|
:param dry_run: If True, remove sensitive keys from the exchange configuration
|
|
"""
|
|
if not dry_run:
|
|
return
|
|
|
|
for key in [k for k in _SENSITIVE_KEYS if k.startswith("exchange.")]:
|
|
if "." in key:
|
|
key1 = key.removeprefix("exchange.")
|
|
if key1 in exchange_config:
|
|
exchange_config[key1] = ""
|