first commit
This commit is contained in:
zj
387
reporter.py
Normal file
387
reporter.py
Normal file
@@ -0,0 +1,387 @@
|
||||
"""
|
||||
ServerGuard - 报告生成模块
|
||||
|
||||
负责将检测结果格式化为各种输出格式。
|
||||
"""
|
||||
|
||||
import json
|
||||
import csv
|
||||
import os
|
||||
from typing import Dict, Any, List
|
||||
from datetime import datetime
|
||||
from io import StringIO
|
||||
|
||||
try:
|
||||
from rich.console import Console
|
||||
from rich.table import Table
|
||||
from rich.panel import Panel
|
||||
from rich import box
|
||||
HAS_RICH = True
|
||||
except ImportError:
|
||||
HAS_RICH = False
|
||||
|
||||
|
||||
class ReportGenerator:
|
||||
"""报告生成器类"""
|
||||
|
||||
def __init__(self):
|
||||
self.console = Console() if HAS_RICH else None
|
||||
|
||||
def generate_report(self, data: Dict[str, Any], format_type: str = 'text') -> str:
|
||||
"""
|
||||
根据指定格式生成报告。
|
||||
|
||||
Args:
|
||||
data: 检测结果数据
|
||||
format_type: 报告格式 (text, json, csv, html)
|
||||
|
||||
Returns:
|
||||
str: 格式化的报告内容
|
||||
"""
|
||||
if format_type == 'json':
|
||||
return self._format_json_report(data)
|
||||
elif format_type == 'csv':
|
||||
return self._format_csv_report(data)
|
||||
elif format_type == 'html':
|
||||
return self._format_html_report(data)
|
||||
else:
|
||||
return self._format_text_report(data)
|
||||
|
||||
def save_report(self, data: Dict[str, Any], format_type: str, filepath: str):
|
||||
"""
|
||||
保存报告到文件。
|
||||
|
||||
Args:
|
||||
data: 检测结果数据
|
||||
format_type: 报告格式
|
||||
filepath: 输出文件路径
|
||||
"""
|
||||
report = self.generate_report(data, format_type)
|
||||
|
||||
# 确保目录存在
|
||||
os.makedirs(os.path.dirname(filepath) or '.', exist_ok=True)
|
||||
|
||||
with open(filepath, 'w', encoding='utf-8') as f:
|
||||
f.write(report)
|
||||
|
||||
def _format_json_report(self, data: Dict[str, Any]) -> str:
|
||||
"""生成 JSON 格式报告。"""
|
||||
return json.dumps(data, indent=2, ensure_ascii=False, default=str)
|
||||
|
||||
def _format_csv_report(self, data: Dict[str, Any]) -> str:
|
||||
"""生成 CSV 格式报告。"""
|
||||
output = StringIO()
|
||||
writer = csv.writer(output)
|
||||
|
||||
# 写入基本信息
|
||||
writer.writerow(['ServerGuard Diagnostic Report'])
|
||||
writer.writerow(['Scan Type', data.get('scan_type', 'unknown')])
|
||||
writer.writerow(['Timestamp', data.get('timestamp', '')])
|
||||
writer.writerow([])
|
||||
|
||||
# 写入各模块数据
|
||||
for module_name, module_data in data.get('modules', {}).items():
|
||||
writer.writerow([f'Module: {module_name.upper()}'])
|
||||
writer.writerow(['Status', module_data.get('status', 'unknown')])
|
||||
|
||||
# 展平嵌套字典
|
||||
self._write_dict_to_csv(writer, module_data, prefix='')
|
||||
writer.writerow([])
|
||||
|
||||
return output.getvalue()
|
||||
|
||||
def _write_dict_to_csv(self, writer, data: Dict[str, Any], prefix: str = ''):
|
||||
"""辅助函数：将字典写入 CSV"""
|
||||
for key, value in data.items():
|
||||
if key == 'status':
|
||||
continue
|
||||
full_key = f"{prefix}.{key}" if prefix else key
|
||||
|
||||
if isinstance(value, dict):
|
||||
self._write_dict_to_csv(writer, value, full_key)
|
||||
elif isinstance(value, list):
|
||||
writer.writerow([full_key, ', '.join(str(v) for v in value)])
|
||||
else:
|
||||
writer.writerow([full_key, value])
|
||||
|
||||
def _format_text_report(self, data: Dict[str, Any]) -> str:
|
||||
"""生成纯文本格式报告。"""
|
||||
lines = []
|
||||
|
||||
# 报告头部
|
||||
lines.append("=" * 70)
|
||||
lines.append("ServerGuard 硬件健康诊断报告")
|
||||
lines.append("=" * 70)
|
||||
lines.append(f"扫描类型: {data.get('scan_type', 'unknown').upper()}")
|
||||
lines.append(f"生成时间: {data.get('timestamp', '')}")
|
||||
if 'stress_duration' in data:
|
||||
lines.append(f"压力测试时长: {data['stress_duration']} 秒")
|
||||
lines.append("=" * 70)
|
||||
lines.append("")
|
||||
|
||||
# 各模块结果
|
||||
for module_name, module_data in data.get('modules', {}).items():
|
||||
lines.append(f"\n[{module_name.upper()}]")
|
||||
lines.append("-" * 70)
|
||||
|
||||
status = module_data.get('status', 'unknown')
|
||||
status_symbol = '✓' if status == 'success' else '⚠' if status == 'warning' else '✗'
|
||||
lines.append(f"状态: {status_symbol} {status.upper()}")
|
||||
|
||||
if 'error' in module_data:
|
||||
lines.append(f"错误: {module_data['error']}")
|
||||
|
||||
# 格式化模块特定数据
|
||||
self._format_module_text(lines, module_name, module_data)
|
||||
|
||||
lines.append("")
|
||||
|
||||
# 报告尾部
|
||||
lines.append("=" * 70)
|
||||
lines.append("报告结束")
|
||||
lines.append("=" * 70)
|
||||
|
||||
return '\n'.join(lines)
|
||||
|
||||
def _format_module_text(self, lines: List[str], module_name: str, data: Dict[str, Any]):
|
||||
"""格式化特定模块的文本输出"""
|
||||
if module_name == 'system':
|
||||
if 'cpu' in data:
|
||||
cpu = data['cpu']
|
||||
lines.append(f"CPU: {cpu.get('model', 'N/A')}")
|
||||
lines.append(f" 核心数: {cpu.get('cores', 'N/A')} 核 / {cpu.get('threads', 'N/A')} 线程")
|
||||
|
||||
if 'memory' in data:
|
||||
mem = data['memory']
|
||||
lines.append(f"内存: 总计 {mem.get('total_gb', 'N/A')} GB, {mem.get('slots_used', 'N/A')} 个插槽")
|
||||
|
||||
if 'storage' in data:
|
||||
lines.append(f"存储设备: {len(data['storage'])} 个设备")
|
||||
|
||||
elif module_name == 'cpu':
|
||||
if 'temperature' in data:
|
||||
temp = data['temperature']
|
||||
lines.append(f"CPU 温度: {temp.get('current_c', 'N/A')}°C")
|
||||
if 'mce_errors' in data:
|
||||
mce = data['mce_errors']
|
||||
lines.append(f"MCE 错误: {mce.get('count', 0)} 个")
|
||||
if 'stress_test' in data:
|
||||
stress = data['stress_test']
|
||||
lines.append(f"压力测试: {'通过' if stress.get('passed') else '失败'}")
|
||||
lines.append(f" 运行时长: {stress.get('duration_seconds', 'N/A')} 秒")
|
||||
|
||||
elif module_name == 'memory':
|
||||
if 'ecc_status' in data:
|
||||
ecc = data['ecc_status']
|
||||
lines.append(f"ECC 支持: {'是' if ecc.get('supported') else '否'}")
|
||||
if ecc.get('errors', 0) > 0:
|
||||
lines.append(f"ECC 错误: {ecc['errors']} 个")
|
||||
if 'stress_test' in data:
|
||||
st = data['stress_test']
|
||||
lines.append(f"内存压力测试: {'通过' if st.get('passed') else '失败'}")
|
||||
if st.get('tool'):
|
||||
lines.append(f" 使用工具: {st.get('tool')}")
|
||||
if st.get('size_mb'):
|
||||
lines.append(f" 测试大小: {st.get('size_mb')} MB")
|
||||
|
||||
elif module_name == 'storage':
|
||||
for device in data.get('devices', []):
|
||||
lines.append(f"设备 {device.get('name', 'N/A')}:")
|
||||
lines.append(f" 型号: {device.get('model', 'N/A')}")
|
||||
lines.append(f" 健康状态: {device.get('health', 'N/A')}")
|
||||
if 'smart_status' in device:
|
||||
smart = device['smart_status']
|
||||
lines.append(f" SMART: {smart.get('overall', 'N/A')}")
|
||||
|
||||
elif module_name == 'sensors':
|
||||
if 'temperatures' in data:
|
||||
lines.append("温度传感器:")
|
||||
for name, value in data['temperatures'].items():
|
||||
lines.append(f" {name}: {value}°C")
|
||||
if 'voltages' in data:
|
||||
lines.append("电压传感器:")
|
||||
for name, value in data['voltages'].items():
|
||||
lines.append(f" {name}: {value}V")
|
||||
|
||||
elif module_name == 'logs':
|
||||
if 'hardware_errors' in data:
|
||||
errors = data['hardware_errors']
|
||||
total = sum(errors.values())
|
||||
lines.append(f"硬件错误总计: {total} 个")
|
||||
for error_type, count in errors.items():
|
||||
if count > 0:
|
||||
lines.append(f" {error_type}: {count} 个")
|
||||
|
||||
def _format_html_report(self, data: Dict[str, Any]) -> str:
|
||||
"""生成 HTML 格式报告。"""
|
||||
html_parts = []
|
||||
|
||||
# HTML 头部
|
||||
html_parts.append("""<!DOCTYPE html>
|
||||
<html lang="zh-CN">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>ServerGuard 诊断报告</title>
|
||||
<style>
|
||||
body {
|
||||
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
|
||||
line-height: 1.6;
|
||||
color: #333;
|
||||
max-width: 1200px;
|
||||
margin: 0 auto;
|
||||
padding: 20px;
|
||||
background: #f5f5f5;
|
||||
}
|
||||
.header {
|
||||
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
|
||||
color: white;
|
||||
padding: 30px;
|
||||
border-radius: 10px;
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
.header h1 {
|
||||
margin: 0;
|
||||
font-size: 2em;
|
||||
}
|
||||
.header .meta {
|
||||
margin-top: 10px;
|
||||
opacity: 0.9;
|
||||
}
|
||||
.module {
|
||||
background: white;
|
||||
border-radius: 8px;
|
||||
padding: 20px;
|
||||
margin-bottom: 20px;
|
||||
box-shadow: 0 2px 4px rgba(0,0,0,0.1);
|
||||
}
|
||||
.module-header {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
border-bottom: 2px solid #eee;
|
||||
padding-bottom: 10px;
|
||||
margin-bottom: 15px;
|
||||
}
|
||||
.module-title {
|
||||
font-size: 1.5em;
|
||||
font-weight: bold;
|
||||
color: #444;
|
||||
}
|
||||
.status {
|
||||
padding: 5px 15px;
|
||||
border-radius: 20px;
|
||||
font-weight: bold;
|
||||
font-size: 0.9em;
|
||||
}
|
||||
.status-success { background: #d4edda; color: #155724; }
|
||||
.status-warning { background: #fff3cd; color: #856404; }
|
||||
.status-error { background: #f8d7da; color: #721c24; }
|
||||
.status-unknown { background: #e2e3e5; color: #383d41; }
|
||||
.info-grid {
|
||||
display: grid;
|
||||
grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
|
||||
gap: 15px;
|
||||
}
|
||||
.info-item {
|
||||
padding: 10px;
|
||||
background: #f8f9fa;
|
||||
border-radius: 5px;
|
||||
}
|
||||
.info-label {
|
||||
font-weight: bold;
|
||||
color: #666;
|
||||
font-size: 0.9em;
|
||||
}
|
||||
.info-value {
|
||||
margin-top: 5px;
|
||||
font-size: 1.1em;
|
||||
}
|
||||
.footer {
|
||||
text-align: center;
|
||||
color: #666;
|
||||
margin-top: 30px;
|
||||
padding: 20px;
|
||||
}
|
||||
.error-box {
|
||||
background: #f8d7da;
|
||||
border: 1px solid #f5c6cb;
|
||||
color: #721c24;
|
||||
padding: 15px;
|
||||
border-radius: 5px;
|
||||
margin: 10px 0;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>""")
|
||||
|
||||
# 报告头部
|
||||
html_parts.append(f"""
|
||||
<div class="header">
|
||||
<h1>🔧 ServerGuard 硬件健康诊断报告</h1>
|
||||
<div class="meta">
|
||||
扫描类型: {data.get('scan_type', 'unknown').upper()} |
|
||||
生成时间: {data.get('timestamp', '')}
|
||||
</div>
|
||||
</div>""")
|
||||
|
||||
# 各模块结果
|
||||
for module_name, module_data in data.get('modules', {}).items():
|
||||
status = module_data.get('status', 'unknown')
|
||||
status_class = f'status-{status}'
|
||||
|
||||
html_parts.append(f"""
|
||||
<div class="module">
|
||||
<div class="module-header">
|
||||
<span class="module-title">{module_name.upper()}</span>
|
||||
<span class="status {status_class}">{status.upper()}</span>
|
||||
</div>""")
|
||||
|
||||
if 'error' in module_data:
|
||||
html_parts.append(f"""
|
||||
<div class="error-box">
|
||||
<strong>错误:</strong> {module_data['error']}
|
||||
</div>""")
|
||||
else:
|
||||
html_parts.append(' <div class="info-grid">')
|
||||
self._format_module_html(html_parts, module_name, module_data)
|
||||
html_parts.append(' </div>')
|
||||
|
||||
html_parts.append(' </div>')
|
||||
|
||||
# 报告尾部
|
||||
html_parts.append("""
|
||||
<div class="footer">
|
||||
<p>由 ServerGuard 生成</p>
|
||||
</div>
|
||||
</body>
|
||||
</html>""")
|
||||
|
||||
return '\n'.join(html_parts)
|
||||
|
||||
def _format_module_html(self, html_parts: List[str], module_name: str, data: Dict[str, Any]):
|
||||
"""格式化特定模块的 HTML 输出"""
|
||||
for key, value in data.items():
|
||||
if key == 'status':
|
||||
continue
|
||||
|
||||
display_key = key.replace('_', ' ').title()
|
||||
|
||||
if isinstance(value, dict):
|
||||
html_parts.append(f"""
|
||||
<div class="info-item">
|
||||
<div class="info-label">{display_key}</div>
|
||||
<div class="info-value">{len(value)} 项数据</div>
|
||||
</div>""")
|
||||
elif isinstance(value, list):
|
||||
html_parts.append(f"""
|
||||
<div class="info-item">
|
||||
<div class="info-label">{display_key}</div>
|
||||
<div class="info-value">{len(value)} 个项目</div>
|
||||
</div>""")
|
||||
else:
|
||||
html_parts.append(f"""
|
||||
<div class="info-item">
|
||||
<div class="info-label">{display_key}</div>
|
||||
<div class="info-value">{value}</div>
|
||||
</div>""")
|
||||
Reference in New Issue
Block a user