gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix linting

Browse Source
This commit is contained in:
zrguo
2025-05-12 16:28:36 +08:00
parent 61a21f8d5d
commit cf4bb148fb
1 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
lightrag/kg/tidb_impl.py
View File

@@ -25,12 +25,21 @@ from sqlalchemy import create_engine, text # type: ignore
def sanitize_sensitive_info(data: dict) -> dict: def sanitize_sensitive_info(data: dict) -> dict:
sanitized_data = data.copy() sanitized_data = data.copy()
sensitive_fields = ['password', 'user', 'host', 'database', 'port', 'ssl_verify_cert', 'ssl_verify_identity'] sensitive_fields = [
for field in sensitive_fields: "password",
if field in sanitized_data: "user",
sanitized_data[field] = '***' "host",
"database",
"port",
"ssl_verify_cert",
"ssl_verify_identity",
]
for field_name in sensitive_fields:
if field_name in sanitized_data:
sanitized_data[field_name] = "***"
return sanitized_data return sanitized_data
class TiDB: class TiDB:
def __init__(self, config, **kwargs): def __init__(self, config, **kwargs):
self.host = config.get("host", None) self.host = config.get("host", None)
@@ -91,9 +100,10 @@ class TiDB:
result = conn.execute(text(sql), params) result = conn.execute(text(sql), params)
except Exception as e: except Exception as e:
sanitized_params = sanitize_sensitive_info(params) sanitized_params = sanitize_sensitive_info(params)
sanitized_params = sanitize_sensitive_info(params) sanitized_error = sanitize_sensitive_info({"error": str(e)})
sanitized_error = sanitize_sensitive_info({'error': str(e)}) logger.error(
logger.error(f"Tidb database,\nsql:{sql},\nparams:{sanitized_params},\nerror:{sanitized_error}") f"Tidb database,\nsql:{sql},\nparams:{sanitized_params},\nerror:{sanitized_error}"
)
raise raise
if multirows: if multirows:
rows = result.all() rows = result.all()
@@ -119,8 +129,10 @@ class TiDB:
conn.execute(text(sql), parameters=data) conn.execute(text(sql), parameters=data)
except Exception as e: except Exception as e:
sanitized_data = sanitize_sensitive_info(data) if data else None sanitized_data = sanitize_sensitive_info(data) if data else None
sanitized_error = sanitize_sensitive_info({'error': str(e)}) sanitized_error = sanitize_sensitive_info({"error": str(e)})
logger.error(f"Tidb database,\nsql:{sql},\ndata:{sanitized_data},\nerror:{sanitized_error}") logger.error(
f"Tidb database,\nsql:{sql},\ndata:{sanitized_data},\nerror:{sanitized_error}"
)
raise raise