add web
This commit is contained in:
17
db.py
17
db.py
@@ -40,6 +40,7 @@ def init_db():
|
||||
state TEXT DEFAULT 'pending',
|
||||
scan_status TEXT DEFAULT 'pending',
|
||||
scan_result TEXT,
|
||||
scan_details_with_code TEXT,
|
||||
issues_count INTEGER DEFAULT 0,
|
||||
security_issues INTEGER DEFAULT 0,
|
||||
ai_review TEXT,
|
||||
@@ -72,16 +73,17 @@ class PRScanDB:
|
||||
"""PR 扫描结果数据库操作类"""
|
||||
|
||||
@staticmethod
|
||||
def save_pr_scan(pr_info: Dict[str, Any], scan_results: Dict[str, Any],
|
||||
report_path: str = None) -> int:
|
||||
def save_pr_scan(pr_info: Dict[str, Any], scan_results: Dict[str, Any],
|
||||
report_path: str = None, scan_details_with_code: Dict = None) -> int:
|
||||
"""
|
||||
保存 PR 扫描结果
|
||||
|
||||
|
||||
Args:
|
||||
pr_info: PR 信息
|
||||
scan_results: 扫描结果
|
||||
report_path: 报告文件路径
|
||||
|
||||
scan_details_with_code: 带代码片段的扫描详情
|
||||
|
||||
Returns:
|
||||
扫描记录 ID
|
||||
"""
|
||||
@@ -116,6 +118,7 @@ class PRScanDB:
|
||||
author = ?,
|
||||
scan_status = ?,
|
||||
scan_result = ?,
|
||||
scan_details_with_code = ?,
|
||||
issues_count = ?,
|
||||
security_issues = ?,
|
||||
ai_review = ?,
|
||||
@@ -129,6 +132,7 @@ class PRScanDB:
|
||||
pr_info.get('author'),
|
||||
'completed',
|
||||
json.dumps(scan_results, ensure_ascii=False),
|
||||
json.dumps(scan_details_with_code, ensure_ascii=False) if scan_details_with_code else None,
|
||||
issues_count,
|
||||
security_issues,
|
||||
json.dumps(scan_results.get('ai', {}), ensure_ascii=False),
|
||||
@@ -143,9 +147,9 @@ class PRScanDB:
|
||||
INSERT INTO pr_scans (
|
||||
pr_number, repo_name, pr_title, pr_url,
|
||||
source_branch, target_branch, author,
|
||||
state, scan_status, scan_result,
|
||||
state, scan_status, scan_result, scan_details_with_code,
|
||||
issues_count, security_issues, ai_review, report_path
|
||||
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
''', (
|
||||
pr_info.get('pr_number'),
|
||||
pr_info.get('repo_name'),
|
||||
@@ -157,6 +161,7 @@ class PRScanDB:
|
||||
'open',
|
||||
'completed',
|
||||
json.dumps(scan_results, ensure_ascii=False),
|
||||
json.dumps(scan_details_with_code, ensure_ascii=False) if scan_details_with_code else None,
|
||||
issues_count,
|
||||
security_issues,
|
||||
json.dumps(scan_results.get('ai', {}), ensure_ascii=False),
|
||||
|
||||
Reference in New Issue
Block a user