From 6ef85d4968ebd453b812d5089a61fa42b1e6ab6c Mon Sep 17 00:00:00 2001
From: zengliwen <zengliwen@kylinos.cn>
Date: Thu, 24 Oct 2024 19:38:28 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E6=95=B0=E6=8D=AE=E5=8F=AF?=
 =?UTF-8?q?=E8=A7=86=E5=8C=96=E9=83=A8=E5=88=86=E4=BB=A3=E7=A0=81=EF=BC=9A?=
 =?UTF-8?q?=E5=AE=9E=E7=8E=B0=E8=BE=93=E5=85=A5=E6=95=B0=E6=8D=AE=E6=A0=A1?=
 =?UTF-8?q?=E9=AA=8C=EF=BC=9B=E8=BE=93=E5=87=BAhtml=E6=8A=A5=E5=91=8A?=
 =?UTF-8?q?=E5=88=B6=E4=BD=9C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/report/report_generator.py | 89 ++++++++++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 src/report/report_generator.py

diff --git a/src/report/report_generator.py b/src/report/report_generator.py
new file mode 100644
index 0000000..42d2759
--- /dev/null
+++ b/src/report/report_generator.py
@@ -0,0 +1,89 @@
+#-*- coding:utf-8 -*-
+"""
+# **********************************************************************************
+# Copyright (c) KylinSoft Co., Ltd. 2024.All rights reserved.
+# [kyclassifier] licensed under the Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#     http://license.coscl.org.cn/MulanPSL2
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
+# PURPOSE.
+# See the Mulan PSL v2 for more details.
+# **********************************************************************************
+"""
+import io
+import os
+import json
+import time
+import sys
+from jinja2 import Environment, FileSystemLoader, select_autoescape
+
+class ReportGenerator(object):
+    """
+    A generator to convert raw data to json file and html file
+    """
+    def __init__(self, category_path, id_path, result_path, file_name):
+        self.category_path = category_path
+        self.id_path = id_path
+        self.result_path = result_path
+        self.file_name = file_name
+        self.flag = os.O_WRONLY | os.O_CREAT | os.O_TRUNC
+        self.mod = 0o640
+        self.work_dir = os.path.dirname(__file__)
+
+    def _check_param_valid(self):
+        for path in [self.category_path, self.id_path, self.base_path]:
+            if not os.path.exists(path):
+                raise Exception("The input data path not exist")
+
+        if not os.path.exists(self.result_path):
+            try:
+                os.makedirs(self.result_path)
+            except Exception as err:
+                print("The input path not exist, generate report failed!")
+
+    def _format_report_data(slef):
+        pass
+
+    def _format_chart_data(self, pie_data,category_data):
+        pass
+    
+    def _generate_html_file(self):
+        start_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time()))
+        pie_data, category_data, id_data, base_data = self._format_report_data()
+        pie_data, category_data,sum_data = self._format_chart_data(pie_data,category_data)
+        env = Environment(loader=FileSystemLoader(os.path.join(os.path.dirname(__file__), 'template')),autoescape=select_autoescape(["html", "xml"]),enable_async=False)
+        template = env.get_template("category_and_id_report.html")
+        report_template = template.render(
+            start_time=start_time,
+            base_data = base_data,
+            list_data=id_data, 
+            category_data=category_data,
+            pie_data=pie_data, 
+            sum_data=sum_data 
+        )
+        try:
+            html_path = os.path.join(self.result_path, self.file_name)
+            with io.open(html_path,'w',encoding='utf-8') as hf:
+                hf.write(report_template) 
+        except (SyntaxError, OSError) as err:
+            raise Exception(err)
+
+    def generate_result_file(self):
+        self._check_param_valid()
+        self._generate_html_file()
+
+if __name__ == "__main__":
+    # category_path: 分类json文件路径
+    # id_path：分层josn文件路径
+    # result_path：输出文件路径
+    # base_path: 部署环境基础信息
+    # file_name：输出文件名
+    fenlei_json_file = sys.argv[1]
+    fenceng_json_file = sys.argv[2]
+    osinfo_json_file = sys.argv[3]
+    result_path = sys.argv[4]
+    result_file_name = sys.argv[5]
+    report_generator = ReportGenerator(fenlei_json_file, fenceng_json_file, osinfo_json_file, result_path, result_file_name)
+    report_generator.generate_result_file()
\ No newline at end of file
-- 
Gitee