ContainerImages
/
prometheus-frigate-exporter
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

log and exit if FRIGATE_STATS_URL not provided

main
Rhys Bailey 2023-12-13 22:22:14 +11:00
parent ad688df46a
commit dc7e614e0f
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
prometheus_frigate_exporter.py
View File

@ -1,12 +1,13 @@
from prometheus_client.core import GaugeMetricFamily, InfoMetricFamily, REGISTRY
from prometheus_client import start_http_server
from urllib.request import urlopen
import json import json
import re import re
import time import time
import sys import sys
import logging import logging
import os import os
from urllib.request import urlopen
from urllib import error
from prometheus_client.core import GaugeMetricFamily, InfoMetricFamily, REGISTRY
from prometheus_client import start_http_server
def add_metric(metric, label, stats, key, multiplier=1.0): def add_metric(metric, label, stats, key, multiplier=1.0):
@ -199,21 +200,20 @@ class CustomCollector(object):
if __name__ == '__main__': if __name__ == '__main__':
logging.basicConfig(format='%(asctime)s - %(message)s', level=logging.INFO) logging.basicConfig(format='%(asctime)s - %(message)s', level=logging.INFO)
try: try:
url = str(os.environ.get('FRIGATE_STATS_URL', 'http://localhost:5000/api/stats')) url = os.environ['FRIGATE_STATS_URL']
REGISTRY.register(CustomCollector(url)) except KeyError:
except IndexError:
logging.error( logging.error(
"Provide Frigate stats url as environment variable to container, " "Provide Frigate stats url as environment variable to container, "
"e.g. FRIGATE_STATS_URL=http://<your-frigate-ip>:5000/api/stats") "e.g. FRIGATE_STATS_URL=http://<your-frigate-ip>:5000/api/stats")
exit() sys.exit()
REGISTRY.register(CustomCollector(url))
port = int(os.environ.get('PORT', 9100)) port = int(os.environ.get('PORT', 9100))
start_http_server(port) start_http_server(port)
logging.info('Started, Frigate API URL: ' + url) logging.info('Started, Frigate API URL: %s', url)
logging.info('Metrics at: http://localhost:' + str(port) + '/metrics') logging.info('Metrics at: http://localhost:%d/metrics', port)
while True: while True:
time.sleep(1) time.sleep(1)