flexmetric 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

flexmetric/metric_process/prometheus_agent.py

@@ -96,7 +96,8 @@ def measure(init_flag,args):
         exec_result.extend(function_results)
     if args.commands:
         cmd_results = process_commands(args.commands_config)
-        exec_result.extend(cmd_results)
+        if cmd_results != None:
+            exec_result.extend(cmd_results)
     # exec_result = process_commands('commands.yaml')
     # print(exec_result)
     global gauges
@@ -138,4 +139,5 @@ def main():
     while True:
         measure(flag,args)
         flag = False
-        time.sleep(5)
+        time.sleep(5)
+main()

{flexmetric-0.1.3.dist-info → flexmetric-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flexmetric
-Version: 0.1.3
+Version: 0.1.4
 Summary: A flexible Prometheus exporter for commands, databases, functions, and scripts.
 Home-page: https://github.com/nikhillingadhal1999/custom_prometheus_agent
 Author: Nikhil Lingadhal

{flexmetric-0.1.3.dist-info → flexmetric-0.1.4.dist-info}/RECORD

@@ -7,10 +7,9 @@ flexmetric/logging_module/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 flexmetric/logging_module/logger.py,sha256=hXj9m2Q_KxJVI5YRHRoK6PXV5tO6NmvuVjq2Ipx_1tE,447
 flexmetric/metric_process/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flexmetric/metric_process/database_processing.py,sha256=OAJ0FbvjQLSqUh85KenuEQ5YG7ghWQYT-pzqogCOafk,2542
-flexmetric/metric_process/process_commands.py,sha256=cHtoyzP0IgZOwoS_BgFZU0VHuOKQO0UVvGOCwWuByNM,2236
-flexmetric/metric_process/prometheus_agent.py,sha256=1C18_n4qsqkxim20raD2LAZJm1zDd9AEPpVmuxDC9MA,5359
-flexmetric-0.1.3.dist-info/METADATA,sha256=FAXGW1RX6ZaOQpbGnNMyIwi_x74UyHwIKmag1ijq7y4,5373
-flexmetric-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flexmetric-0.1.3.dist-info/entry_points.txt,sha256=urVePn5EWr3JqNvkYP7OsB_h2_Bqvv-Wq1MJRBexm8A,79
-flexmetric-0.1.3.dist-info/top_level.txt,sha256=zBlrNwKhXUNhgu9RRZnXxYwYnmE-eocRe6wKSmQROA4,11
-flexmetric-0.1.3.dist-info/RECORD,,
+flexmetric/metric_process/prometheus_agent.py,sha256=pZ0Czv1ZQvyH0JhuNac5rjP1PC0ixcXM3HsQ0J-WxaY,5402
+flexmetric-0.1.4.dist-info/METADATA,sha256=adZgMG3ZOx9mlJwhCy3GHdih68q4CJODNb0MxKCSt8Y,5373
+flexmetric-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flexmetric-0.1.4.dist-info/entry_points.txt,sha256=urVePn5EWr3JqNvkYP7OsB_h2_Bqvv-Wq1MJRBexm8A,79
+flexmetric-0.1.4.dist-info/top_level.txt,sha256=zBlrNwKhXUNhgu9RRZnXxYwYnmE-eocRe6wKSmQROA4,11
+flexmetric-0.1.4.dist-info/RECORD,,

flexmetric/metric_process/process_commands.py

@@ -1,73 +0,0 @@
-import subprocess
-import yaml
-import re
-
-# 1. Read YAML commands
-def read_commands_from_yaml(file_path):
-    with open(file_path, 'r') as f:
-        config = yaml.safe_load(f)
-    return config.get('commands', [])
-
-# 2. Execute command with timeout
-def execute_command_with_timeout(command, timeout):
-    try:
-        result = subprocess.run(command, shell=True, capture_output=True, text=True, timeout=timeout)
-        if result.returncode != 0:
-            return ''
-        return result.stdout.strip()
-    except subprocess.TimeoutExpired:
-        return ''
-
-def process_commands(config_file):
-    commands = read_commands_from_yaml(config_file)
-    all_results = []
-
-    for cmd_info in commands:
-        command = cmd_info['command']
-        label_name = cmd_info['label']
-        timeout = cmd_info.get('timeout_seconds', 30)
-        label_column = cmd_info.get('label_column', -1)
-        value_column = cmd_info.get('value_column', 0)
-        fixed_label_value = cmd_info.get('label_value')
-
-        raw_output = execute_command_with_timeout(command, timeout)
-        lines = raw_output.strip().splitlines()
-
-        result_list = []
-
-        for line in lines:
-            parts = line.strip().split()
-            if not parts:
-                continue
-            if label_column == 'fixed':
-                label = fixed_label_value or 'label'
-            else:
-                try:
-                    label = parts[label_column]
-                except IndexError:
-                    label = 'unknown'
-            try:
-                raw_value = parts[value_column]
-                cleaned_value = re.sub(r'[^\d\.\-]', '', raw_value)
-                value = float(cleaned_value) if cleaned_value else 1
-            except (IndexError, ValueError):
-                value = 1
-
-            result_list.append({
-                'label': label,
-                'value': value
-            })
-
-        formatted = {
-            'result': result_list,
-            'labels': [label_name]
-        }
-        all_results.append(formatted)
-
-    return all_results
-
-
-# # Example usage:
-# if __name__ == "__main__":
-#     results = process_commands('/Users/nlingadh/code/custom_prometheus_agent/src/commands.yaml')
-#     print(results)