openubmc-bingo 0.5.277__py3-none-any.whl → 0.6.2__py3-none-any.whl

bmcgo/tasks/task.py

@@ -20,13 +20,13 @@ import tempfile
 import filecmp
 from string import Template
 from multiprocessing import Process
-from collections import OrderedDict
 
 import yaml
-from conans.model.profile import Profile
 from bmcgo.utils.config import Config
 from bmcgo.utils.tools import Tools
 from bmcgo import errors
+from bmcgo import misc
+from bmcgo.component.deploy import GraphNode
 from bmcgo.utils.component_post import ComponentPost
 from bmcgo.functional.simple_sign import BmcgoCommand as SimpleSign
 
@@ -40,12 +40,21 @@ class Task(Process):
 
     def __init__(self, config: Config, work_name=""):
         super(Task, self).__init__()
+        self.community_name = misc.community_name()
         log_file = os.path.join(config.temp_path, "log", "task.log")
         self.tools: Tools = Tools(work_name, log_file=log_file)
         self.log = self.tools.log
         self.config: Config = config
         self.built_type_check()
         self.work_name = work_name
+        # 记录每个组件的GraphNode信息
+        self.graph_nodes: dict[str, GraphNode] = {}
+        # 设置默认的厂商名，产品可以manifest.yml的base/vendor中设置
+        # manifest未配置vendor时，使用OPENUBMC_DEFAULT_VENDOR环境变量
+        # OPENUBMC_DEFAULT_VENDOR环境变量未设置时，使用openubmc
+        self.vendor = self.get_manufacture_config("base/vendor", misc.vendor())
+        os.environ["OPENUBMC_PRODUCT_VENDOR"] = self.vendor
+        self.customization_dir = os.path.join(self.config.build_path, "customizations")
 
     @property
     def conan_home(self):
@@ -210,7 +219,7 @@ class Task(Process):
         """
         return self.config.get_manufacture_config(key, default_val)
 
-    def get_profile_config(self) -> (Profile, OrderedDict):
+    def get_profile_config(self):
         return self.tools.get_profile_config(self.config.profile)
 
     def prepare_conan(self):
@@ -320,7 +329,15 @@ class Task(Process):
                 raise OSError(f"未知条件: {key}, 请检查此种条件是否在配置内")
         return True
 
-    def _component_cust_action(self, action: str):
+    def _component_cust_action_v2(self, action: str):
+        profile, _ = self.get_profile_config()
+        for name in os.listdir(self.customization_dir):
+            path_dir = os.path.join(self.customization_dir, name)
+            self.info(f"开始执行组件 {name} 定制化 {action}，定制脚本 {path_dir}/include/customization.py")
+            post = ComponentPost(self.config, path_dir, profile)
+            post.post_work(os.getcwd(), action)
+
+    def _component_cust_action_v1(self, action: str):
         conan_install = f"{self.config.build_path}/conan_install"
 
         # 优先处理rootfs定制化脚本
@@ -344,6 +361,12 @@ class Task(Process):
             post = ComponentPost(self.config, os.path.join(conan_install, comp), profile)
             post.post_work(os.getcwd(), action)
 
+    def _component_cust_action(self, action: str):
+        if misc.conan_v2():
+            return self._component_cust_action_v2(action)
+        else:
+            return self._component_cust_action_v1(action)
+
     def _local_signature(self, unsigned_file, cms_output, crl_output, ca_output):
         """自签名方法"""
         self_sign_config = self.config.bconfig.hpm_self_sign