From 8020f71afd34d7696d7933659df2d763ab05542f Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sat, 4 May 2024 16:31:17 +0200
Subject: Adding upstream version 1.37.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 collectors/python.d.plugin/pandas/pandas.chart.py | 89 +++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 collectors/python.d.plugin/pandas/pandas.chart.py

(limited to 'collectors/python.d.plugin/pandas/pandas.chart.py')

diff --git a/collectors/python.d.plugin/pandas/pandas.chart.py b/collectors/python.d.plugin/pandas/pandas.chart.py
new file mode 100644
index 0000000..8eb4452
--- /dev/null
+++ b/collectors/python.d.plugin/pandas/pandas.chart.py
@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+# Description: pandas netdata python.d module
+# Author: Andrew Maguire (andrewm4894)
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+import pandas as pd
+
+try:
+    import requests
+    HAS_REQUESTS = True
+except ImportError:
+    HAS_REQUESTS = False
+
+from bases.FrameworkServices.SimpleService import SimpleService
+
+ORDER = []
+
+CHARTS = {}
+
+
+class Service(SimpleService):
+    def __init__(self, configuration=None, name=None):
+        SimpleService.__init__(self, configuration=configuration, name=name)
+        self.order = ORDER
+        self.definitions = CHARTS
+        self.chart_configs = self.configuration.get('chart_configs', None)
+        self.line_sep = self.configuration.get('line_sep', ';')
+
+    def run_code(self, df_steps):
+        """eval() each line of code and ensure the result is a pandas dataframe"""
+
+        # process each line of code
+        lines = df_steps.split(self.line_sep)
+        for line in lines:
+            line_clean = line.strip('\n').strip(' ')
+            if line_clean != '' and line_clean[0] != '#':
+                df = eval(line_clean)
+                assert isinstance(df, pd.DataFrame), 'The result of each evaluated line of `df_steps` must be of type `pd.DataFrame`'
+
+        # take top row of final df as data to be collected by netdata
+        data = df.to_dict(orient='records')[0]
+
+        return data
+
+    def check(self):
+        """ensure charts and dims all configured and that we can get data"""
+
+        if not HAS_REQUESTS:
+            self.warn('requests library could not be imported')
+
+        if not self.chart_configs:
+            self.error('chart_configs must be defined')
+
+        data = dict()
+
+        # add each chart as defined by the config
+        for chart_config in self.chart_configs:
+            if chart_config['name'] not in self.charts:
+                chart_template = {
+                    'options': [
+                        chart_config['name'],
+                        chart_config['title'],
+                        chart_config['units'],
+                        chart_config['family'],
+                        chart_config['context'],
+                        chart_config['type']
+                        ],
+                    'lines': []
+                    }
+                self.charts.add_chart([chart_config['name']] + chart_template['options'])
+
+            data_tmp = self.run_code(chart_config['df_steps'])
+            data.update(data_tmp)
+
+            for dim in data_tmp:
+                self.charts[chart_config['name']].add_dimension([dim, dim, 'absolute', 1, 1])
+
+        return True
+
+    def get_data(self):
+        """get data for each chart config"""
+
+        data = dict()
+
+        for chart_config in self.chart_configs:
+            data_tmp = self.run_code(chart_config['df_steps'])
+            data.update(data_tmp)
+
+        return data
-- 
cgit v1.2.3