/
exporter.py
137 lines (116 loc) · 4.62 KB
/
exporter.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
from prometheus_client import start_http_server, Gauge
import re
import time
import os
import subprocess
import argparse
TEMP_REGEX = re.compile('t=-?([0-9]+)')
INTERNAL_TEMP_REGEX = re.compile('temp=([0-9]+\.[0-9])')
SENSOR_PATH = "/sys/bus/w1/devices"
CRC_ERROR="Could not read sensor. Wrong CRC."
DEFAULT_VALUE_ERROR="Could not read sensor. Sensor returned default value."
def read_command_arguments():
parser = argparse.ArgumentParser(description="Exporter of connected 1-Wire Sensors to Prometheus.io monitoring.")
parser.add_argument("--port",
metavar="PORT",
type=int,
default=8001,
help="The port used to export the temperatures to prometheus.io. Default is 8001.")
parser.add_argument("--internal",
default=False,
action="store_true",
help="Export the internal CPU Temperature of the Raspberry Pi. Requires the vcgencmd.")
return parser.parse_args()
def find_sensors():
ret = []
for device in os.listdir(SENSOR_PATH):
if "w1_bus_master" not in device:
sensor_under_test = Sensor(device)
test_value, error = sensor_under_test.read_sensor()
if test_value is not None or ( error == CRC_ERROR or error == DEFAULT_VALUE_ERROR):
ret.append(sensor_under_test)
return ret
def get_callback(device):
def read():
return read_sensor(device)
return read
class Sensor (object):
def __init__(self, id):
self.id = id
self.last_value = 0
self.miss_reads = 0
self.error_gauge = None
def set_error_gauge(self, gauge):
self.error_gauge = gauge
self.unset_missread()
def set_missread(self):
if self.error_gauge is not None:
self.error_gauge.set(1)
def unset_missread(self):
if self.error_gauge is not None:
self.error_gauge.set(0)
def __call__(self, *args, **kwargs):
value, _ = self.read_sensor()
if value is None:
self.miss_reads +=1
if self.miss_reads > 5:
return 85
return self.last_value
else:
self.miss_reads = 0
self.last_value = value
return value
def __repr__(self):
return "Sensor ID: " + self.__str__()
def __str__(self):
return self.id
def read_sensor(self):
try:
with open("%s/%s/w1_slave" % (SENSOR_PATH, self.id), "r") as sensor:
crc = l = None
try:
crc = sensor.readline()
if "YES" not in crc:
raise Exception(CRC_ERROR)
l = sensor.readline()
match = TEMP_REGEX.search(l)
ret_val = float(match.group(1))/1000
if ret_val == 85:
raise Exception(DEFAULT_VALUE_ERROR)
else:
self.unset_missread()
return ret_val, ""
except Exception, e:
print self.id, "Ups something went wrong.", e.message
if crc is not None:
print "CRC Line: ", crc
if l is not None:
print "l Line: ", l
self.set_missread()
return None, e.message
except IOError, e:
print e.message
self.set_missread()
return None, e.message
def read_raspberry_pi_temperature():
p = subprocess.Popen(["/usr/bin/vcgencmd", "measure_temp"], stdout=subprocess.PIPE)
match = INTERNAL_TEMP_REGEX.search(p.stdout.read())
return float(match.group(1))
def register_prometheus_gauges(export_internal_raspberry=False):
g = Gauge("sensor_temperature_in_celsius", "Local room temperature around the raspberry pi", ["sensor"])
error_g = Gauge("faulty_sensor_read", "Is 1 if the sensor could not be read.", ["sensor"])
sensors = find_sensors()
print "Found sensors:", ", ".join(map(lambda x: str(x), sensors))
for sensor in sensors:
g.labels(str(sensor)).set_function(sensor)
sensor.set_error_gauge(error_g.labels(str(sensor)))
if export_internal_raspberry:
g = Gauge("cpu_temperature_in_celsius", "CPU Temperature of the Raspberry Pi")
g.set_function(read_raspberry_pi_temperature)
return sensors
if __name__ == "__main__":
args = read_command_arguments()
start_http_server(args.port)
register_prometheus_gauges(args.internal)
while True:
time.sleep(10000)