noham/prometheus-qbittorrent-exporter
1
0
Fork 0
mirror of https://github.com/NohamR/prometheus-qbittorrent-exporter.git synced 2025-05-23 16:49:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #1 from Gabisonfire/master

Browse Source 
Added the ability to include uncategorized torrents.
This commit is contained in:
Esteban Sanchez 2020-10-22 23:11:35 +02:00 committed by GitHub
commit 53dc38f796
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
qbittorrent_exporter/exporter.py
View File

@ -3,6 +3,7 @@ import os
import sys
import signal
import faulthandler
from attrdict import AttrDict
from qbittorrentapi import Client, TorrentStates
from prometheus_client import start_http_server
from prometheus_client.core import GaugeMetricFamily, CounterMetricFamily, REGISTRY
@ -122,8 +123,9 @@ class QbittorrentMetricsCollector():
return []
metrics = []
categories.Uncategorized = AttrDict({'name': 'Uncategorized', 'savePath': ''})
for category in categories:
category_torrents = [t for t in self.torrents if t['category'] == category]
category_torrents = [t for t in self.torrents if t['category'] == category or (category == "Uncategorized" and t['category'] == "")]
for status in self.TORRENT_STATUSES:
status_prop = f"is_{status}"