pytbox 0.0.6__tar.gz → 0.0.7__tar.gz

{pytbox-0.0.6 → pytbox-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytbox
-Version: 0.0.6
+Version: 0.0.7
 Summary: A collection of Python integrations and utilities (Feishu, Dida365, VictoriaMetrics, ...)
 Author-email: mingming hou <houm01@foxmail.com>
 License: MIT

{pytbox-0.0.6 → pytbox-0.0.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pytbox"
-version = "0.0.6"
+version = "0.0.7"
 description = "A collection of Python integrations and utilities (Feishu, Dida365, VictoriaMetrics, ...)"
 authors = [{ name = "mingming hou", email = "houm01@foxmail.com" }]
 license = { text = "MIT" }

{pytbox-0.0.6 → pytbox-0.0.7}/src/pytbox/categraf/build_config.py

@@ -26,8 +26,19 @@ class BuildConfig:
         self.instances = load_config_by_file(instances)
         self.output_dir = output_dir
     
-    
-    
+    def _get_template(self, template_name):
+        return env.get_template(template_name)
+
+    def common(self, input_name):
+        template = self._get_template(f'input.{input_name}/{input_name}.toml.j2')
+        render_data = template.render()
+        target_dir = Path(self.output_dir) / f'input.{input_name}'
+        if not target_dir.exists():
+            target_dir.mkdir(parents=True, exist_ok=True)
+            
+        with open(Path(self.output_dir) / f'input.{input_name}' / f'{input_name}.toml', 'w', encoding='utf-8') as f:
+            f.write(render_data)
+
     def ping(self):
         instances = self.instances['ping']['instance']
         render_data = ping_template.render(instances=instances)
@@ -44,10 +55,26 @@ class BuildConfig:
         target_dir = Path(self.output_dir) / 'input.prometheus'
         if not target_dir.exists():
             target_dir.mkdir(parents=True, exist_ok=True)
-            
         with open(Path(self.output_dir) / 'input.prometheus' / 'prometheus.toml', 'w', encoding='utf-8') as f:
             f.write(render_data)
 
+    def vsphere(self):
+        template = self._get_template('input.vsphere/vsphere.toml.j2')
+        instances = self.instances['vsphere']['instance']
+        print(instances)
+        render_data = template.render(instances=instances)
+        target_dir = Path(self.output_dir) / 'input.vsphere'
+        if not target_dir.exists():
+            target_dir.mkdir(parents=True, exist_ok=True)
+        with open(Path(self.output_dir) / 'input.vsphere' / 'vsphere.toml', 'w', encoding='utf-8') as f:
+            f.write(render_data)
+
     def run(self):
+        # self.common('cpu')
+        # self.common('mem')
+        # self.common('net')
+        # self.common('disk')
+        # self.common('diskio')
+        self.vsphere()
         # self.ping()
-        self.prometheus()        
+        # self.prometheus()   

{pytbox-0.0.6 → pytbox-0.0.7}/src/pytbox.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytbox
-Version: 0.0.6
+Version: 0.0.7
 Summary: A collection of Python integrations and utilities (Feishu, Dida365, VictoriaMetrics, ...)
 Author-email: mingming hou <houm01@foxmail.com>
 License: MIT