From fe1042902a06fb2ff6ee2eca3cba63da65d408a8 Mon Sep 17 00:00:00 2001
From: Volodymyr Boyko <volodymyrx.boiko@intel.com>
Date: Thu, 11 Mar 2021 02:14:51 +0630
Subject: [PATCH] Add thermals

Signed-off-by: Volodymyr Boyko <volodymyrx.boiko@intel.com>
---
 .../sonic_platform/chassis.py                 |   2 +
 .../sonic_platform/thermal.py                 | 109 ++++++++++++++++++
 2 files changed, 111 insertions(+)
 create mode 100644 platform/barefoot/sonic-platform-modules-bfn-montara/sonic_platform/thermal.py

diff --git a/platform/barefoot/sonic-platform-modules-bfn-montara/sonic_platform/chassis.py b/platform/barefoot/sonic-platform-modules-bfn-montara/sonic_platform/chassis.py
index fd8f5a9370e2..42364d6e5688 100644
--- a/platform/barefoot/sonic-platform-modules-bfn-montara/sonic_platform/chassis.py
+++ b/platform/barefoot/sonic-platform-modules-bfn-montara/sonic_platform/chassis.py
@@ -6,6 +6,7 @@
     from sonic_platform.sfp import Sfp
     from sonic_platform.psu import Psu
     from sonic_platform.fan_drawer import fan_drawer_list_get
+    from sonic_platform.thermal import thermal_list_get
     from eeprom import Eeprom
 except ImportError as e:
     raise ImportError(str(e) + "- required module not found")
@@ -28,6 +29,7 @@ def __init__(self):
             self._psu_list.append(psu)
 
         self._fan_drawer_list = fan_drawer_list_get()
+        self._thermal_list = thermal_list_get()
 
     def get_name(self):
         """
diff --git a/platform/barefoot/sonic-platform-modules-bfn-montara/sonic_platform/thermal.py b/platform/barefoot/sonic-platform-modules-bfn-montara/sonic_platform/thermal.py
new file mode 100644
index 000000000000..3a39c10a1e1f
--- /dev/null
+++ b/platform/barefoot/sonic-platform-modules-bfn-montara/sonic_platform/thermal.py
@@ -0,0 +1,109 @@
+try:
+    import subprocess
+
+    from sonic_platform.bfn_extensions.platform_sensors import platform_sensors_get
+    from sonic_platform_base.thermal_base import ThermalBase
+except ImportError as e:
+    raise ImportError (str(e) + "- required module not found")
+
+'''
+data argument is in "sensors -A -u" format, example:
+coretemp-isa-0000
+Package id 0:
+  temp1_input: 37.000
+  temp1_max: 82.000
+  temp1_crit: 104.000
+  temp1_crit_alarm: 0.000
+Core 0:
+  temp2_input: 37.000
+  ...
+'''
+def _sensors_chip_parsed(data: str):
+    def kv(line):
+        k, v, *_ = [t.strip(': ') for t in line.split(':') if t] + ['']
+        return k, v
+
+    chip, *data = data.strip().split('\n')
+    chip = chip.strip(': ')
+
+    sensors = []
+    for line in data:
+        if not line.startswith(' '):
+            sensor_label = line.strip(': ')
+            sensors.append((sensor_label, {}))
+            continue
+
+        if len(sensors) == 0:
+            raise RuntimeError(f'invalid data to parse: {data}')
+
+        attr, value = kv(line)
+        sensor_label, sensor_data = sensors[-1]
+        sensor_data.update({attr: value})
+
+    return chip, dict(sensors)
+
+'''
+Example of returned dict:
+{
+    'coretemp-isa-0000': {
+        'Core 1': { "temp1_input": 40, ...  },
+        'Core 2': { ... }
+    }
+}
+'''
+def _sensors_get() -> dict:
+    data = platform_sensors_get(['-A', '-u']) or ''
+    data += subprocess.check_output("/usr/bin/sensors -A -u",
+                shell=True, text=True)
+    data = data.split('\n\n')
+    data = [_sensors_chip_parsed(chip_data) for chip_data in data if chip_data]
+    data = dict(data)
+    return data
+
+def _value_get(d: dict, key_prefix, key_suffix=''):
+    for k, v in d.items():
+        if k.startswith(key_prefix) and k.endswith(key_suffix):
+            return v
+    return None
+
+# Thermal -> ThermalBase -> DeviceBase
+class Thermal(ThermalBase):
+    def __init__(self, chip, label):
+        self.__chip = chip
+        self.__label = label
+        self.__name = f"{chip}:{label}".lower().replace(' ', '-')
+
+    def __get(self, attr_prefix, attr_suffix):
+        sensor_data = _sensors_get().get(self.__chip, {}).get(self.__label, {})
+        value = _value_get(sensor_data, attr_prefix, attr_suffix)
+        if value is not None: return value
+        raise NotImplementedError
+
+    # ThermalBase interface methods:
+    def get_temperature(self) -> float:
+        return float(self.__get('temp', 'input'))
+
+    def get_high_threshold(self) -> float:
+        return float(self.__get('temp', 'max'))
+
+    def get_high_critical_threshold(self) -> float:
+        return float(self.__get('temp', 'crit'))
+
+    # DeviceBase interface methods:
+    def get_name(self):
+        return self.__name
+
+    def get_presence(self):
+        return True
+
+    def get_status(self):
+        return True
+
+def thermal_list_get():
+    l = []
+    for chip, chip_data in _sensors_get().items():
+        for sensor, sensor_data in chip_data.items():
+            # add only temperature sensors
+            if _value_get(sensor_data, "temp") is not None:
+                l.append(Thermal(chip, sensor))
+    return l