PaiGramTeam/PamGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PamGram.git synced 2024-11-25 07:36:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

🔒 Filter bot_token in log

Browse Source
This commit is contained in:
xtaodada 2023-06-15 13:08:50 +08:00
parent dcbb6b1b60
commit 32dcb5bd6b
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
utils/log/_traceback.py
View File

@ -22,6 +22,7 @@ from rich.traceback import (
Traceback as BaseTraceback, Traceback as BaseTraceback,
) )
from core.config import config
from utils.log._style import MonokaiProStyle from utils.log._style import MonokaiProStyle
if TYPE_CHECKING: if TYPE_CHECKING:
@ -199,7 +200,7 @@ class Traceback(BaseTraceback):
name=frame_summary.f_code.co_name, name=frame_summary.f_code.co_name,
locals={ locals={
key: pretty.traverse( key: pretty.traverse(
value, Traceback.filter_value(value),
max_length=locals_max_length, max_length=locals_max_length,
max_string=locals_max_string, max_string=locals_max_string,
max_depth=locals_max_depth, max_depth=locals_max_depth,
@ -343,3 +344,9 @@ class Traceback(BaseTraceback):
if frame.locals if frame.locals
else syntax else syntax
) )
@staticmethod
def filter_value(value: Any) -> Any:
if isinstance(value, str):
return value.replace(config.bot_token, "TOKEN")
return value