8898a030c1e2e258a53ae3c5931fe8b4967ff157
[vuplus_dvbapp] / lib / python / Components / Sensors.py
1 from Components.FanControl import fancontrol
2
3 class Sensors:
4         # (type, name, unit, directory)
5         TYPE_TEMPERATURE = 0
6         # (type, name, unit, fanid)
7         TYPE_FAN_RPM = 1
8         
9         def __init__(self):
10                 # (type, name, unit, sensor_specific_dict/list)
11                 self.sensors_list = []
12                 self.addSensors()
13                 
14         def getSensorsCount(self, type = None):
15                 if type is None:
16                         return len(self.sensors_list)
17                 count = 0
18                 for sensor in self.sensors_list:
19                         if sensor[0] == type:
20                                 count += 1
21                 return count
22         
23         # returns a list of sensorids of type "type"
24         def getSensorsList(self, type = None):
25                 if type is None:
26                         return range(len(self.sensors_list))
27                 list = []
28                 for sensorid in range(len(self.sensors_list)):
29                         if self.sensors_list[sensorid][0] == type:
30                                 list.append(sensorid)
31                 return list
32         
33         
34         def getSensorType(self, sensorid):
35                 return self.sensors_list[sensorid][0]
36         
37         def getSensorName(self, sensorid):
38                 return self.sensors_list[sensorid][1]
39         
40         def getSensorValue(self, sensorid):
41                 value = -1
42                 sensor = self.sensors_list[sensorid]
43                 if sensor[0] == self.TYPE_TEMPERATURE:
44                         f = open("%s/value" % sensor[3], "r")
45                         value = int(f.readline().strip())
46                         f.close()
47                 elif sensor[0] == self.TYPE_FAN_RPM:
48                         value = fancontrol.getFanSpeed(sensor[3])
49                 return value
50         
51         def getSensorUnit(self, sensorid):
52                 return self.sensors_list[sensorid][2]
53
54         def addSensors(self):
55                 import os
56                 if os.path.exists("/proc/stb/sensors"):
57                         for dirname in os.listdir("/proc/stb/sensors"):
58                                 if dirname.find("temp", 0, 4) == 0:
59                                         f = open("/proc/stb/sensors/%s/name" % dirname, "r")
60                                         name = f.readline().strip()
61                                         f.close()
62                                         
63                                         f = open("/proc/stb/sensors/%s/unit" % dirname, "r")
64                                         unit = f.readline().strip()
65                                         f.close()
66                                         
67                                         self.sensors_list.append((self.TYPE_TEMPERATURE, name, unit, "/proc/stb/sensors/%s" % dirname))
68                 for fanid in range(fancontrol.getFanCount()):
69                         if fancontrol.hasRPMSensor(fanid):
70                                 self.sensors_list.append((self.TYPE_FAN_RPM, _("Fan %d") % (fanid + 1), "rpm", fanid))
71         
72 sensors = Sensors()