Example prometheus script for generating a file with metrics for textfile collector

example_textfile_metrics.py

#!/usr/bin/env python3

from prometheus_client import CollectorRegistry, Gauge, write_to_textfile
from psutil import process_iter
from socket import gethostname

hostname = gethostname()
pid_count = {}
for proc in process_iter(['pid', 'name', 'username']):
    if proc.info['username'] not in pid_count:
        pid_count[proc.info['username']] = 0

    pid_count[proc.info['username']] += 1 

registry = CollectorRegistry()
cnt = Gauge('user_pid_count', 'count of pids on the node', ['username', 'hostname'], registry=registry)
ts = Gauge('last_success_unixtime', 'last textfile export time ', registry=registry)

for user, count in pid_count.items():
    cnt.labels(user, hostname).set(count)


ts.set_to_current_time()
write_to_textfile('/tmp/user_pid_count.prom', registry)