simono41/prometheus-immich-exporter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Added scraping for uncategorized torrents

Browse source
This commit is contained in:
Gab 2020-10-20 02:05:17 -04:00
parent 0b3fb04728
commit 40d9bb964e
2 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
README.md
View file

@ -31,6 +31,7 @@ The application reads configuration using environment variables:
| `QBITTORRENT_PORT` | | qbittorrent server port |
| `QBITTORRENT_USER` | `""` | qbittorrent username |
| `QBITTORRENT_PASS` | `""` | qbittorrent password |
| `QBITTORRENT_INCLUDE_UNCATEGORIZED` | `"false"` | Include uncategorized torrents |
| `EXPORTER_PORT` | `8000` | Exporter listening port |
| `EXPORTER_LOG_LEVEL` | `INFO` | Log level. One of: `DEBUG`, `INFO`, `WARNING`, `ERROR`, `CRITICAL` |

12
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
@ -24,6 +25,7 @@ class QbittorrentMetricsCollector():
"errored",
"paused",
]
INCLUDE_UNCATEGORIZED = False
def __init__(self, config):
self.config = config
@ -34,6 +36,7 @@ class QbittorrentMetricsCollector():
username=config["username"],
password=config["password"],
)
self.INCLUDE_UNCATEGORIZED = config["include_uncategorized"] == 'true'
def collect(self):
try:
@ -122,6 +125,8 @@ class QbittorrentMetricsCollector():
return []
metrics = []
if self.INCLUDE_UNCATEGORIZED:
categories.Uncategorized = AttrDict({'name': 'Uncategorized', 'savePath': ''})
for category in categories:
category_torrents = [t for t in self.torrents if t['category'] == category]
@ -166,7 +171,8 @@ def main():
"username": os.environ.get("QBITTORRENT_USER", ""),
"password": os.environ.get("QBITTORRENT_PASS", ""),
"exporter_port": int(os.environ.get("EXPORTER_PORT", "8000")),
"log_level": os.environ.get("EXPORTER_LOG_LEVEL", "INFO")
"log_level": os.environ.get("EXPORTER_LOG_LEVEL", "INFO"),
"include_uncategorized": os.environ.get("QBITTORRENT_INCLUDE_UNCATEGORIZED", "false"),
}
# Register signal handler
@ -194,6 +200,10 @@ def main():
logger.info("Exporter is starting up")
REGISTRY.register(QbittorrentMetricsCollector(config))
logger.info(
f"Including uncategorized torrents: {config['include_uncategorized']}"
)
# Start server
start_http_server(config["exporter_port"])
logger.info(